ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-21 13:29:20 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,324 Commits 751 Branches 38 Tags
f2edb06bb00a3da569217f05db5101992d3cd52a
Commit Graph

11 Commits

Author SHA1 Message Date
linqunAMD
feec597551 Fix a typo in intrin_wmma_bf16_16x16x16_bf16_w32 (#2727) 
__builtin_amdgcn_wmma_bf16_16x16x16_bf16_w32 is only available in gfx11.

[ROCm/composable_kernel commit: 00fd72b2d4]
 2025-09-03 08:07:09 +08:00
Anton Gorenko
db016cf6da DeviceGemm_Wmma_CShuffleV3 with BlockGemmPipelineVersion::v3 (#2096) 
* Prepare files for DeviceGemm_Wmma_CShuffleV3

* Implement main part of CShuffleV3 with block pipeline v3 for WMMA

* Remove unused functions and template params for A/B descriptors

* Support both gfx11 and gfx12

* Enable SplitK for gfx12 and disable for gfx11

* Added RowColRow layout for DeviceGemmV2 fp16

* Added more instances for Row, Col, Row data layout

* Added instances for DeviceGemm_Wmma_CShuffleV3, Col, Row, Row data layout

* Added instances for DeviceGemm_Wmma_CShuffleV3, Col, Col, Row data layout

* Added more instances for DeviceGemm_Wmma_CShuffleV3, Row, Row, Row data layout

* Fix formatting

* Add documentation

Based on 5585c3121e

* Enable gemm_universal profiling for gfx11/12

* Add WMMA intrinsics for F8/BF8

* Support F8/BF8 DeviceGemm_Wmma_CShuffleV3, add basic instances

* Add BF16 instances and tests

* Fix test_gemm_universal_wmma_fp8 by adding CK_USE_WMMA_FP8

---------

Co-authored-by: Anca Hamuraru <anca@streamhpc.com>

[ROCm/composable_kernel commit: edd92fc546]
 2025-04-28 10:14:21 +05:00
Illia Silin
ea702b3631 Add bf16 and int8 wmma gemms for Navi3x and Navi4x. (#1671) 
* add bf16 gemms for gfx11/gfx12

* reduce the input values in test_gemm

* add int8 wmma gemm instances for gfx11/gfx12

* add example gemm_wmma_int8

* fix bug in gemm_wmma_int8 test

* increase bf16 gemm test tolerance

* update the dates and clean-up commented-out instances

[ROCm/composable_kernel commit: 8aba2724cc]
 2024-11-18 14:07:04 -08:00
Illia Silin
123aae9e6e enable compilation for generic navi targets (#1645) 
[ROCm/composable_kernel commit: 75c5bfa364]
 2024-11-07 14:14:42 -08:00
Illia Silin
57ae3ae99f Merging the gfx12 code into public repo. (#1362) 
[ROCm/composable_kernel commit: 941d1f7ce0]
 2024-06-27 00:33:34 -07:00
Illia Silin
b63148b2b1 Add support for more Navi2x and Navi3x models. (#1152) 
* add support for navi2x and navi3x models

* fix syntax

* use common macro for different mi300 architectures

[ROCm/composable_kernel commit: 180f16f9ac]
 2024-02-02 11:35:26 -08:00
Illia Silin
b57fbee2f1 update copyright headers (#726) 
[ROCm/composable_kernel commit: b94fd0b227]
 2023-05-31 18:46:57 -05:00
Haocong WANG
459469f66a Fix arch limitation bug (#639) 
[ROCm/composable_kernel commit: ea028ac65a]
 2023-03-15 07:44:13 -07:00
Haocong WANG
9687ad0b61 [Navi3x] Multiple issue fix (#612) 
* Change gridwise gemm mD blockwise gemm to naive

* RRR Gemm fix

* Fix RCR gemm bug

* Isolate wmma instructions

* Update amd_inline_asm.hpp

* Update amd_wmma.hpp

* Update amd_wmma.hpp

* fix syntax and update Jenkinsfile

---------

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

[ROCm/composable_kernel commit: 087e310589]
 2023-03-10 17:04:28 -06:00
Haocong WANG
9d5e41b586 [Navi3x-LWPCK-545] Block-wise GEMM + Real GEMM_WMMA_FP16 (#541) 
* wmma_op + unit test

* add arch limitation to wmma test

* change arch limitation

* Refactor + Add all type unit test(int4 compile failed)

* Add f32_16x16x16_bf16 unit test

* tempsave

* tempsave

* tempsave

* runtime bug, cannot find symbol

* workaround for incorrect HIP warpSize return value

* debugging

* tempsave

* Correctness OK, waiting for optimization

* Tidy up + format

* temp save

* temp save, reproduce the v_bfi_b32 issue

* add inline asm for wmmaop test

* tidy up

* clean some debug purpose code

* discard some codes

* clang format

* clang format

* compiler issue fixed + increase tile size

[ROCm/composable_kernel commit: 919aeb1f52]
 2023-01-16 20:06:01 -06:00
Haocong WANG
3baad464d0 [Navi3x-LWPCK-449] wmma_op + unit test (#484) 
* wmma_op + unit test

* add arch limitation to wmma test

* change arch limitation

* Refactor + Add all type unit test(int4 compile failed)

* Add f32_16x16x16_bf16 unit test

* Remote int4 related

* delete deprecated test

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
Co-authored-by: Chao Liu <chao.liu2@amd.com>

[ROCm/composable_kernel commit: abf9cc6c5c]
 2022-12-02 11:41:13 -06:00