ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
22d438ae9e02b674e1656263df07799ee06fb466
composable_kernel/device_operation/include
History
Chao Liu 22d438ae9e Add gridwise GEMM pipeline (#89) 
* clean up

* add mutilple thread scratch to ThreadwiseTensorSliceTransfer_v3r1

* add 2 stage prefetch

* add more sanity check into transform_tensor_descriptor

* tweak

* enabling 2 stage prefetch to exsiting gridwise gemm; tweak

* enabling 2 stage prefetch to exsiting gridwise gemm

* move gridwise gemm pipeline in class; clean up

* add some irregular tile size

* update CalculateHasMainK0BlockLoop for multi-stage-prefetch

* refactor gridwise gemm pipeline class
2022-02-23 17:23:49 -06:00
..
conv_utils.hpp
Unify Convolution FWD XDL 1D/2D implementation. (#93)
2022-02-23 10:44:20 -06:00
convolution_forward_specialization.hpp
Fusion Conv+Bias+ReLU(+Add) (#62)
2021-12-26 07:43:42 -07:00
convolution_utility.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_base.hpp
Fusion Conv+Bias+ReLU(+Add) (#62)
2021-12-26 07:43:42 -07:00
device_batched_gemm_xdl.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_conv2d_fwd_xdl_c_shuffle_bias_activation_add_nhwc_kyxc_nhwk.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_conv2d_fwd_xdl_c_shuffle_bias_activation_nhwc_kyxc_nhwk.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_conv2d_fwd_xdl_nhwc_kyxc_nhwk.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_conv3d_fwd_naive_ndhwc_kzyxc_ndhwk.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_conv3d_fwd_xdl_ndhwc_kzyxc_ndhwk.hpp
Unify Convolution FWD XDL 1D/2D implementation. (#93)
2022-02-23 10:44:20 -06:00
device_conv_fwd_bias_activation_add.hpp
Fusion Conv+Bias+ReLU(+Add) (#62)
2021-12-26 07:43:42 -07:00
device_conv_fwd_bias_activation.hpp
Fusion Conv+Bias+ReLU(+Add) (#62)
2021-12-26 07:43:42 -07:00
device_conv_fwd.hpp
Fusion Conv+Bias+ReLU(+Add) (#62)
2021-12-26 07:43:42 -07:00
device_convnd_fwd_xdl_nhwc_kyxc_nhwk.hpp
Unify Convolution FWD XDL 1D/2D implementation. (#93)
2022-02-23 10:44:20 -06:00
device_gemm_bias_activation_add.hpp
GEMM+Bias+ReLU+Add (#76)
2022-02-06 22:32:47 -06:00
device_gemm_bias_activation.hpp
GEMM+Bias+ReLU+Add (#76)
2022-02-06 22:32:47 -06:00
device_gemm_xdl_c_shuffle_bias_2d.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_gemm_xdl_c_shuffle_bias_activation_add.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_gemm_xdl_c_shuffle_bias_activation.hpp
Conv3d new (#94)
2022-02-22 22:45:28 -06:00
device_gemm_xdl_c_shuffle.hpp
Add gridwise GEMM pipeline (#89)
2022-02-23 17:23:49 -06:00
device_gemm_xdl_splitk.hpp
Add small tile size for fp16/fp32 and NN layout (#80)
2022-02-11 15:49:06 -06:00
device_gemm_xdl.hpp
Add gridwise GEMM pipeline (#89)
2022-02-23 17:23:49 -06:00
device_gemm.hpp
Support alpha beta scaling for GEMM (#78)
2022-02-11 00:48:41 -06:00
device_operation_instance.hpp
Fusion Conv+Bias+ReLU(+Add) (#62)
2021-12-26 07:43:42 -07:00
gemm_specialization.hpp
Add small tile size for fp16/fp32 and NN layout (#80)
2022-02-11 15:49:06 -06:00
tensor_layout.hpp
Unify Convolution FWD XDL 1D/2D implementation. (#93)
2022-02-23 10:44:20 -06:00