Andriy Roshchenko 79b0bfeb41 MX GEMM - Add FP8 GEMM Tests for Different Layouts (#2152) ...

* Add gemm_mx_fp8_bf8 example with row-major B

* Add more overloads of MX MFMA instructions

* Add MK_KN (RRR) tests

* Add KM_NK (CCR) tests

* Add more problem sizes to Large tests

* Add test_gemm_mx to the list of regression tests

2025-05-01 11:55:48 -06:00

..

batched_gemm

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

Split the instances by architecture. (#1223)

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

batched_gemm_softmax_gemm_permute

reduce test size to avoid timeout on specific silicon (#1966)

2025-03-11 09:15:26 -07:00

batchnorm

Resolve some data type issues and cmake policy. (#940)

2023-09-26 08:39:11 -07:00

block_swizzle_test

initial stream-k implementation with example (#699)

2023-07-26 14:18:15 -05:00

block_to_ctile_map

update copyright headers (#726)

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

ck_tile

Add the MI355 support for CK TILE GEMM (#2046)

2025-04-03 11:48:54 -07:00

contraction

Fix continous dim selection in contraction (#1336)

2024-06-18 10:26:49 +02:00

conv_tensor_rearrange

Fixing most of the cppcheck errors. (#1142)

2024-01-24 13:47:48 -08:00

conv_util

Add Grouped Conv Fwd Large Tensor kernel (#1432)

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

convnd_bwd_data

Split the instances by architecture. (#1223)

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

convnd_fwd

Split the instances by architecture. (#1223)

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

data_type

Temporarily disable MX FP4 device tests (#2112)

2025-04-22 16:08:48 -05:00

elementwise_normalization

Refactoring cmake files to build data types separately. (#932)

2023-09-20 22:15:56 -07:00

gemm

Refactoring cmake files to build data types separately. (#932)

2023-09-20 22:15:56 -07:00

gemm_add

Split the instances by architecture. (#1223)

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

gemm_layernorm

Split the instances by architecture. (#1223)

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

gemm_mx

MX GEMM - Add FP8 GEMM Tests for Different Layouts (#2152)

2025-05-01 11:55:48 -06:00

gemm_reduce

Split the instances by architecture. (#1223)

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

gemm_split_k

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_streamk

MI308 fix for streamk 1-Tile floating point exception (#2101)

2025-04-21 11:44:07 -07:00

grouped_convnd_bwd_data

Integrate universal gemm with conv bwd data and add SplitK (#1315)

2025-04-28 23:54:49 +02:00

grouped_convnd_bwd_weight

Add support for GKCYX grouped conv weight (#2023)

2025-04-02 23:59:49 +02:00

grouped_convnd_fwd

Add grouped conv fwd 3d GKCYX instances for f32, f16, bf16 (#2069)

2025-04-16 11:00:55 +02:00

grouped_gemm

Polished Grouped GEMM APIs and new BF16 instances (#1600)

2024-11-27 13:02:44 +01:00

magic_number_division

update copyright headers (#726)

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

mx_mfma_op

MX GEMM - New GEMM pipeline for MX data types (#2059)

2025-04-15 17:17:07 -06:00

normalization_bwd_data

Grouped GEMM Multiple D tile loop. (#1247)

2024-04-25 15:12:53 -05:00

normalization_bwd_gamma_beta

Grouped GEMM Multiple D tile loop. (#1247)

2024-04-25 15:12:53 -05:00

normalization_fwd

layernorm and groupnorm backward data (#1083)

2023-12-19 04:23:11 +08:00

permute_scale

Split the instances by architecture. (#1223)

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

pool

OCP FP8 support for gfx12. (#1710)

2024-12-03 08:42:55 -08:00

position_embedding

[CK_TILE] support group from cmdline (#1295)

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

reduce

Rewrite *sh reduce unit tests to gtest: part 1 (#1407)

2024-08-12 16:28:10 +02:00

reference_conv_fwd

update copyright headers (#726)

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

scatter_gather

[CK_TILE] add scatter_gather (#1609)

2024-10-29 18:19:29 +08:00

smfmac_op

Merge from internal (#1857)

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

softmax

[HotFix] add config and version files to pass on build info (#856)

2023-08-23 11:36:17 -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

Split the instances by architecture. (#1223)

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

CMakeLists.txt

MX GEMM - Add FP8 GEMM Tests for Different Layouts (#2152)

2025-05-01 11:55:48 -06:00