ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-17 03:19:48 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
7217d2c92633f81df6e6d105403c04f573597fbf
composable_kernel/library/src/tensor_operation_instance/gpu
History
Andriy Roshchenko c0957c8334 Restrict MX GEMM instantiation to GFX950 arch (#2157) 
[ROCm/composable_kernel commit: 79beaacdd1]
2025-05-05 08:18:22 -07:00
..
avg_pool2d_bwd
Pool2d max/avg kernel in the BWD version (#1494)
2024-09-12 11:47:52 +02:00
avg_pool3d_bwd
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
batched_gemm
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
batched_gemm_add_relu_gemm_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_b_scale
Added Int4 mixed batch gemm support (#1839)
2025-02-10 11:17:02 +08:00
batched_gemm_bias_permute
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_gemm
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_multi_d
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
batched_gemm_reduce
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
batched_gemm_softmax_gemm
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
batched_gemm_softmax_gemm_permute
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
batchnorm
Refactor elementwise kernels (#1222)
2024-04-19 13:31:17 +02:00
column_to_image
Add support for groups in Img2Col/Col2Img (#1007)
2023-10-31 10:46:32 +01:00
contraction_bilinear
Extend support for contraction 6D (#1207)
2024-04-09 23:46:21 +02:00
contraction_scale
Extend support for contraction 6D (#1207)
2024-04-09 23:46:21 +02:00
conv1d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv2d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
conv2d_fwd
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
conv2d_fwd_bias_relu
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
conv2d_fwd_bias_relu_add
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
conv3d_bwd_data
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
elementwise
Refactor elementwise kernels (#1222)
2024-04-19 13:31:17 +02:00
elementwise_normalization
Refactoring cmake files to build data types separately. (#932)
2023-09-20 22:15:56 -07:00
gemm
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_ab_scale
[Block Scale GEMM] Optimized block scale gemm (#1950)
2025-03-11 10:11:21 -07:00
gemm_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_add_fastgelu
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_add_fastgelu
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_add_multiply
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_relu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_add_relu_add_layernorm
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_add_silu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_b_scale
Added Int4 mixed batch gemm support (#1839)
2025-02-10 11:17:02 +08:00
gemm_bias_add_reduce
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_bilinear
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_fastgelu
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_multi_abd
Split the gemm_multi_abd instances. (#1306)
2024-05-23 09:17:02 -07:00
gemm_multiply_add
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_multiply_multiply
[Block Scale GEMM] Optimized block scale gemm (#1950)
2025-03-11 10:11:21 -07:00
gemm_multiply_multiply_wp
New instances for gemm_multiply_multiply_weightpreshuffle operator (#2061)
2025-04-08 15:14:53 -07:00
gemm_mx
Restrict MX GEMM instantiation to GFX950 arch (#2157)
2025-05-05 08:18:22 -07:00
gemm_reduce
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_splitk
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_streamk
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
gemm_universal
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00
gemm_universal_batched
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_universal_reduce
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
gemm_universal_streamk
f8/bf16 GEMM Stream-K (#1879)
2025-03-31 20:30:17 -06:00
grouped_conv1d_bwd_weight
Enable grouped conv bwd wei bf16 NGCHW (#1589)
2024-10-22 16:18:28 +02:00
grouped_conv1d_fwd
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv2d_bwd_data
Integrate universal gemm with conv bwd data and add SplitK (#1315)
2025-04-28 23:54:49 +02:00
grouped_conv2d_bwd_weight
Add support for GKCYX grouped conv weight (#2023)
2025-04-02 23:59:49 +02:00
grouped_conv2d_fwd
Add grouped conv fwd 16x16 mfma instruction instances (#2140)
2025-04-30 09:49:37 +02:00
grouped_conv2d_fwd_dynamic_op
Temporary disable part of dynamic op conv instances (#1630)
2024-11-04 13:34:17 -08:00
grouped_conv3d_bwd_data
Integrate universal gemm with conv bwd data and add SplitK (#1315)
2025-04-28 23:54:49 +02:00
grouped_conv3d_bwd_data_bilinear
Integrate universal gemm with conv bwd data and add SplitK (#1315)
2025-04-28 23:54:49 +02:00
grouped_conv3d_bwd_data_scale
Integrate universal gemm with conv bwd data and add SplitK (#1315)
2025-04-28 23:54:49 +02:00
grouped_conv3d_bwd_weight
Add support for GKCYX grouped conv weight (#2023)
2025-04-02 23:59:49 +02:00
grouped_conv3d_bwd_weight_bilinear
Re-enable fp8 types for all architectures. (#1470)
2024-08-16 16:07:52 -06:00
grouped_conv3d_bwd_weight_scale
Re-enable fp8 types for all architectures. (#1470)
2024-08-16 16:07:52 -06:00
grouped_conv3d_fwd
Add grouped conv fwd 16x16 mfma instruction instances (#2140)
2025-04-30 09:49:37 +02:00
grouped_conv3d_fwd_bilinear
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_fwd_convinvscale
Add a convinvscale op, related instances and examples (#1307)
2024-06-10 14:48:49 -05:00
grouped_conv3d_fwd_convscale
Adding Instances and Examples for FP8-based Scaled Convolution and AMAX Reduction. (#1473)
2024-08-21 15:22:41 -07:00
grouped_conv3d_fwd_convscale_add
Adding more instances of grouped convolution 3d forward for FP8 with ConvScale+Bias element-wise operation. (#1412)
2024-07-24 15:49:55 -05:00
grouped_conv3d_fwd_convscale_relu
Adding Instances and Examples for FP8-based Scaled Convolution and AMAX Reduction. (#1473)
2024-08-21 15:22:41 -07:00
grouped_conv3d_fwd_dynamic_op
Temporary disable part of dynamic op conv instances (#1630)
2024-11-04 13:34:17 -08:00
grouped_conv3d_fwd_scale
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_fwd_scaleadd_ab
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_conv3d_fwd_scaleadd_scaleadd_relu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
grouped_gemm_bias
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm_fastgelu
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
grouped_gemm_fixed_nk
Added bf16 instances grouped gemm fixed nk (#1825)
2025-01-20 09:13:09 -08:00
grouped_gemm_fixed_nk_multi_abd
bf16A_Int8B with fastgelu/bias (#1264)
2024-04-26 07:26:30 -05:00
grouped_gemm_tile_loop
Remove CK_USE_AMD_MFMA_GFX950 (#1935)
2025-03-04 10:32:25 -08:00
image_to_column
Add support for groups in Img2Col/Col2Img (#1007)
2023-10-31 10:46:32 +01:00
max_pool_bwd
This commit contains implementation of max pool2d for f8 type (#1506)
2024-09-16 10:15:06 +02:00
mha
Refactor CI performance tests. (#1726)
2024-12-06 13:04:25 -08:00
normalization_bwd_data
layernorm and groupnorm backward data (#1083)
2023-12-19 04:23:11 +08:00
normalization_bwd_gamma_beta
layernorm & groupnorm bwd gamma beta (#1133)
2024-01-25 19:53:15 +08:00
normalization_fwd
layernorm and groupnorm backward data (#1083)
2023-12-19 04:23:11 +08:00
permute_scale
Adding Instances and Examples for FP8-based Scaled Convolution with ReLU Activation and AMAX Reduction. (#1469)
2024-08-20 10:30:56 -05:00
pool2d_fwd
Add pool2d int8 and fp8 instances (#1508)
2024-09-13 10:18:21 -07:00
pool3d_fwd
OCP FP8 support for gfx12. (#1710)
2024-12-03 08:42:55 -08:00
quantization
Split the instances by architecture. (#1223)
2024-04-02 09:42:17 -07:00
reduce
Adding Instances and Examples for FP8-based Scaled Convolution with ReLU Activation and AMAX Reduction. (#1469)
2024-08-20 10:30:56 -05:00
softmax
enable compilation of INSTANCES_ONLY for Windows (#1082)
2023-12-20 14:34:53 -08:00
transpose
Hip tensor permute unit test (#1068)
2023-12-18 21:35:00 -06:00
CMakeLists.txt
DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096)
2025-04-28 10:14:21 +05:00