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

fix: compiler version

Browse Source 
Signed-off-by: Zihua Wu <13583761+lucifer1004@users.noreply.github.com>
This commit is contained in:
Zihua Wu
2025-04-23 00:06:18 +00:00
parent c14cad0c06
commit 78c7fa347e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
deep_gemm/jit/compiler.py
View File

@@ -176,7 +176,8 @@ class Compiler(abc.ABC):
class NvccCompiler(Compiler): class NvccCompiler(Compiler):
@staticmethod @staticmethod
def __version__() -> Tuple[int, int]: def __version__() -> Tuple[int, int]:
major, minor, _ = map(int, cuda.bindings.__version__.split('.')) _, version = get_nvcc_compiler()
major, minor = map(int, version.split('.'))
return (major, minor) return (major, minor)
@classmethod @classmethod
@@ -203,8 +204,7 @@ class NvccCompiler(Compiler):
class NvrtcCompiler(Compiler): class NvrtcCompiler(Compiler):
@staticmethod @staticmethod
def __version__() -> Tuple[int, int]: def __version__() -> Tuple[int, int]:
_, version = get_nvcc_compiler() major, minor = map(int, cuda.bindings.__version__.split('.')[:2])
major, minor = map(int, version.split('.'))
return (major, minor) return (major, minor)
@staticmethod @staticmethod