 Ray Wang
|
ec5e9ed0b8
|
Fix SM90 MQA logits (#229)
|
2025-11-19 10:50:36 +08:00 |
|
|
Ray Wang
|
2f9d87877e
|
Use larger MMA shape (#227)
|
2025-11-14 11:38:15 +08:00 |
|
|
Chenggang Zhao
|
07b82fb8cd
|
Fix old CUDA compatibility
|
2025-10-01 20:29:15 +08:00 |
|
|
Simon Mo
|
59f2c07cf2
|
Add SM100 kernels (#201)
Signed-off-by: simon-mo <simon.mo@hey.com>
|
2025-09-29 17:07:28 +08:00 |
|
|
Chenggang Zhao
|
80ceeb2c76
|
Add SM90 kernels (#200)
|
2025-09-29 17:00:23 +08:00 |
|
|
Ray Wang
|
3f71de7aa9
|
Make various updates and fixes (#198)
|
2025-09-25 16:19:07 +08:00 |
|
|
|
79f48ee15a
|
Fix multicast bug and optimize masked GEMM (#193)
* Fix multicast bug and profile masked GEMM
* Updates and lint
---------
Co-authored-by: Kuai Yu <yukuai@deepseek.com>
Co-authored-by: Chenggang Zhao <chenggangz@deepseek.com>
|
2025-09-12 17:12:27 +08:00 |
|
|
Ray Wang
|
f85ec649d7
|
Make various updates and fixes: (#164)
- Add BF16 support for SM90 and SM100
- Refactor Python APIs
- Other fixes and code refactoring
|
2025-08-15 18:32:35 +08:00 |
|
|
zhonghui-J
|
3254b758e2
|
Polish get_best_configs modeling. (#158)
|
2025-08-14 16:50:21 +08:00 |
|
|
LJC00118
|
7b6b5563b9
|
Fix smxx layout assertion (#154)
|
2025-08-05 10:38:06 +08:00 |
|
|
Ray Wang
|
d9c363f86f
|
Make various updates and fixes:
- Add support for legacy CUDA versions; now compatible with CUDA 12.3 and newer
- Add support for NVRTC compilation
- Other fixes and code refactoring
|
2025-08-02 19:52:22 -07:00 |
|
|
Chenggang Zhao
|
c50deed14c
|
Code lint
|
2025-07-30 10:39:30 +08:00 |
|
|
LJC00118
|
6bc75b549e
|
Fix smxx layout assertion (#141)
* Fix assertion error in smxx_layout.hpp for mn % 4 != 0 cases
* Fix assertion error in smxx_layout.hpp for mn % 4 != 0 cases
* Align submodule files
* Fix assertion error in smxx_layout.hpp for mn % 4 != 0 cases
* fix(smxx_layout): support mn%4!=0 and num_groups>1 via torch
* fix(smxx_layout): support mn%4!=0 and num_groups>1 via torch
* fix: correct logic for entering get_mn_major_tma_aligned_packed_ue8m0_tensor_torch
|
2025-07-30 10:36:54 +08:00 |
|
|
Ray Wangandfzyzcjy
|
9da4a23561
|
Add more GPU architectures support (#112)
* Add more GPU architectures support
* Update layout.py
* Optimize performance, Add SM90 support, Add 1D2D SM100 support
* Add fmtlib submodule at commit 553ec11
---------
Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
|
2025-07-18 11:32:22 +08:00 |
|