biondizzle/DeepGEMM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use driver API

Browse Source
This commit is contained in:
Chenggang Zhao
2025-08-19 11:29:15 +08:00
parent 51d1e9cdd3
commit ea9c5d9270
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@@ -28,9 +28,9 @@ third_party_include_dirs = [
'third-party/cutlass/include/cutlass',
]
# Use driver API for older CUDA compatibility
if int(os.environ.get('DG_JIT_USE_DRIVER_API', '0')):
cxx_flags.append('-DDG_JIT_USE_DRIVER_API')
# Use runtime API
if int(os.environ.get('DG_JIT_USE_RUNTIME_API', '0')):
cxx_flags.append('-DDG_JIT_USE_RUNTIME_API')
class CustomBuildPy(build_py):