Enrico Degregori 7414a0f4d4 Wmma support for gemm_reduce (#3145) ...

* Initial implementation GEMM+Reduce:

 - device struct
 - epilogue struct

* Fix tests, improve profiler and add initial instances

* Add instances

* Fix compilation error

* Address review comments

* Fix logging

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

2025-11-12 11:23:54 -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

Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)

2025-09-12 08:17:07 -07:00

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

Split the instances by architecture. (#1223)

2024-04-02 09:42:17 -07:00

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

Switch to C++20 standard for all CMake targets. (#2536)

2025-07-22 10:52:10 -07:00

block_to_ctile_map

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

ck_tile

[CK_TILE] Fix gemm_quant (#3186)

2025-11-11 08:23:57 -08: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

Add Grouped Conv Fwd Large Tensor kernel (#1432)

2024-08-06 10:06:10 +02:00

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

poc convert fnuz fp8 to non-native dtype similar to ocp (#2871)

2025-09-18 22:51:01 -07:00

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

Wmma support for multiple Ds based GEMMs (#2613)

2025-09-05 16:31:08 +02:00

gemm_b_scale

Support b_scale: (#2350)

2025-07-24 18:49:58 -07:00

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

fix copy-paste bug in get_matrix_b; re-enable all tests in multi_abd (#2939)

2025-09-26 22:55:18 -04:00

gemm_multiply_multiply_wp

Fix splitk preshuffle (#3137)

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

gemm_mx

MX GEMM - Add FP6 GEMM Test (#2488)

2025-07-11 15:32:12 -06:00

gemm_reduce

Wmma support for gemm_reduce (#3145)

2025-11-12 11:23:54 -08: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

WMMA support for GEMM reduce (#2823)

2025-09-12 21:36:43 +02:00

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

Split the instances by architecture. (#1223)

2024-04-02 09:42:17 -07:00

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

[CK_TILE] support group from cmdline (#1295)

2024-05-28 11:13:21 +08:00

quantization

Added wmma support for gemm quantization: (#2841)

2025-09-16 16:23:29 -07:00

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

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

scatter_gather

Fix build errors on windows (#2456)

2025-07-16 07:58:23 -07:00

smfmac_op

Merge from internal (#1857)

2025-02-07 15:05:05 -07:00

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

update copyright headers (#726)

2023-05-31 18:46:57 -05:00

transpose

Split the instances by architecture. (#1223)

2024-04-02 09:42:17 -07:00

wmma_op

Fix issue with multiple targets and remove smfmac tests from unsupported test targets (#1372)

2024-07-03 23:34:38 -07:00

wrapper

Extend XDL kernel to Support RDNA3/4 - Part 4 (#2724)

2025-09-12 08:17:07 -07:00

CMakeLists.txt

Fix splitk preshuffle (#3137)

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