ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-02 20:51:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
781cacd2e60ffbf358aaeeeee315a9b9d69c43a6
composable_kernel/include/ck/tensor_operation/gpu/device
History
ltqin 781cacd2e6 NHWC Conv2d Bwd weight fp16 ckprofiler and test (#166) 
* change backward weight name

* start add bwd weight lib and profiler

* change tuning paramter

* change output info

* add bwd weight test

* change test info

* using conv_util

* change wgt to weight

* add }

* add fp32
2022-04-04 20:32:00 -05:00
..
conv_utils.hpp
Unified implementation of 1d/2d/3d conv bwd-data. fp32/fp16/bfp16/int8 (#134)
2022-03-29 10:52:25 -05:00
convolution_backward_data_specialization.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
convolution_forward_specialization.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
convolution_utility.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_base.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_batched_gemm_reduce_xdl_cshuffle.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_batched_gemm_xdl.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv2d_backward_weight_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
NHWC Conv2d Bwd weight fp16 ckprofiler and test (#166)
2022-04-04 20:32:00 -05:00
device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv2d_fwd_xdl_c_shuffle_bias_activation_add_nhwc_kyxc_nhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv2d_fwd_xdl_c_shuffle_bias_activation_nhwc_kyxc_nhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv2d_fwd_xdl_nhwc_kyxc_nhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv3d_fwd_naive_ndhwc_kzyxc_ndhwk.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_conv3d_fwd_xdl_ndhwc_kzyxc_ndhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_conv_backward_weight.hpp
NHWC Conv2d Bwd weight fp16 ckprofiler and test (#166)
2022-04-04 20:32:00 -05:00
device_conv_bwd_data.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_conv_fwd_bias_activation_add.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_conv_fwd_bias_activation.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_conv_fwd.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_convnd_bwd_data_xdl_ndhwc_kzyxc_ndhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_convnd_fwd_xdl_nhwc_kyxc_nhwk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_bias_activation_add.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_gemm_bias_activation.hpp
Reorganize files, Part 1 (#119)
2022-03-08 21:46:36 -06:00
device_gemm_bias.hpp
Gemm+Reduce Fusion (#128)
2022-03-23 22:18:42 -05:00
device_gemm_reduce_xdl_cshuffle.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_reduce.hpp
Batched gemm and reduction (#156)
2022-03-30 11:21:18 -05:00
device_gemm_xdl_c_shuffle_bias_2d.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_xdl_c_shuffle_bias_activation_add.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_xdl_c_shuffle_bias_activation.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_xdl_c_shuffle.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_xdl_cshuffle.hpp
Tune & add conflict-free LDS gemm kernels (#159)
2022-03-31 12:58:41 -05:00
device_gemm_xdl_splitk_c_shuffle.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_xdl_splitk.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm_xdl.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_gemm.hpp
batched_gemm: use profiler in ctest (#163)
2022-03-30 21:32:49 -05:00
device_grouped_gemm_xdl.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_pool2d_fwd_nhwc_nhwc.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_pool2d_fwd.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
device_reduce_blockwise_second_call.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
device_reduce_blockwise.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
device_reduce_common.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
device_reduce_multiblock_atomic_add.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
device_reduce_multiblock_partial_reduce.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
device_reduce_threadwise.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
device_reduce.hpp
Reduction for int8 and bfloat16 (#125)
2022-03-22 14:35:14 -05:00
gemm_specialization.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
reduction_operator_mapping.hpp
Compile for gfx908 and gfx90a (#130)
2022-03-31 12:33:34 -05:00
tensor_layout.hpp
batched_gemm: use profiler in ctest (#163)
2022-03-30 21:32:49 -05:00