chenchenghao/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-28 09:40:49 +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
csrc/jit/handle.hpp
+1 -1
View File
@@ -8,7 +8,7 @@
namespace deep_gemm { namespace deep_gemm {
#if CUDART_VERSION >= 12080 and not defined(DG_JIT_USE_DRIVER_API) #if CUDART_VERSION >= 12080 and defined(DG_JIT_USE_RUNTIME_API)
// Use CUDA runtime API // Use CUDA runtime API
using LibraryHandle = cudaLibrary_t; using LibraryHandle = cudaLibrary_t;
setup.py
+3 -3
View File
@@ -28,9 +28,9 @@ third_party_include_dirs = [
'third-party/cutlass/include/cutlass', 'third-party/cutlass/include/cutlass',
] ]
# Use driver API for older CUDA compatibility # Use runtime API
if int(os.environ.get('DG_JIT_USE_DRIVER_API', '0')): if int(os.environ.get('DG_JIT_USE_RUNTIME_API', '0')):
cxx_flags.append('-DDG_JIT_USE_DRIVER_API') cxx_flags.append('-DDG_JIT_USE_RUNTIME_API')
class CustomBuildPy(build_py): class CustomBuildPy(build_py):