Enrico Degregori 507d81c3af Fix splitk preshuffle (#3137) ...

* Fix splitK multiply_multiply_wp

* Add tests for gemm_multiply_multiply_wp

* Add tests for gemm_universal_preshuffle (KBatch = 1)

* Add tests gemm_blockscale_wp

* Fix splitk gemm universal preshuffle

* Run new tests on arch supporting fp8

* Restore example

* Fix strides profiler

* Fix tests

* Fix clang format

* Finalize profiler preshuffle with tolerances

* Minor improvements to splitk related changes

* Address review comments: clang format and ckProfiler typo

* Remove b_k_split_offset from SplitKBatchOffset struct

2025-11-03 11:59:01 -08:00

..

batched_gemm

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

batched_gemm_b_scale

implement device batched gemm b scale for wmma (#2825)

2025-10-16 11:00:42 -07:00

batched_gemm_gemm

…

batched_gemm_multi_d

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

batched_gemm_reduce

…

batched_gemm_softmax_gemm

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

batched_gemm_softmax_gemm_permute

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

batchnorm

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

block_swizzle_test

…

block_to_ctile_map

…

ck_tile

Replace CK_TILE_PIPELINE macros with a common enum

2025-11-03 09:35:05 -07:00

contraction

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

conv_tensor_rearrange

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

conv_util

…

convnd_bwd_data

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

convnd_fwd

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

data_type

…

elementwise_normalization

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

gemm

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

gemm_add

…

gemm_b_scale

…

gemm_blockscale_wp

Fix splitk preshuffle (#3137)

2025-11-03 11:59:01 -08:00

gemm_layernorm

WMMA gemm_add_relu_add_layernorm (#2989)

2025-10-31 11:19:26 -07:00

gemm_multi_abd

…

gemm_multiply_multiply_wp

Fix splitk preshuffle (#3137)

2025-11-03 11:59:01 -08:00

gemm_mx

…

gemm_reduce

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

gemm_split_k

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

gemm_universal

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

gemm_universal_preshuffle

Fix splitk preshuffle (#3137)

2025-11-03 11:59:01 -08:00

gemm_universal_reduce

…

gemm_universal_streamk

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

grouped_convnd_bwd_data

Fixed handling of split-K autodeduce argument for grouped convolution (#3024)

2025-10-17 15:36:39 +03:00

grouped_convnd_bwd_weight

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

grouped_convnd_fwd

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

grouped_convnd_fwd_activation

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

grouped_gemm

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

magic_number_division

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

mx_mfma_op

Rearrange pointers to fix the reinterpret_cast issue (#3077)

2025-10-23 10:54:13 -07:00

normalization_bwd_data

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

normalization_bwd_gamma_beta

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

normalization_fwd

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

permute_scale

…

pool

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

position_embedding

…

quantization

…

reduce

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

reference_conv_fwd

…

scatter_gather

…

smfmac_op

…

softmax

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

2025-09-30 08:24:40 -07:00

space_filling_curve

…

transpose

…

wmma_op

…

wrapper

…

CMakeLists.txt

Fix splitk preshuffle (#3137)

2025-11-03 11:59:01 -08:00