DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-05-07 01:24:21 +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
Show Stats 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):
@staticmethod
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)
@classmethod
@ -203,8 +204,7 @@ class NvccCompiler(Compiler):
class NvrtcCompiler(Compiler):
@staticmethod
def __version__() -> Tuple[int, int]:
_, version = get_nvcc_compiler()
major, minor = map(int, version.split('.'))
major, minor = map(int, cuda.bindings.__version__.split('.')[:2])
return (major, minor)
@staticmethod