ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 09:16:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
595d23be14d3a23a0df1aae8350497f47ba04137
composable_kernel/library/src/tensor_operation_instance/gpu
History
Rostyslav Geyyer eac50708d9 Add instances/ckProfiler/client example for fp8/fp16 mixed precision Gemm (#853) 
* Add ComputeType arg to splitk device and gridwise ops

* Update for gridwise op compatibility

* Update bf16 and int8 splitk gemm examples with ComputeType

* Add instances

* Update ckProfiler for mixed precision cases

* Add a mixed precision splitK gemm client example

---------

Co-authored-by: zjing14 <zhangjing14@gmail.com>
2023-08-22 09:34:49 -05:00
..
batched_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batched_gemm_add_relu_gemm_add
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batched_gemm_bias_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batched_gemm_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batched_gemm_multi_d
Disable DL kernels by default. (#816)
2023-07-26 11:06:45 -05:00
batched_gemm_reduce
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batched_gemm_softmax_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batched_gemm_softmax_gemm_permute
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
batchnorm
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
contraction_bilinear
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
contraction_scale
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
conv1d_bwd_data
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
conv2d_bwd_data
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
conv2d_fwd
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
conv2d_fwd_bias_relu
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
conv2d_fwd_bias_relu_add
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
conv3d_bwd_data
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
elementwise
Padded Generic Kernel Instance (#730)
2023-06-16 23:43:11 -05:00
elementwise_normalization
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
gemm
Implement DPP8 based GEMM for Navi21 (#826)
2023-08-14 15:46:27 -05:00
gemm_add_add_fastgelu
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
gemm_add_fastgelu
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
gemm_add_multiply
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gemm_add_relu_add_layernorm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
gemm_bias_add_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gemm_bilinear
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
gemm_fastgelu
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
gemm_reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
gemm_splitk
Add instances/ckProfiler/client example for fp8/fp16 mixed precision Gemm (#853)
2023-08-22 09:34:49 -05:00
gemm_streamk
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
grouped_conv1d_bwd_weight
Support NHWGC conv2d_bwd_weight (#769)
2023-07-12 08:25:02 -05:00
grouped_conv1d_fwd
Enable grouped conv with small K or C (#822)
2023-08-09 10:40:55 -05:00
grouped_conv2d_bwd_data
Grouped 3d conv backward data support (#799)
2023-07-18 11:01:33 -05:00
grouped_conv2d_bwd_weight
Support NHWGC conv2d_bwd_weight (#769)
2023-07-12 08:25:02 -05:00
grouped_conv2d_fwd
Enable grouped conv with small K or C (#822)
2023-08-09 10:40:55 -05:00
grouped_conv3d_bwd_data
Grouped 3d conv backward data support (#799)
2023-07-18 11:01:33 -05:00
grouped_conv3d_bwd_weight
Grouped conv bwd wei NDHWGC/NDHWGK (#804)
2023-07-21 12:00:55 -05:00
grouped_conv3d_fwd
Enable grouped conv with small K or C (#822)
2023-08-09 10:40:55 -05:00
grouped_gemm
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
grouped_gemm_fastgelu
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
normalization
Add Normalization splitk instances (#829)
2023-08-12 01:31:31 +08:00
pool3d_fwd
Refactor pool fwd (#815)
2023-08-15 02:25:28 +08:00
quantization
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
reduce
update copyright headers (#726)
2023-05-31 18:46:57 -05:00
softmax
Allow building CK for specific data types and split off last remaining DL instances. (#830)
2023-08-07 14:56:10 -07:00
CMakeLists.txt
Disable DL kernels by default. (#816)
2023-07-26 11:06:45 -05:00