Thomas Ning
e5fb690945
Fix the Composable Kernel CI and versions incompatibility ( #4640 )
...
## Motivation
This PR has 4 patches:
1. Fix the CI error of grouped gemm.
2. Fix the incompatibility of old linux version.
3. Fix the potential errors of flatmm.
4. Address the previous comments of abquant eight warps pipeline
solution.
---------
Co-authored-by: illsilin_amdeng <Illia.Silin@amd.com >
2026-02-18 06:59:37 -08:00
Max Podkorytov
6cf89bbca9
[CK-Tile] move out memory operation from cshuffle epilogue class ( #3359 )
...
* initial poc
* factor out common parts in operator()
* cv4
* rest of the universal gemm pipelines
* fix test
* remove boilerplate from tile engine
* fix example
* fix example
* format
* fix tests build for gemm
* remove base pipeline codegen from gemm instance builder
* unify v3 logic with the rest of universal gemm pipelines
* fix build for multi abd test
* fix test gemm multi d
* fix build for weight preshuffle
* fix grouped gemm test
* fix grouped gemm multi d test
* fix grouped gemm preshuffle
* fix grouped gemm example except for quant
* fix gemm preshuffle
* fix splitk 2 stage example
* fix batched gemm example
* fix multid example
* fix multiabd example
* fix batched gemm test
* fixup
* fix examples build
* fix grouped gemm test build
* fix smoke builder
* hacky poc
* fix tile engine
* kill the lambda
* maybe fix test build
* more fixes
* clang-format
* save temp
* clang-format
* mostly fix examples
* clang-format
* remove dead code
* more cleanup
* fix fmha bwd build (default epilogue set/add appears to be broken)
* fix default epilogue tests but not correctness
* clang-format
* fix bquant
* clang-format
* cleanup dead code
* rearrange make windows for readability
* restore changes to IsSupportedArgument
* fix smoke-builder
* clang-format
* fixup rename class
* build fixes
* clang-format
* fix builder
* fixup
* remove set from builder tests
* fix test
* clang-format
* re-refactor the kernels
* clang-format
* fix header license
* remove memory operation from conv bwd test
* clang-format
* clang-format example,include
* clang-format test
* build fixes
* clang-format
* solve compilation error
* fix the CI
* solve compilation error
* clang format
* solve merge conflict
* solve merge conflict
* solve the gfx11 error
* solve test error
* moar build fixes
* remove AtomicAddRequiresKBatchGreaterThanOne test since the property is removed from the kernel scope
---------
Co-authored-by: Thomas Ning <Thomas.Ning@amd.com >
[ROCm/composable_kernel commit: e339101e9c ]
2026-01-04 03:28:14 -08:00
yadaish
fc3ffa0d75
[CK_TILE] support split-k a16w4 gemm1 ( #3389 )
...
* initial version to support moe gemm1 split-k
* add missing args
* fix build warning
* update reference
* for split-k disable bias and weight
* remove debug log
* fix format
* fix div by zero errors
* fix cmake config
* update
* resolve conflicts
* remove useless changes
* reformat
* fix
* remove useless changes
* fix ci
---------
Co-authored-by: lalala-sh <Jiaxing.Wen@amd.com >
Co-authored-by: root <root@smci355-ccs-aus-m01-25.cs-aus.dcgpu >
[ROCm/composable_kernel commit: dae85ead64 ]
2025-12-29 23:05:35 +08:00
yadaish
e76ee195df
Dev/a8w4 and a8w8splitk ( #3447 )
...
* Ck moe bs splitk pr (#3440 )
* splitk kick-off. Compilation fail
* splitk hack pass
* fix scale offset calc.
* clang-format for a8w8_moe_blk_gemm1 splitk change
* fix testcase error
---------
Co-authored-by: oscar <huaiguxu@amd.com >
Co-authored-by: huaiguxu <145733371+huaiguxu@users.noreply.github.com >
* Zan/moe a8w4 (#3441 )
* update
* update
* update ck moe a8w4
* update
* update
* update
* compile pass
* update
* update
* python3 op_tests/test_moe_2stage.py -t 16 -e 1 -k 1 -dim 256,256 ready
* support new a8w4 kernel
* update
* update ck_tile
* re format
* update
* update
* fix conflict
* fix build
* update ck_tile moe
* fix clang format
* fix the problem
* fix accruacy issue
* fix
---------
Co-authored-by: oscar <huaiguxu@amd.com >
Co-authored-by: huaiguxu <145733371+huaiguxu@users.noreply.github.com >
Co-authored-by: Zzz9990 <zanzhang@amd.com >
Co-authored-by: felix <felix.li@amd.com >
[ROCm/composable_kernel commit: c0ee71d735 ]
2025-12-19 09:26:52 +08:00
Yi DING
fb72fea980
[CK_TILE] Add FP8xF4 Flatmm ( #3401 )
...
* Refactor policy
* fix a bank conflict
* Enable mixed mx flatmm
* Update
[ROCm/composable_kernel commit: 57e1e4a848 ]
2025-12-17 10:01:48 +08:00
Zzz9990
09e81b46ba
[CK_TILE MOE] add NT & preshuffle permute to cktile MOE ( #3377 )
...
* update coherence
---------
Co-authored-by: Zzz9990 <Zzz9990>
[ROCm/composable_kernel commit: 1aa93ef551 ]
2025-12-10 10:03:28 +08:00
lalala-sh
9691ccf03c
fix a16w4 moe bugs ( #3373 )
...
* fix valid mask bug
* update format
[ROCm/composable_kernel commit: 6f0966e1e9 ]
2025-12-09 17:54:55 +08:00
Yi DING
8b98fe0353
[CK_TILE] Optimize Flatmm MXFP4 by Eliminating Runtime Division by 2 ( #3287 )
...
* [CK_TILE] Optimize Flatmm MXFP4 by Eliminating Runtime Division by 2
* typo
[ROCm/composable_kernel commit: 878b4e7f46 ]
2025-12-08 19:20:44 +08:00
msaffari-amd
1055485a38
Add validity checks for MoE FlatMM scatter and enable bf16 hardware atomic-add ( #3236 )
...
* Add validity checks for MoE FlatMM scatter and enable bf16 hardware atomic
* correct clang-format
* removed unused rtol_atol variable from example code
* clang format correction
* remove unused varable max_accumulated_value from example
[ROCm/composable_kernel commit: f875ab0bbc ]
2025-11-28 09:43:01 +01:00
Aviral Goel
ee7a68b10f
chore(copyright): update copyright header for include directory ( #3293 )
...
[ROCm/composable_kernel commit: de6466481f ]
2025-11-26 11:00:05 -07:00
Yi DING
e27e760d5a
[CK_TILE] Add Flatmm MX FP8 ( #3208 )
...
* Use async for flatmm mxfp4
* Fix preshuffle
* Add flatmm mxfp8
* Thanks, Copilot
* Thanks Copilot again~
[ROCm/composable_kernel commit: 47e2ed838e ]
2025-11-20 10:35:15 +08:00
Max Podkorytov
b1faa0c1c5
[CK-Tile] Remove usage of tile partitioner's full gemm shape ( #3204 )
...
gemm shape should be used from the pipeline instead (where it gets from a problem description struct)
[ROCm/composable_kernel commit: a3a4eb12bd ]
2025-11-18 09:56:40 -08:00
BingYuan.Zhou
3800080d25
fix build error ( #3195 )
...
Co-authored-by: root <root@hjbog-srdc-39.amd.com >
[ROCm/composable_kernel commit: 4d629cd2b0 ]
2025-11-14 09:46:13 +08:00
Yi DING
acec30dd09
[CK_TILE] Add mxfp4 flatmm ( #3080 )
...
* Squashed commit of the following:
commit 3e1a851dad834776efbe4fe365ac82c4ed312010
Author: Ding, Yi <yi.ding@amd.com >
Date: Thu Oct 23 06:10:54 2025 +0000
Fix & clean after rebase
commit 1edf485092f44411da9a1796a4a6b72d5cdb67c6
Author: Ding, Yi <yi.ding@amd.com >
Date: Wed Oct 22 10:46:13 2025 +0000
Squashed commit of the following:
commit 5276b28a51dac7b5d2106fbae8e78de190ee0de1
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 22 02:04:27 2025 -0500
fix bandwidth calculation
commit d645bb20c6d879154c30ecd82bbff4d2a9206750
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 22 00:58:59 2025 -0500
updates
commit 0fa7e6b88aaf81a36034aa7607746de295de4263
Author: mtgu0705 <mtgu@amd.com >
Date: Fri Sep 19 00:39:46 2025 -0500
fix a bug, set the A DS_read preload size to 4 for MXFP4
commit 50cafa824e2267f2b2f0dfeeb93e69a673630c61
Author: mtgu0705 <mtgu@amd.com >
Date: Thu Sep 18 01:19:03 2025 -0500
fix a_wrap preload issue for large MPerBlock.
commit e6333bbbc6ef540e24f92095040085f1ed59041e
Author: mtgu0705 <mtgu@amd.com >
Date: Wed Sep 17 21:34:03 2025 -0500
optimized the VGPR repack issue for MXFP4
commit e99e4932c401b9f6d1893dd5044c2827d6b3f145
Author: Gino Lu <gino.lu@amd.com >
Date: Wed Sep 17 04:19:44 2025 -0500
fix time error
commit 4586ce6da7fba0514f2e01a8124c76b7d494e124
Author: mtgu0705 <mtgu@amd.com >
Date: Wed Sep 17 03:58:00 2025 -0500
updated, function passed.
commit c4f25e7579573db5681b9160f6bdb1349f3566f1
Author: mtgu0705 <mtgu@amd.com >
Date: Tue Sep 16 22:21:39 2025 -0500
fix, function partially passed
commit a51b56eb6b00b99a4e8d2802dbf5b5b5277b54d8
Author: mtgu0705 <mtgu@amd.com >
Date: Tue Sep 16 03:01:12 2025 -0500
fix, reference function passed, next check kernel function
commit 5b02643ebab18960e8f9ba66c6bd2f91774f9cae
Author: Gino Lu <gino.lu@amd.com >
Date: Tue Sep 16 02:29:01 2025 -0500
let pack/unpack return pk_fp4_t
commit 76d37c5d4b17530e95c6fced31bff66a35d54b8f
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 15 20:50:26 2025 -0500
fix
commit e5be3e162b9a20e5355bd556d2b27afb6d8bf085
Author: Gino Lu <gino.lu@amd.com >
Date: Mon Sep 15 05:51:06 2025 -0500
fix bug
commit 39a024efe4aa773df589712b1290803bb5ab5d1d
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 15 04:02:05 2025 -0500
fix core dump issue, function is not correct.
commit 16c49d268cfe065b5112b960b2d852b26552686a
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 15 03:03:02 2025 -0500
updates, build pass
commit fe7a961852dee6eff3be3cf1e0d0fabec5cd42ee
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 15 00:05:18 2025 -0500
updates
commit aaf9fe8022a72df59e04e4d5886dca3ba9c23400
Author: Gino Lu <gino.lu@amd.com >
Date: Sun Sep 14 23:40:28 2025 -0500
fix bug
commit a3da89290e1553b85fbf1171c07e93ac0f5584db
Author: Gino Lu <gino.lu@amd.com >
Date: Fri Sep 12 03:28:50 2025 -0500
fix interface
commit c5ff747e72d877461ba61dc19a0fe15527d3161e
Author: Gino Lu <gino.lu@amd.com >
Date: Fri Sep 12 02:53:50 2025 -0500
add interface in warp_gemm_impl
commit 0a48d369e601cc798589fc59e0784bdbfc0a22f9
Author: mtgu0705 <mtgu@amd.com >
Date: Wed Sep 10 05:03:08 2025 -0500
updates some fixes.
commit aaa2beca30ff5546d171a2028d1894fd4e131d4e
Author: mtgu0705 <mtgu@amd.com >
Date: Tue Sep 9 04:37:42 2025 -0500
fix after merge ginolu/add_wgmfma_dispatcher
commit bf87449b09cba690922b2f3f78ba39bf1b1e472e
Merge: 05ab58e3d 991d7fdbb
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 8 22:09:15 2025 -0500
Merge remote-tracking branch 'origin/ginolu/add_wgmfma_dispatcher' into mtgu/cktile_mxfp4_flatmm_dev
commit 05ab58e3de2b708aceda63d704089c0fa89437ae
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 8 21:42:47 2025 -0500
update mx flatmm tail pipeline
commit 991d7fdbb726d65091a91b5cc2800f798a6661fc
Merge: ad046084a b2f280046
Author: Gino Lu <gino.lu@amd.com >
Date: Mon Sep 8 19:10:23 2025 -0500
Merge branch 'develop' into ginolu/add_wgmfma_dispatcher
commit ad046084a2f6e4ebf0cd8b47d0d72b74815061fa
Author: Gino Lu <gino.lu@amd.com >
Date: Mon Sep 8 19:09:55 2025 -0500
fix type error
commit 42e16b43a035364a42789d7ce45a1e6a7d1d2609
Author: mtgu0705 <mtgu@amd.com >
Date: Mon Sep 8 04:01:40 2025 -0500
update hotloop pipeline
commit c2f69745346545087c8ce24acaba2961bb93ef0b
Merge: adbeeb90b 8b4be3a0e
Author: Gino Lu <gino.lu@amd.com >
Date: Fri Sep 5 04:22:26 2025 -0500
Merge branch 'develop' into ginolu/add_wgmfma_dispatcher
commit adbeeb90be1533f8aeb8c1d5aea6470d45a455a0
Author: Gino Lu <gino.lu@amd.com >
Date: Fri Sep 5 04:21:26 2025 -0500
fix clang format
commit e2378ac393bb79ac80a8eef84677bffce86d9e0a
Author: mtgu0705 <mtgu@amd.com >
Date: Wed Sep 3 10:00:54 2025 -0500
some updates
commit bdc18a2269db49ff88e1ef1db30f83ea430d7544
Merge: 6c5cea2b7 feec59755
Author: asleepzzz <hanwen.chang@amd.com >
Date: Wed Sep 3 13:22:03 2025 +0800
Merge branch 'develop' into ginolu/add_wgmfma_dispatcher
commit 6c5cea2b7a306f5d0ad346cb9baf6370ea2a73fe
Author: Gino Lu <gino.lu@amd.com >
Date: Mon Sep 1 02:11:02 2025 -0500
fix vec size error
commit 76d1dfa352087dfd5867c8909b73726d3a1e853e
Author: Gino Lu <gino.lu@amd.com >
Date: Mon Sep 1 01:23:39 2025 -0500
fix format error
commit a9061aaa1b4bfaa9db102c75b9d74863f39708a9
Author: mtgu0705 <mtgu@amd.com >
Date: Sat Aug 30 03:19:07 2025 -0500
update codes
commit 0caa184a271a8824ef40f87de456d0fa2500c8ad
Author: mtgu0705 <mtgu@amd.com >
Date: Fri Aug 29 11:27:33 2025 -0500
init ck_tile mxfp4 flatmm
commit 5d46a6635f04bd69b76f7eda1438862e271b987a
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Thu Aug 28 08:02:50 2025 +0000
Add bias for f16xf4 moe_flatmm
commit dd112dc302d17f541737671a3ac557d7c09ff969
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 27 13:39:47 2025 +0000
update case construction
commit b1aca68a073d82c7b3c7bb53286e5f415999edc1
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Tue Aug 26 12:32:29 2025 +0000
support swiglu activaion and use rcpf to accelerate silu
commit 49235bd42349a84fc2ebd7ad0b100cc2545bb80a
Author: Gino Lu <gino.lu@amd.com >
Date: Tue Aug 26 02:33:55 2025 -0500
first commit
commit c169e39d6381b932cf7098cc118db29df91da1cb
Author: root <root@smci355-ccs-aus-m02-25.cs-aus.dcgpu >
Date: Fri Aug 22 04:01:59 2025 -0500
add line to last
commit 318f9bf317306454941bbf394c1940023edcf0ac
Author: root <root@smci355-ccs-aus-m02-25.cs-aus.dcgpu >
Date: Fri Aug 22 03:20:46 2025 -0500
adjust A_LDS descriptor to avoid bankconflict
commit 9d066120ed068d6d102da25d619e170a28a04d18
Author: root <root@smci355-ccs-aus-m02-25.cs-aus.dcgpu >
Date: Thu Aug 21 09:46:52 2025 -0500
enable hotloop
commit 61a895e6b821798970afffd0e9432a21e2f04df8
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Thu Aug 21 09:12:21 2025 +0000
support atomic_pk_add_bf16 on gfx950
commit 9f14864e45f21d8c1bc70a94988fb86c2c0017d8
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Thu Aug 21 06:58:55 2025 +0000
use int64_t as expert stride to avoid overflow
commit e63af46b32e1139a1e59dee6f46b9971047c4026
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 20 13:53:32 2025 +0000
use v4i32 as the storage type for B to avoid repack operation
commit 6cf0224dd8a229bf2be726ca861c736c9b5f5415
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 20 06:40:03 2025 +0000
add pk_fp4_t and e8m0_t support for amd_buffer_load_impl
commit 67a591f2240b0b035029edad904627f98b3839fd
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 20 04:39:14 2025 +0000
optimize cvt_pkf4_to_f16 implementation
commit 51c7126e77e9b17af694eaa57040e487f9d443e8
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Tue Aug 19 14:56:46 2025 +0000
optimize A_LDS descriptor to avoid bankconflict
commit c113160f326353290a2878d7b8febf7daed91d71
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 18 18:43:37 2025 +0000
fix gate-up when GU_NRepeat > 1
commit a45ca0e9934ca4bb9114f65621d5c9582d937a45
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 18 17:28:11 2025 +0000
add fp16xf4 moe
commit dc8c8e484804f7bca10c8f0764540af3b5884e83
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Sun Aug 17 17:51:18 2025 +0000
rename example
commit b177c967141cfdc401d3f36bf17830fe99893600
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Fri Aug 15 06:20:46 2025 +0000
remove additional check when e8m0->float
commit d467f9688c3d35f391e15089135edb1ad1d38b05
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Thu Aug 14 09:34:12 2025 +0000
eliminate repeat dequant
commit 1b20674b26ab3ce6bd2f710dd729fd4cc0f79428
Merge: faa3c0278 7d02625e7
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 13 16:51:49 2025 +0000
Merge remote-tracking branch 'origin/moe_flatmm' into feat-mixed_input_flatmm
commit faa3c0278cf11b7105a4302dea3a4416520b2cc7
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 13 16:16:48 2025 +0000
update f16xMXF4
commit a2a2e1dab05501cc2136133236c01c08d51db4ea
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 13 10:48:53 2025 +0000
update scale-preshuffle for MXF4
commit eac9667feb899419dda1628164c092b969852660
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 11 11:24:34 2025 +0000
update
commit 7d02625e7678882af653f52c2a4ddaf64568a41c
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 11 08:38:23 2025 +0000
optimize gemm2 atomic_add pattern
commit d5f3c3e3ec72d0e6739467c4dc0b4e209f6d1192
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 11 07:59:47 2025 +0000
update scale for mxfp4
commit 15db198084614466bd4cfd4943fcb549cab2069a
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 11 07:56:14 2025 +0000
update case construction
commit 5dff349d82a5f70b6eea821d2622df51f90ef200
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 11 06:03:06 2025 +0000
update granularity control
commit d32cdc52144f65ec473f4ec8e45ea23968811184
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 11 03:42:46 2025 +0000
fix TileConfig
commit 26f38c5716304ee5f84e5c4f6f88144d9f3dddaf
Author: Gino Lu <gino.lu@amd.com >
Date: Thu Aug 7 21:37:28 2025 +0800
Add e8m0 scaled convert into CK_TILE (#2617 )
* first commit
* remove redundent code
* modify according to comments.
* fix type_convert error with scaled_type_convert
commit 419041478745f65dfec18859e75a13d975089519
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Fri Aug 8 20:19:16 2025 +0000
add mixed_prec fp16xfp4
commit 92e2a8b0308b9b107df9d2fd63a961efce706402
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Thu Aug 7 09:22:04 2025 +0000
debug mixed_prec flatmm
commit dea3ce80496ebcb00512979f0c3bb897f25e11a5
Merge: fde443bc3 b4f45fe14
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Wed Aug 6 16:49:47 2025 +0800
Merge pull request #2626 from ROCm/felix/flatmm_fix_splitk
fix split k
commit d480e8150358cc4ef8b05e25afe299141fad4fde
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Aug 6 08:33:33 2025 +0000
add moe_flatmm
commit b4f45fe14d11569f34de40c8a205cd6760b61357
Author: coderfeli <coderfeli@163.com >
Date: Wed Aug 6 02:45:31 2025 +0000
fix split k
commit fde443bc38fe60e52195817ecb2c7b20d772eedb
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Aug 4 07:16:36 2025 +0000
fix flatmm with scaling when WarpTileM == 32
commit 5a0667afa889a5af8c6b8509232eabd50cf5efef
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Fri Aug 1 11:01:23 2025 +0000
optimize scaling epilogue
commit 5c3502bbf71833c6f6f7d4a1cc4f4fd93811f522
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Fri Aug 1 07:28:38 2025 +0000
fix wrong config for fp8 scaling
commit eb2d0653cdb86603cb11539cbac466b6431b58b7
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Jul 30 06:20:30 2025 +0000
prune debug message
commit 0c089cb56343a39e02a1ee38e9cabeb71ba35e92
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Jul 30 04:52:08 2025 +0000
fix compile error
commit 61759ca30ce3787f70e228c3919b3e4d354016dd
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Tue Jul 29 15:42:58 2025 +0000
Add persistent option on flatmm for tuning
commit b36dc5dd55f15fc1ce8eb21637bdec862e56a883
Author: AMD-dteng <dteng@amd.com >
Date: Tue Jul 29 22:48:00 2025 +0800
update pipeline v1: add atomic IGLP schedule
commit f886f26994454fc2b4fc3433c86bf699767a2a7c
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Thu Jul 24 09:09:27 2025 +0000
fix error log throwing
commit 4b4686ab144daa9061fbda17f3df4c17600c8e9a
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Mon Jul 28 08:24:51 2025 +0000
crz idea
commit 7099af44a81be41431ba70ae60827b60116d02d2
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Sun Jul 27 11:57:38 2025 +0000
Add permuteN optimzization when NRepeat % 2 == 0 on flatmm
commit b147524c92e69a267337c8e48b6e64bcb1483551
Author: sjfeng <j514681085@icloud.com >
Date: Sun Jul 27 17:24:08 2025 +0800
try to remove c_shuffle_lds
commit 2dd94f59d1a7740a5689e1713ed45588cd0d55dd
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Fri Jul 25 07:41:48 2025 +0000
fix loop-dim mismatch and improve c_shuffle alu parallelism
commit 4e93f0c5e27806adc070e4caa81661069295751c
Merge: 3f12ef5aa 0eb7455f1
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Thu Jul 24 08:46:51 2025 +0000
merge flatmm -scale
commit 3f12ef5aa52ced1bff3bfb57b878358330e9e095
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Thu Jul 24 16:19:58 2025 +0800
revert delete of inc file
commit 08c3a0d184d7581dc5be364f5b36f16fb4a8d6fa
Author: solin <bingzhou@amd.com >
Date: Thu Jul 24 04:38:16 2025 +0000
reorg flatmm code
commit 0eb7455f106604d5254ed16b0daeda68e2a148e3
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Jul 23 19:12:31 2025 +0000
fix flatmm syntax error on gfx950
commit 695ff87e68fdcbe28452c1805cd4dbb643c45495
Author: Feng Shijie <Shijie.Feng@amd.com >
Date: Wed Jul 23 19:04:22 2025 +0000
support flatmm scaling
commit e3c29d9dea8758db96b998982ccc8bd1c4e8298d
Author: valarLip <340077269@qq.com >
Date: Wed Jul 23 08:44:12 2025 +0000
merge flatmm pipe v0 from dteng_flatmm_opt
commit 425c366fa4c30426ff36cade89b39fd8cb7b9732
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Wed Jul 23 15:38:12 2025 +0800
build pass
commit 6b377a9481535696de40f175d7e2159263d21bdc
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Wed Jul 23 07:20:26 2025 +0000
fix bug
commit b6dc58d1ea676fe480c0243ae098c875498f6d6a
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Wed Jul 23 15:01:53 2025 +0800
sync
commit 904359f401866ee810484e6b8f5b46d79d9e25c8
Author: valarLip <340077269@qq.com >
Date: Tue Jul 22 08:09:35 2025 +0000
adaptive scheduler instead of Macro definition
commit f29916c17228c17de9923aab62e7d72d7a30f4e9
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Thu Jul 17 08:40:35 2025 +0000
fix tail handler bug
commit e2c60a90929fec955d91db909d50db538d58363b
Author: lalala-sh <Jiaxing.Wen@amd.com >
Date: Wed Jul 16 10:12:19 2025 +0000
merge from dteng_flatmm_opt
---------
Co-authored-by: lalala-sh <Jiaxing.Wen@amd.com >
Co-authored-by: AMD-dteng <dteng@amd.com >
Co-authored-by: solin <bingzhou@amd.com >
Co-authored-by: sjfeng <j514681085@icloud.com >
Co-authored-by: valarLip <340077269@qq.com >
Co-authored-by: asleepzzz <hanwen.chang@amd.com >
Co-authored-by: Feng Shijie <Shijie.Feng@amd.com >
Co-authored-by: coderfeli <coderfeli@163.com >
Co-authored-by: Gino Lu <gino.lu@amd.com >
Co-authored-by: mtgu0705 <mtgu@amd.com >
* Fix crash on small M
* Apply suggestion from @Copilot
---------
Co-authored-by: lalala-sh <Jiaxing.Wen@amd.com >
Co-authored-by: AMD-dteng <dteng@amd.com >
Co-authored-by: solin <bingzhou@amd.com >
Co-authored-by: sjfeng <j514681085@icloud.com >
Co-authored-by: valarLip <340077269@qq.com >
Co-authored-by: asleepzzz <hanwen.chang@amd.com >
Co-authored-by: Feng Shijie <Shijie.Feng@amd.com >
Co-authored-by: coderfeli <coderfeli@163.com >
Co-authored-by: Gino Lu <gino.lu@amd.com >
Co-authored-by: mtgu0705 <mtgu@amd.com >
[ROCm/composable_kernel commit: e135dd518d ]
2025-10-31 11:29:05 +08:00
lalala-sh
63e0a73bd3
[CK_TILE] Update flatmm related kernels ( #3022 )
...
---------
Co-authored-by: Ding, Yi <yi.ding@amd.com >
Co-authored-by: felix <felix.li@amd.com >
[ROCm/composable_kernel commit: 211d64e18a ]
2025-10-22 22:36:11 +08:00
Khushbu Agarwal
bb5eeef2af
Fix for Add the API to load SGPR ( #2913 )
...
* Revert "Revert "[CK-Tile] Add the API to load SGPR (#2878 )" (#2904 )"
This reverts commit 5cc40c160f .
* Fix: sgpr minor issue
* cyclic dependency resolved
* clang formatted
* removing unused variable
* clang formatted
---------
Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com >
[ROCm/composable_kernel commit: b56e5d1d79 ]
2025-09-25 10:32:42 -07:00
asleepzzz
5cc40c160f
Revert "[CK-Tile] Add the API to load SGPR ( #2878 )" ( #2904 )
...
This reverts commit fb5e953a05 .
[ROCm/composable_kernel commit: f161b5b738 ]
2025-09-23 14:33:51 -07:00
Thomas Ning
fb5e953a05
[CK-Tile] Add the API to load SGPR ( #2878 )
...
* Have a workable version for SGPR
* have a workable version for atomic add
* Revert "have a workable version for atomic add"
This reverts commit 792377a590c26cfff9c8f545d9a9e8484a7422eb.
* substitute with the new sgpr read api
* update the CHANGELOG
* have a workable version for atomic add
* Revert "have a workable version for atomic add"
This reverts commit 792377a590c26cfff9c8f545d9a9e8484a7422eb.
* change to static for logic
* have a workable version for atomic add
* Revert "have a workable version for atomic add"
This reverts commit 792377a590c26cfff9c8f545d9a9e8484a7422eb.
[ROCm/composable_kernel commit: 2cbbf5dcb3 ]
2025-09-23 01:23:56 -07:00
linqunAMD
7022f756e6
[CK_TILE] Fix flatmm on gfx11 and gfx12 ( #2790 )
...
1. Correct shuffle_b and MakeBFlatDramTileDistribution according to WMMA warp layout
2. Add FlatmmConfig16_Wmma for gfx11 and gfx12
[ROCm/composable_kernel commit: df4ee556d6 ]
2025-09-10 08:28:00 +08:00
linqunAMD
615ca9842d
Support Wave32 in CK_TILE - Part 1 ( #2594 )
...
* Support wave32/wave64 in CK_TILE - Part 1
* remove blocksize in kernel launch
* fix build error
* fix clang format
* fix clang format 2
* fix clang format 3
* fix fmha build error
* fix fmha build 2
* fix fmha build 3
* fix build error 4
* address review comment
* update change log
* replace KernelBlockSize with kBlockSize
* fix CI fail
* fix clang format
* address review comment and rebase code.
* fix universal test fail
---------
Co-authored-by: Lin, Qun <Quentin.Lin+amdeng@amd.com >
Co-authored-by: Thomas Ning <Thomas.Ning@amd.com >
[ROCm/composable_kernel commit: 9fcc1ee9fd ]
2025-08-18 10:08:31 -07:00
Aviral Goel
8bf96c18c6
Integration of a new pipeline for weight preshuffle into gemm examples ( #2516 )
...
* something khushbu can help with
* v1 v2 works with flatmm develop
* v0 v1 v2 numerical error gone
* Fixing numerical error, and interchange preshuffle configs to match with flatmm
* Refactor GEMM pipeline configurations and integrate preshuffle support
- Updated preshuffle pipeline definitions to include multiple versions (V1, V2, V3).
- Changed the pipeline constant from CK_TILE_PIPELINE_PRESHUFFLE to CK_TILE_PIPELINE_PRESHUFFLE_V3 in relevant configurations.
- Removed obsolete code and comments
* clang format
* fix vectorloadsize bug
* add the Preshuffle3
* update kwarp calculation in gemm utils
* update vector size A and B correctly in V2 pipeline; Added few more changes to align with dteng's branch
* fix: add CK_GFX950_SUPPORT macro for gfx950 detection
* default disable rotating buffer
* docs(CHANGELOG): update changelog for rocm 7.0
* Revert "docs(CHANGELOG): update changelog for rocm 7.0"
This reverts commit 2bc16fff84 .
* Remove unused Preshuffle V3 pipeline and related code; update gemm function to use Preshuffle V2; clean up comments and formatting in various files.
* revert example/ck_tile/flatmm to its original state
* remove comment added by second author
* switch to xor ALDSDescriptor
* modify the MakeALdsDescriptor()
* temporary profiling script
* getting rid of line marker compiler error
* UniversalWeightPreshufflePipelineAgBgCrPolicy now derives from UniversalGemmBasePolicy
* add a minor fix for the config
* typo fix
* Fix formatting in lambda function for WeightPreshufflePipelineAGmemBGmemCRegV2
* revert change in include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1.hpp
* revert change in include/ck_tile/core/arch/amd_buffer_addressing.hpp
* reenable the GemmSpatiallyLocalTilePartitioner
* make GemmConfigPreshuffle_1 for v1 pipeline, GemmConfigPreshuffle_2 for v2 pipeline
* remove hardcoded true for preshuffle bool template argument
* rename script
* remove gemm_profilie.sh script
* merge conflict resolve
* clang formatted
* typo fix
* Remove duplicate include of block_gemm_areg_bsmem_creg_v2r1.hpp in gemm.hpp
* Remove commented-out code in UniversalWeightPreshufflePipelineAgBgCrPolicy
* Fix missing newline at end of file in run_gemm_example.inc
* Remove unused barrier call in BlockWeightPreshuffleASmemBSmemCRegV1
* addressing review comments
* removing debug code
* addressing review comments
* Revert "addressing review comments"
This reverts commit 29c45192ba .
* updating tile_engine code
* addressing review comments
---------
Co-authored-by: amd-khushbu <khuagarw@amd.com >
Co-authored-by: ThomasNing <thomas.ning@amd.com >
[ROCm/composable_kernel commit: 1441a0a7ee ]
2025-08-01 00:04:54 -07:00
Khushbu Agarwal
e34599e8a9
Merge flatmm Operator with universal gemm ( #2434 )
...
* Initial commit
* Adding new tile partitioner to flatmm
* intermediate changes
* debugging kernels
* Updating flatmm example to universal gemm example
* updated flatmm kernel to run via gemmKernel
* update universal gemm to incorporate flatmm
* debug
* Fix flatmm call
* Fixing other kernels and tests for API changes
* clang formatted
* fixing gemm tests
* added test for flatmm and simplify kernel arguments
* adding flatmm test
* fix test for flatmm
* simplify gemm kernel with flatmm
* remove flatmm related files
* addressing review comments and code clean up
* resolving empty file
* resolving empty file
* clang formatted
* addressing review comments
* enable persistent kernel for flatmm
* reverted the removed files for flatmm
* reverted the removed files for flatmm
* changed flatmm to weightPReshuffle; removed the _1 added in teh faltmm example
* some more renames
* clang formatted
[ROCm/composable_kernel commit: d239b91fd5 ]
2025-07-11 08:27:55 -07:00
Thomas Ning
bc6af0fa49
Fix the CK Tile related operators ( #2356 )
...
* fix the flatmm
* Fix the pipeline
* address the comment
[ROCm/composable_kernel commit: 3c4cdfac4f ]
2025-06-16 17:38:52 -07:00
Illia Silin
0f4d68633b
Revert "fix the flatmm ( #2349 )" ( #2352 )
...
This reverts commit fc65195605 .
[ROCm/composable_kernel commit: 5523df4b2d ]
2025-06-16 07:54:55 -07:00
Thomas Ning
fc65195605
fix the flatmm ( #2349 )
...
[ROCm/composable_kernel commit: d996bc78be ]
2025-06-16 02:17:53 -07:00
BingYuan.Zhou
977b7d0928
Flatmm merge ( #2168 )
...
* sync with function interface of cshuffleepiloge,fix flatmm build fail
* move code from solin/flatmm which add mfma16*16*32fp8 and optimize flatmm
---------
Co-authored-by: solin <bingzhou@amd.com >
[ROCm/composable_kernel commit: 6a3960c1e1 ]
2025-05-08 12:59:57 +08:00
BingYuan.Zhou
4ec293cb4b
[flatmm] implement basic fp16 flatmm ( #2089 )
...
* [flatmm] implement basic fp16 flatmm
* fix CI build fail
---------
Co-authored-by: root <root@hjbog-srdc-50.amd.com >
Co-authored-by: solin <bingzhou@amd.com >
[ROCm/composable_kernel commit: eaf1f0bf3b ]
2025-04-16 16:51:17 +08:00