File tree 1 file changed +8
-6
lines changed
1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -36,12 +36,14 @@ $ python run.py --op gemm
36
36
37
37
We depend on the following projects as a source of customized Triton or CUTLASS kernels:
38
38
39
- * (Required) [ FBGEMM] ( https://github.com/pytorch/FBGEMM )
40
- * (Required) [ kernels] ( https://github.com/triton-lang/kernels )
41
- * (Required) [ generative-recommenders] ( https://github.com/facebookresearch/generative-recommenders )
42
- * (Optional) [ ThunderKittens] ( https://github.com/HazyResearch/ThunderKittens )
43
- * (Optional) [ cutlass-kernels] ( https://github.com/ColfaxResearch/cutlass-kernels )
44
- * (Optional) [ flash-attention] ( https://github.com/Dao-AILab/flash-attention )
39
+ * (CUDA, HIP) [ kernels] ( https://github.com/triton-lang/kernels )
40
+ * (CUDA, HIP) [ generative-recommenders] ( https://github.com/facebookresearch/generative-recommenders )
41
+ * (CUDA, HIP) [ Liger-Kernel] ( https://github.com/linkedin/Liger-Kernel )
42
+ * (CUDA) [ xformers] ( https://github.com/facebookresearch/xformers )
43
+ * (CUDA) [ flash-attention] ( https://github.com/Dao-AILab/flash-attention )
44
+ * (CUDA) [ FBGEMM] ( https://github.com/pytorch/FBGEMM )
45
+ * (CUDA) [ ThunderKittens] ( https://github.com/HazyResearch/ThunderKittens )
46
+ * (CUDA) [ cutlass-kernels] ( https://github.com/ColfaxResearch/cutlass-kernels )
45
47
46
48
47
49
## License
You can’t perform that action at this time.
0 commit comments