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 |
|
yukuai26
|
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 |
|
Chenggang Zhao
|
ea9c5d9270
|
Use driver API
|
2025-08-28 09:40:49 +08:00 |
|
Chenggang Zhao
|
0e49c3353b
|
Refactor compiler version checks and arch flags
|
2025-08-27 09:28:21 +08:00 |
|
PGFLMG
|
3a93f4eb28
|
Fix B200 cu128 NVCC compilation failed (#173)
|
2025-08-27 09:07:18 +08:00 |
|
Chenggang Zhao
|
9c3783beb2
|
Fix CUBIN symbol name compatibility
|
2025-08-26 17:43:26 +08:00 |
|
Chenggang Zhao
|
f20256fd50
|
Compatible with CUDA 13
|
2025-08-22 17:30:47 +08:00 |
|
xiweny
|
affdb1cd90
|
Add sm_100f support and make nvcc 13 happy (#157)
Signed-off-by: Xiwen Yu <13230610+VALLIS-NERIA@users.noreply.github.com>
|
2025-08-22 17:19:32 +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 |
|
dan_the_3rd
|
4b4e4f20dd
|
Update system.hpp (#133)
|
2025-07-28 17:01:05 +08:00 |
|
Chenggang Zhao
|
187656694f
|
Code lint
|
2025-07-21 11:00:50 +08:00 |
|
Ray Wang
|
436a56314c
|
Use std::filesystem::directory_iterator instead of std::filesystem::recursive_directory_iterator to avoid an ABI breakage we met (#131)
|
2025-07-21 10:44:20 +08:00 |
|
Ray Wang
|
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 |
|