Andriy Roshchenko ffa13455a2 MX FP GEMM - Test MX FP8 MFMA Instructions (#1902) ...

* Refactored `load_A_row_major` to follow scale mapping

* Refactored `load_A_col_major` to follow scale mapping

* Refactored `load_B_col_major` to follow scale mapping

* Verified non-scaled test

* Verified scaled tests

* Used ReferenceMXGemm for verification

* Updated license headers

2025-02-21 13:35:54 -07:00

..

amd_address_space.hpp

update copyright headers (#726)

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

amd_buffer_addressing.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

amd_ck_fp8.hpp

Merge from internal (#1857)

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

amd_gemm_dpp.hpp

Redesign the DPP8 GEMM kernel to use warp-wise component (#863)

2023-09-06 11:44:09 -05:00

amd_inline_asm.hpp

Implement the fp16xint4 scale weight only kernel for Ali (#1786)

2025-01-03 18:35:21 +08:00

amd_lds.hpp

Implement direct loads split-K GEMM kernel (#1137)

2024-02-07 01:08:34 +01:00

amd_smfmac.hpp

Added structural sparsity blockwise gemm (#1435)

2024-09-11 15:19:42 +02:00

amd_wave_read_first_lane.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

amd_wmma.hpp

Add bf16 and int8 wmma gemms for Navi3x and Navi4x. (#1671)

2024-11-18 14:07:04 -08:00

amd_xdlops.hpp

MX FP GEMM - Test MX FP8 MFMA Instructions (#1902)

2025-02-21 13:35:54 -07:00

array_multi_index.hpp

update copyright headers (#726)

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

array.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

blkgemmpipe_scheduler.hpp

[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)

2025-02-20 14:00:27 -08:00

c_style_pointer_cast.hpp

update copyright headers (#726)

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

common_header.hpp

FP8 enablement - add a pseudorandom number generator, add conversion methods (#708)

2023-06-19 11:20:35 -05:00

container_element_picker.hpp

update copyright headers (#726)

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

container_helper.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

data_type.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

debug.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

dynamic_buffer.hpp

fix a bug for int4 scale weight only kernel (#1820)

2025-01-19 11:18:18 +08:00

e8m0.hpp

Merge from internal (#1857)

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

enable_if.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

env.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

f8_utils.hpp

Remove reinterpret_cast uses that result in undefined behaviour. (#1445)

2024-08-07 11:49:02 -07:00

flush_icache.hpp

Universal gemm flush cache (#1251)

2024-04-25 15:07:14 -05:00

functional2.hpp

update copyright headers (#726)

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

functional3.hpp

update copyright headers (#726)

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

functional4.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

functional.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

generic_memory_space_atomic.hpp

update copyright headers (#726)

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

get_id.hpp

update copyright headers (#726)

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

get_shift.hpp

Batchnorm splitk single kernel (#771)

2023-07-06 10:58:55 -05:00

ignore.hpp

update copyright headers (#726)

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

inner_product_dpp8.hpp

Redesign the DPP8 GEMM kernel to use warp-wise component (#863)

2023-09-06 11:44:09 -05:00

inner_product.hpp

Fix the DL kernel issues on Navi3x. (#998)

2023-10-19 09:34:39 -07:00

integral_constant.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

is_detected.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

is_known_at_compile_time.hpp

Add optimized copy to ck wrapper (#1126)

2024-01-19 11:29:00 +01:00

loop_scheduler.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

magic_division.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

math_v2.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

math.hpp

Extend available elementwise operations with conv examples (#995)

2023-10-19 17:23:19 +02:00

multi_index.hpp

update copyright headers (#726)

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

mxf4_utils.hpp

Merge from internal (#1857)

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

mxf6_utils.hpp

Merge from internal (#1857)

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

mxf8_utils.hpp

Merge from internal (#1857)

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

mxfp_utils.hpp

Merge from internal (#1857)

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

number.hpp

update copyright headers (#726)

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

random_gen.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

reduction_common.hpp

Batchnorm splitk single kernel (#771)

2023-07-06 10:58:55 -05:00

reduction_enums.hpp

update copyright headers (#726)

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

reduction_functions_accumulate.hpp

update copyright headers (#726)

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

reduction_operator.hpp

Remove unsupported (fp8) type from Add memory operation. (#1521)

2024-09-20 09:40:45 +02:00

scaled_type_convert.hpp

Merge from internal (#1857)

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

sequence_helper.hpp

update copyright headers (#726)

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

sequence.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

span.hpp

update copyright headers (#726)

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

static_buffer.hpp

Jing's contribution: prototype of mixed precision gemm FP16/BF16xint4 GEMM (#1762)

2025-01-02 11:48:06 +08:00

statically_indexed_array_multi_index.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

statically_indexed_array.hpp

update copyright headers (#726)

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

synchronization.hpp

Merging the gfx12 code into public repo. (#1362)

2024-06-27 00:33:34 -07:00

thread_group.hpp

update copyright headers (#726)

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

transpose_vectors.hpp

[GEMM] Gemm universal device operation (#1154)

2024-04-13 21:03:18 -05:00

tuple_helper.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

tuple.hpp

Codegen hipRTC compilation (#1579)

2025-01-31 09:48:39 -08:00

type_convert.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

type.hpp

Rebase the PR #1520 to ROCm repo. (#1574)

2025-02-20 18:58:14 -08:00

workgroup_barrier.hpp

initial stream-k implementation with example (#699)

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

workgroup_synchronization.hpp

Batchnorm splitk single kernel (#771)

2023-07-06 10:58:55 -05:00