ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 05:31:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d7d782909655d31ab5e125a9220c2a9396d1ff21
composable_kernel/include/ck/tensor_operation/gpu/device
History
zjing14 d7d7829096 Batched Gemm with multiD (#329) 
* add batched_gemm_multiD

* add ds

* rename file

* add batched_gemm_bias example

* add batch_strides into bmm_c_permute

* clean

* rename example_28 to example_29

Co-authored-by: Chao Liu <chao.liu2@amd.com>
2022-07-22 09:33:50 -05:00
..
convolution_backward_data_specialization.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
convolution_backward_weight_specialization.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
convolution_forward_specialization.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
device_5ary_elementwise.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
device_base.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_batched_gemm_c_permute_xdl.hpp
Batched Gemm with multiD (#329)
2022-07-22 09:33:50 -05:00
device_batched_gemm_c_permute.hpp
Batched Gemm with multiD (#329)
2022-07-22 09:33:50 -05:00
device_batched_gemm_multi_d_xdl.hpp
Batched Gemm with multiD (#329)
2022-07-22 09:33:50 -05:00
device_batched_gemm_multi_d.hpp
Batched Gemm with multiD (#329)
2022-07-22 09:33:50 -05:00
device_batched_gemm_reduce_xdl_cshuffle.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
device_batched_gemm_xdl.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
device_batched_gemm.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
device_binary_elementwise.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
device_cgemm_4gemm_xdl_cshuffle.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_cgemm.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_contraction_multiple_d_xdl_cshuffle.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
device_contraction_multiple_d.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
device_conv2d_backward_weight_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv2d_bwd_data_xdl_nhwc_kyxc_nhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv2d_fwd_xdl_c_shuffle_bias_activation_add_nhwc_kyxc_nhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv2d_fwd_xdl_c_shuffle_bias_activation_nhwc_kyxc_nhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv2d_fwd_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
device_conv2d_fwd_xdl_nhwc_kyxc_nhwk.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
device_conv3d_fwd_naive_ndhwc_kzyxc_ndhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv3d_fwd_xdl_ndhwc_kzyxc_ndhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv_backward_weight.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv_bwd_data.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv_fwd_bias_activation_add.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv_fwd_bias_activation.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_conv_fwd.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_convnd_backward_weight_xdl_c_shuffle_nhwc_kyxc_nhwk.hpp
add conv1d/3d bwd weight instances (#318)
2022-07-08 15:42:20 -05:00
device_convnd_bwd_data_xdl_ndhwc_kzyxc_ndhwk.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_convnd_fwd_xdl_nhwc_kyxc_nhwk.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
device_elementwise.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
device_gemm_bias_add_reduce_xdl_cshuffle.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
device_gemm_bias_c_permute_xdl.hpp
Gemm + bias + c_permute (#312)
2022-06-30 19:55:09 -05:00
device_gemm_bias_c_permute.hpp
Gemm + bias + c_permute (#312)
2022-06-30 19:55:09 -05:00
device_gemm_dl.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm_multiple_d_xdl_cshuffle.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
device_gemm_multiple_d.hpp
N-D Tensor Contraction example, instance, and client example (#270)
2022-07-07 14:31:11 -05:00
device_gemm_reduce_xdl_cshuffle.hpp
external api for gemm + layernorm (#285)
2022-06-27 14:25:10 -05:00
device_gemm_reduce.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm_splitk.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm_xdl_cshuffle.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm_xdl_layernorm_cshuffle.hpp
Single-kernel GEMM + layernorm (#263)
2022-07-01 01:38:00 -05:00
device_gemm_xdl_splitk_c_shuffle.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm_xdl_splitk.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm_xdl.hpp
Improve external interface for GEMM and GEMM+add+add+fastgelu (#311)
2022-06-30 22:11:00 -05:00
device_gemm.hpp
Grouped Gemm device with multiD grid (#319)
2022-07-21 10:07:01 -05:00
device_grouped_gemm_xdl.hpp
Grouped Gemm device with multiD grid (#319)
2022-07-21 10:07:01 -05:00
device_grouped_gemm.hpp
Grouped Gemm device with multiD grid (#319)
2022-07-21 10:07:01 -05:00
device_layernorm.hpp
Standalone layernorm (#315)
2022-07-13 11:16:14 -05:00
device_normalization.hpp
Standalone sweep once softmax kernel w/ ckProfiler (#295)
2022-06-30 12:08:50 -05:00
device_pool2d_fwd_nhwc_nhwc.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_pool2d_fwd.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_reduce_common.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_reduce_multiblock.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_reduce_threadwise.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_reduce.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
device_softmax.hpp
Standalone sweep once softmax kernel w/ ckProfiler (#295)
2022-06-30 12:08:50 -05:00
device_unary_elementwise.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
gemm_specialization.hpp
Gemm+Bilinear (#316)
2022-07-02 09:15:38 -05:00
reduction_operator_mapping.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00
tensor_layout.hpp
add license in file (#303)
2022-06-24 23:32:43 -05:00