3 ms·
It seems mostly Python, which is nice: https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm/jit_kernels/gemm.py https://github.com/deepseek-ai/DeepGEMM/b
by fulafel 2y ago
It seems mostly Python, which is nice: https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm/jit_kernels/gemm.py https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm/...
- azurezyq 2y agoThe secret is here https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm%2Finclude%2Fdeep_gemm%2Ffp8_gemm.cuh https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm%...
- fulafel 2y agoAnd the yield bit modification: https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm/jit/interleave_ffma.py https://github.com/deepseek-ai/DeepGEMM/blob/main/deep_gemm/...