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

Drop support for Windows

Browse Source
This commit is contained in:
Chenggang Zhao 2025-05-07 11:24:35 +08:00
parent a29b331c48
commit 41b4cff7c8
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
deep_gemm/jit/compiler.py
View File

@ -1,7 +1,6 @@
import functools import functools
import hashlib import hashlib
import os import os
import platform
import re import re
import subprocess import subprocess
import time import time
@ -52,8 +51,7 @@ def get_nvcc_compiler() -> Tuple[str, str]:
if os.getenv('DG_JIT_NVCC_COMPILER'): if os.getenv('DG_JIT_NVCC_COMPILER'):
paths.append(os.getenv('DG_JIT_NVCC_COMPILER')) paths.append(os.getenv('DG_JIT_NVCC_COMPILER'))
nvcc_bin = 'nvcc.exe' if platform.system() == 'Windows' else 'nvcc' paths.append(os.path.join(CUDA_HOME, 'bin', 'nvcc'))
paths.append(os.path.join(CUDA_HOME, 'bin', nvcc_bin))
# Try to find the first available NVCC compiler # Try to find the first available NVCC compiler
least_version_required = '12.3' least_version_required = '12.3'
@ -187,11 +185,7 @@ class NVCCCompiler(Compiler):
@classmethod @classmethod
def flags(cls) -> List[str]: def flags(cls) -> List[str]:
if platform.system() != 'Windows':
cxx_flags = ['-fPIC', '-O3', '-fconcepts', '-Wno-deprecated-declarations', '-Wno-abi'] cxx_flags = ['-fPIC', '-O3', '-fconcepts', '-Wno-deprecated-declarations', '-Wno-abi']
else:
cxx_flags = ['/O2', '/std:c++20']
return [*super().flags(), *[f'-I{d}' for d in cls.include_dirs()], return [*super().flags(), *[f'-I{d}' for d in cls.include_dirs()],
'-gencode=arch=compute_90a,code=sm_90a', '-gencode=arch=compute_90a,code=sm_90a',
'-cubin', '-O3', '--expt-relaxed-constexpr', '--expt-extended-lambda', '-cubin', '-O3', '--expt-relaxed-constexpr', '--expt-extended-lambda',