ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 10:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
80117e7ecc0c378c3afd6c5c3f8f42e7e9690751
composable_kernel/include/ck/tensor_operation/gpu/device
History
Johannes Graner 80117e7ecc [CK Grouped Gemm] Fix workspace stride in two stage kernel (#3412) 
* Use correct workspace stride

* Use correct stride in elementwise kernel

* Fix test by adding padder

* No UTF-8 in comments

* Remove unnecessary changes

* Remove more unnecessary changes

* Use non-padded strides for workspace

* Disable two stage kernel for RRR+MNKPadding+kbatch>2

Partially fixes AICK-441

[ROCm/composable_kernel commit: 323e014799]
2025-12-19 10:04:48 +01:00
..
impl
[CK Grouped Gemm] Fix workspace stride in two stage kernel (#3412)
2025-12-19 10:04:48 +01:00
conv_tensor_rearrange_op.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
convolution_backward_data_specialization.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
convolution_backward_weight_specialization.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
convolution_forward_specialization.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_avgpool_bwd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_base.hpp
[ck][gfx12] support contraction on gfx12 (#3421)
2025-12-15 07:16:01 -08:00
device_batched_contraction_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm_e_permute.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm_gemm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm_multi_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm_multiple_d_gemm_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm_softmax_gemm_permute.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm_softmax_gemm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batched_gemm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batchnorm_backward.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batchnorm_forward.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_batchnorm_infer.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_cgemm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_contraction_multiple_abd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_contraction_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_conv_bwd_data.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_conv_fwd_bias_activation_add.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_conv_fwd_bias_activation.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_conv_fwd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_conv_tensor_rearrange.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_elementwise_normalization.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_elementwise_scale.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_elementwise.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_bias_e_permute.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_dequantB.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_multiple_abd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_multiple_d_ab_scale.hpp
Wmma support for gemm_ab_scale (#3314)
2025-12-11 09:06:20 +01:00
device_gemm_multiple_d_layernorm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_multiple_d_multiple_r.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_mx.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_reduce.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_splitk.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_streamk_v2.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_streamk.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm_v2.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_gemm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_contraction_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_conv_bwd_data_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_conv_bwd_weight_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_conv_bwd_weight.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_conv_fwd_multiple_abd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_conv_fwd_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_conv_fwd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm_fixed_nk.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm_multi_abd_fixed_nk.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm_multi_abd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm_softmax_gemm_permute.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm_splitk.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm_tile_loop.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_grouped_gemm.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_max_pool_bwd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_multiple_reduce.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_normalization_bwd_data.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_normalization_bwd_gamma_beta.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_normalization_fwd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_permute.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_pool_fwd.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_put_element.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_reduce_multi_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_reduce.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_softmax.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
device_splitk_contraction_multiple_d.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
gemm_specialization.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
helper.hpp
chore(copyright): update copyright header for include directory (#3224)
2025-11-18 10:17:18 -08:00
masking_specialization.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
matrix_padder.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
reduction_operator_mapping.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
tensor_layout.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
tensor_specialization.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
welford_helper.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00