DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Several fixes

Browse Source
This commit is contained in:
Chenggang Zhao
2025-05-07 09:57:39 +08:00
parent 317e83581d
commit 83f6e9537e
4 changed files with 30 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
deep_gemm/jit_kernels/runtime.py
View File

@@ -176,7 +176,7 @@ class FP8GemmRuntime(Runtime):
using namespace deep_gemm;
__global__ void dummy_kernel() {{
void *ptr = (void *)&fp8_gemm_kernel<
auto ptr = reinterpret_cast<void*>(&fp8_gemm_kernel<
{kwargs['N']},
{kwargs['K']},
{kwargs['BLOCK_M']},
@@ -191,7 +191,7 @@ __global__ void dummy_kernel() {{
{kwargs['NUM_TMA_MULTICAST']},
{'true' if kwargs['IS_TMA_MULTICAST_ON_A'] else 'false'},
GemmType::{kwargs['GEMM_TYPE']}
>;
>);
}}
'''
if int(os.getenv('DG_JIT_DEBUG', 0)):