Commit Graph

1961 Commits

Author SHA1 Message Date
Yanxing-Shi
a0f1615c09 Merge remote-tracking branch 'upstream/develop' into support_engine_benchmark 2025-05-21 09:48:23 +00:00
Yanxing-Shi
bb66c2af3e disable warning output & enable default config 2025-05-21 09:47:57 +00:00
Aviral Goel
990d645578 added gemm universal example in readme (#2216) 2025-05-20 15:35:07 -07:00
SamiAario-AMD
380bca2b85 Fix 11_add_rmsnorm2d_rdquant (#2207) 2025-05-20 15:15:28 -07:00
Thomas Ning
1386924749 Add the instances for small sized GEMM in preshuffle and improve CMake Flag (#2212)
* Add small instance, add the bug fix, & improve the example CMake

* clang format
2025-05-20 15:05:08 -07:00
Yanxing-Shi
1bd07d12fc Merge remote-tracking branch 'upstream/develop' into support_engine_benchmark 2025-05-20 16:09:19 +00:00
Yanxing-Shi
4dcbc7e3d8 fix jenkins & delete extra header 2025-05-20 16:08:30 +00:00
Yanxing-Shi
3506722e6a add benchmark gemm executable 2025-05-20 15:41:19 +00:00
Sami Remes
d1e6f0982d [CK_TILE] Grouped GEMM tile loop (#2146)
* Add trait to use a persistent kernel and split the entrypoints in grouped gemm

* Some helper functions for persistent kernel case

* Get max occupancy grid using device properties

* Implement tile loop in main entry point to grouped gemm

* Enable GridSize() on device

* Handle offset tile index using real current block index

* Add persistent kernel choice to grouped gemm example

* Use a for-loop for iterating over the group

* Reduce VGPR spills by early-exit

* Enable persistent kernel choice in grouped_gemm example

* Add persistent kernel option to grouped_gemm test

* Fix formatting with remod.py

* Remove GridUpdateBlocks as blocks are now iteratively computed

* Add comment about VGPR spilling

* Fix formatting

* Use CK_TILE_HOST instead of __host__

* Enable all Row/Col combinations in grouped gemm unit test

* Add some KBatch=2 cases to grouped gemm tests

* Fix SplitK for grouped gemm

* Enable pipeline hotloop/tailnumber selection in-kernel for grouped gemm

* Add type traits

* Split examples to regular and tileloop

* Formatting

* Use hipExtStreamGetCUMask to get current active CUs for the given stream

* Align test and example kernel config, and disable validation for splitk repeats

* Remove debug options from CMakeLists.txt

* Separate the code paths for persistent/non-persistent in test

* Fix formatting

* Address review comments

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
2025-05-20 17:18:57 +03:00
Yanxing-Shi
3e66716bd0 merge develop 2025-05-20 09:01:24 +00:00
Yanxing-Shi
ee6b7f9246 add kernel instance object 2025-05-20 08:57:48 +00:00
Aviral Goel
c4929225f6 remove debug statements from CMakeLists (#2204) 2025-05-19 17:31:04 -07:00
Jan Patrick Lehr
0970f22221 [CMake] Disable newly added compiler warning -Wnrvo (#2210)
Recently a new warning was added to Clang to warn when no copy-elision
on return happens. That prevents our CK build. This disables the
warning.
2025-05-19 17:30:15 -07:00
jefyang1
f18170064d Use new mfma instructions for FP8 on gfx950 (#2202)
* Add logic to use new mfma instructions for fp8 bf8

* Fix example_gemm_xdl_fp8_pk_i4_bpreshuffle_v3 on gfx950 and run clang format

* Update include/ck/tensor_operation/gpu/warp/xdlops_gemm.hpp

Co-authored-by: Andriy Roshchenko <107577548+andriy-ca@users.noreply.github.com>

* Fix intrin_mfma f8 calls due to merge mistake

---------

Co-authored-by: Andriy Roshchenko <107577548+andriy-ca@users.noreply.github.com>
2025-05-19 17:29:51 -07:00
Andriy Roshchenko
57e0f5df29 MX GEMM - Expand MX MFMA Testing to BF8, FP6, and BF6 Data Types (#2199)
* Unify test interface for different layouts.

* WIP: Introducing FP4/FP6/FP8 abstractions

* WIP: Introducing packed storage abstraction

* WIP: Introducing packed storage abstraction

* WIP: Improved support for FP6 data type

* Refactor packed storage for f6_t

* WIP: FP6 MFMA test

* Test if we correctly represent all FP6/FP4 numbers

* Additional output for failed FP4 test.

* More failing conversion tests

* Even more failing conversion tests

* Working FP6 MFMA tests

* Expand MX MFMA testing to BF8/6

* Update and verify MX MFMA test for packed types

* Fix fp4 and fp6 conversions on host

* Working MX MFMA tests for FP8/6/4

* Cleanup

* Add missing type

* Cleanup

* Final cleanup

* Restrict FP6/4 values output to CK_LOGGING=1

* Use CHAR_BIT instead of number 8

* Fix typo

* Remove FP6 and FP4 from the list of native types

---------

Co-authored-by: Rostyslav Geyyer <rosty.geyyer@amd.com>
2025-05-19 16:52:51 -05:00
jefyang1
b8b12bb81e Fix example_grouped_gemm_multiple_d_xdl_fp16 on gfx950 (#2203)
* Fix example_grouped_gemm_multiple_d_xdl_fp16 on gfx950

* Run clang format
2025-05-19 14:25:50 -07:00
Yanxing-Shi
5b83f76eb0 fix codegen bug 2025-05-19 14:03:16 +00:00
Yanxing-Shi
b3caa67694 fix csv bug 2025-05-19 11:44:26 +00:00
Yanxing-Shi
9897410acf refactor profiler 2025-05-19 10:42:57 +00:00
Bartłomiej Kocot
6342f6b5e8 Restore oddc instances (#2201) 2025-05-16 18:42:02 -07:00
arai713
5b3430b868 Narrowing error fix for codegen compilation (#2194)
* removed comment with special characters

* fix for arg/template change after merge from develop

---------

Co-authored-by: Thomas Ning <Thomas.Ning@amd.com>
2025-05-16 11:11:54 -07:00
Illia Silin
40668c9a99 Build and store CK library deb package for all targets daily. (#2196)
* generate and store library package for all targets

* use ninja to build packages for all targets

* make sure to use ftime-trace when using ninja

* make sure build trace only runs on gfx9

* archive lib package and stash only library package
2025-05-16 07:40:53 -07:00
Yanxing-Shi
c821b1253a format python 2025-05-16 10:41:30 +00:00
Yanxing-Shi
adeb051095 Merge remote-tracking branch 'upstream/develop' into support_engine_benchmark 2025-05-16 10:38:59 +00:00
Yanxing-Shi
012c77125a recover benchmark_gemm and fix 2025-05-16 10:37:59 +00:00
Mateusz Ozga
fa3c6811d8 Disable conv for Filter1x1Stride1Pad0 when K or C is even (#2186) 2025-05-16 10:18:47 +02:00
Po Yen Chen
791802b381 [CK_TILE] fMHA batch_prefill block index & logits soft-capping optimizations (#2198)
* Write soft-sign in inline asm

* Change tile idx computation

* Add macro to turn off soft-sign asm opt

* Use simple for loop to avoid register spill

* Only do block id transform for masking cases
2025-05-16 15:14:46 +08:00
Yanxing-Shi
fb63cd5923 merge 2025-05-16 06:55:32 +00:00
Po Yen Chen
8cb0474b3d Use only qr_async pipeline for batch_prefill (#2195) 2025-05-15 11:47:29 -07:00
Khushbu Agarwal
3d8d6e75e4 Adding validation for tile sizes in Tile Engine (#2189)
* Adding validation for tile sizes

* Add architecture in config, and shuffle lines of code in warp_gemm.hpp

* Enable MFMA for gfx950, and invalid tile handling
2025-05-15 10:28:31 -07:00
Casey-Shi
fde83689d5 Merge branch 'develop' into support_engine_benchmark 2025-05-15 22:21:58 +08:00
Bartłomiej Kocot
7c0e29cc0f Extend 64x64 with 4 waves instances for grouped conv bwd wei (#2187)
* Extend 64x64 with 4 waves instnaces for grouped conv bwd wei

* Fix

* fix

* fix
2025-05-15 16:21:34 +02:00
Yanxing-Shi
68a4aff0b1 fix config 2025-05-15 14:20:17 +00:00
Yanxing-Shi
d6f31b680a modify changelog 2025-05-15 14:12:45 +00:00
Yanxing-Shi
d4107f55cf remove pydantic module 2025-05-15 13:54:26 +00:00
Yanxing-Shi
fc092038f7 fix README 2025-05-15 12:37:00 +00:00
Yanxing-Shi
ccf18b90e6 add asm cache control 2025-05-15 12:20:28 +00:00
Yanxing-Shi
457315dd8a Merge remote-tracking branch 'upstream/support_engine_benchmark' into support_engine_benchmark 2025-05-15 11:18:11 +00:00
Yanxing-Shi
047f6e4480 python format 2025-05-15 11:16:13 +00:00
Yanxing-Shi
62d2a63f43 add benchmark for cold and warmp up 2025-05-15 11:11:18 +00:00
Yanxing-Shi
cfbbae9bd6 fix 2025-05-15 06:15:38 +00:00
illsilin
3d58544b7d add pydantic module to the docker image 2025-05-14 09:59:55 -07:00
Yanxing-Shi
53c4429f37 format 2025-05-14 15:46:50 +00:00
Yanxing-Shi
2843f8e59d Merge remote-tracking branch 'upstream/develop' into support_engine_benchmark 2025-05-14 15:40:26 +00:00
Yanxing-Shi
16654510da fix 2025-05-14 15:12:27 +00:00
Yanxing-Shi
7fa1d4daea add changelog 2025-05-14 13:59:25 +00:00
Yanxing-Shi
4bbe7eca09 add cmake option & modify 2025-05-14 09:17:37 +00:00
BingYuan.Zhou
41c17d0a95 fix moe sorting build fail (#2190)
* fix moe sorting build fail

* refile code

---------

Co-authored-by: solin <bingzhou@amd.com>
2025-05-14 09:31:26 +08:00
Illia Silin
58f9e9ffbc Update the buffer load/store intrinsic names for clang>=20. (#2192)
* fix the buffer load/store intrinsic names

* fix clang format
2025-05-13 10:18:14 -07:00
Yanxing-Shi
58ab4eb617 remove comment 2025-05-13 16:22:22 +00:00