Skip to content

Commit b16bbe4

Browse files
authored
3rdparty: add dependency to cutlass and composable kernels (#269)
For group gemm in MoE and LoRA models, we use the combination of custom CUDA kernels and cutlass & composable kernels (for NVIDIA & AMD GPUs, correspondingly). This PR adds dependency to the two repos.
1 parent 2bdacfe commit b16bbe4

File tree

3 files changed

+8
-0
lines changed

3 files changed

+8
-0
lines changed

.gitmodules

+6
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,9 @@
77
[submodule "3rdparty/mscclpp"]
88
path = 3rdparty/mscclpp
99
url = https://github.com/microsoft/mscclpp.git
10+
[submodule "3rdparty/cutlass"]
11+
path = 3rdparty/cutlass
12+
url = https://github.com/NVIDIA/cutlass.git
13+
[submodule "3rdparty/composable_kernels"]
14+
path = 3rdparty/composable_kernels
15+
url = https://github.com/ROCm/composable_kernel.git

3rdparty/composable_kernels

Submodule composable_kernels added at 5055b3b

3rdparty/cutlass

Submodule cutlass added at 033d9ef

0 commit comments

Comments
 (0)