Commit Graph

742 Commits

Author SHA1 Message Date
coderfeli
d8731a7599 add logits 2025-04-16 09:31:37 +00:00
Bernard
9f73300e80 hack for cap_logits naive 2025-04-16 09:25:10 +00:00
coderfeli
795e2b1e43 fix 2 2025-04-15 09:52:10 +00:00
coderfeli
ff281e135d fix multi batch and hack page idx core 2025-04-15 01:21:23 +00:00
coderfeli
234528e06c tile gather support async pipe ok 2025-04-12 14:45:38 +00:00
coderfeli
c2cdfda718 merge pa 2025-04-11 02:40:23 +00:00
coderfeli
b9d4a20a42 v cache ok 2025-04-09 12:00:59 +00:00
coderfeli
4b5859f717 refine apis 2025-04-08 03:42:46 +00:00
coderfeli
8d62ff557a rename to tile_scatter_gather 2025-04-08 02:50:46 +00:00
coderfeli
bfbd28c9d8 support page dim configure in tile window 2025-04-08 01:53:17 +00:00
coderfeli
738a7427cb fix hacks 2025-04-07 12:10:55 +00:00
coderfeli
4e644a33ab fix k origin 2025-04-07 10:04:22 +00:00
coderfeli
57c9d84eb1 run ok 2025-04-07 03:45:21 +00:00
coderfeli
fe2ea699e5 paged window run ok 2025-04-05 13:45:46 +00:00
coderfeli
867a4e527c fix bugs 2025-04-04 13:40:10 +00:00
coderfeli
491178276f fix fp8 scale 2025-04-03 11:10:37 +00:00
lalala-sh
3037d8bac1 mul int4 scale 2025-04-03 18:06:12 +08:00
root
20f6674bf6 fix no quant case 2025-04-03 02:46:01 +00:00
root
b2b34fffbb fix fp8 16x16 2025-04-02 16:27:52 +00:00
root
85f83330b5 fuse moe activation 2025-04-02 07:02:09 +00:00
root
45a0463f1f moe gemm draft v0.1 2025-03-26 08:14:11 +00:00
lalala-sh
938f4234f6 scatter gemm v1 2025-03-25 07:28:50 +00:00
coderfeli
98cee8d02b fix merge 2025-03-18 05:45:04 +00:00
coderfeli
5f49b91237 merge develop 2025-03-18 04:49:40 +00:00
aledudek
5095906975 Async grouped gemm v3 (#1940)
* Fully async grouped gemm

* Remove commented code

* Remvoe maybe_unused

* host kernel args

* Checkpoint segfault debugging...

* Working part1

* Working part2

* Remvoe comments...

* Use void ptr for gemm kernel host args

* Fix device_grouped_gemm_multiple_d_dl build issue

* Fix device_grouped_gemm_xdl build issue
2025-03-17 16:42:43 +01:00
Bartłomiej Kocot
c2e4898b4b Grouped conv bwd data NGCHW (#1967)
* Grouped conv bwd data NGCHW

* fixes

* fix

* Improvements

* Fix

* Fix

* add client example
2025-03-17 13:32:00 +01:00
coderfeli
7dbdff9f9f moe sorting fix moebuf 2025-03-17 06:20:57 +00:00
coderfeli
5eaa36be18 mork to support 13w tokens 2025-03-17 01:45:34 +00:00
coderfeli
ef8c1333b9 use uint32 2025-03-17 01:45:09 +00:00
coderfeli
6c0e021235 revert v1 test 2025-03-17 01:39:57 +00:00
coderfeli
bccc5192cf fix uint32 2025-03-17 01:18:32 +00:00
coderfeli
da2659d502 input output all ok 2025-03-15 14:26:30 +00:00
coderfeli
d1e999c05c int64 index ok now 2025-03-15 13:28:49 +00:00
coderfeli
f911cf7396 impl int64 but result not correct 2025-03-14 13:01:07 +00:00
coderfeli
d4925e1637 fix output oob 2025-03-14 03:19:26 +00:00
carlushuang
3e81279d26 Reapply "[CK_TILE] support hdim=192/128 pair for deepseekv3 (#1961)" … (#1971)
* Reapply "[CK_TILE] support hdim=192/128 pair for deepseekv3 (#1961)" (#1969)

This reverts commit 8cbcd3e0d0.

* fix codegen problem

* Update config.hpp

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
2025-03-13 11:41:39 +08:00
lalala-sh
948aaa5641 prepare moe gemm 2025-03-13 03:33:55 +00:00
illsilin
f8464d2087 fix clang format 2025-03-12 20:21:14 -07:00
coderfeli
d85c034977 fix2 2025-03-13 02:30:07 +00:00
coderfeli
8b05fa935d fix coredump in e2e test 2025-03-13 02:12:18 +00:00
feli
251afab3b7 ck_moe: fix useless code and remove usless oob (#1972)
* fix useless code and remove usless oob

* clang format

---------

Co-authored-by: coderfeli <coderfeli@163.com>
2025-03-12 09:22:42 -07:00
Illia Silin
4c97cc511e use old instrinsics with staging compiler (#1970) 2025-03-12 07:29:09 -07:00
feli
2585c78940 Merge branch 'develop' into ck_moe_rm_oob 2025-03-12 16:05:59 +08:00
coderfeli
40542296de clang format 2025-03-12 08:05:12 +00:00
Illia Silin
8cbcd3e0d0 Revert "[CK_TILE] support hdim=192/128 pair for deepseekv3 (#1961)" (#1969)
This reverts commit 7a93b16ff6.
2025-03-11 10:40:18 -07:00
Haocong WANG
cbd74c2d12 [Block Scale GEMM] Optimized block scale gemm (#1950)
* Added two kernel for M=32 problem

* Comment the first one

* Enable multiply_multiply for Scale_Block_M = 1 for deepseek

* Modify the a_thread offset since the A data load is different from B.

* edit fp8 ab scale for Scale_Block_M=1

* edit GemmSpec to MNKPadding

* enable blockwise pipelie v1 and v2. v1 is work for small K.

* add instance for gemm_ab_scale

* fix cmakelist of ckProfiler

* optimize blockscale gemm. todo: reduce vgpr usage

* fix a correctness bug

* sanity checked

* revert ckprofiler cmake changes

* clang format

* revert unnecessary changes.

* remove commented codes.

* split weight preshuffle library targets

* bring back enable-post-misched=0

* fix build issues for gemm_multiply_multiply_fp8 instances

* fix clang format

* add verbose build flag when building for all targets

* reduce path names for new instances

* fix paths in cmake

* refactor gemm_multiply_multiply library target

* fix a bug in example

* fix example 65 cmake

* reduce the number of threads when building libs for all targets to 50

* use ninja to build for all targets

* reduce teh number of threads when building for all targets

* reduce the number of threads to 32 when building libs for all targets to 50

---------

Co-authored-by: mtgu0705 <mtgu@amd.com>
Co-authored-by: chenjun <junchen2@amd.com>
Co-authored-by: illsilin <Illia.Silin@amd.com>
Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
2025-03-11 10:11:21 -07:00
carlushuang
7a93b16ff6 [CK_TILE] support hdim=192/128 pair for deepseekv3 (#1961)
* support hdim=192/128 pair

* remove useless print

* update
2025-03-11 21:07:40 +08:00
coderfeli
1508cd514a fix useless code and remove usless oob 2025-03-11 07:05:05 +00:00
Mingtao Gu
0db7c8f0b2 Ck int4 moe develop (#1949)
* Add Gemm fp8xint4 example and kernel, function pass.

* Init Gemm_fp8xint4 Bpreshuffle

* Added gemm_fp8xint4_Bpreshuffle files, function not checked yet

* General fix.

* fp8xint4 bpreshuffle function pass

* fix.

* init b preshuffle dequant in VGPR.

* fix bug, function pass.

* move b thread dequant copy to blockwise.

* fix bug, function now passes.

* modified the tile size to 256, 128x128x128.

* fixed a bug.

* Initial int4 moe, compile pass, function not check.

* fix bug in moe_gemm1.cpp, now function pass.

* test expert = 8 and function pass.

* Added moe_pk_i4_gemm2, function pass.

* Added b preshuffle pipeline v3 support.

* fixed merge issue. fp8xint4 and fp8xint4_bpreshuffle function pass.

* Split the blockwise pipeline for fp8xint4.

* commit missing files

* opt gemm2 to 2x2 wave

* fix swizzle = false

* update int4 moe with latest input changes.

* update tile size.

* enable pipeline v3.

* fix nswizzle = true

* commit a version for compiler debug.

* Updated transfer_v3r1_gather to support pk_i4_t type.

* for int4 moe2 for type_convert support.

* remove some values between mfma instructions.

* fix int4 moe

* Updated transfer_v3r1_gather to support pk_i4_t type.

* i4 support lds multiple shuffle

* fixed int4 moe tflops calculation.

* Modified CshuffleCShuffleMXdlPerWavePerShuffle to 1 to suit C multiple shuffle

* updated gemm2.

* change int4 moe example names

* fix and format code.

* format.

* format codes.

* update fp8xint4 example tile size.

* add <unordered_map> header

* fixed.

* format.

* Added conditional compilation for int4 -> fp8 conversion kernels

---------

Co-authored-by: mtgu0705 <mtgu@amd.com>
Co-authored-by: coderfeli <coderfeli@163.com>
2025-03-10 11:16:44 +08:00
Thomas Ning
9d51d17dd0 Fix on the error (#1956) 2025-03-07 13:43:52 -08:00