We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 149007b commit bc02c75Copy full SHA for bc02c75
1 file changed
setup.py
@@ -83,8 +83,10 @@ def detect_gpu_backend() -> str:
83
backend = os.environ.get('GPU4PYSCF_GPU_BACKEND')
84
if backend is None:
85
backend = os.environ.get('GPU4PYSCF_BACKEND')
86
- if backend is None:
+ if backend is None and not get_hip_version is None:
87
backend = 'HIP'
88
+ else:
89
+ backend = 'CUDA'
90
return normalize_backend(backend)
91
92
@@ -159,6 +161,7 @@ def run(self):
159
161
if hip_arch:
160
162
cmd.append(f'-DCMAKE_HIP_ARCHITECTURES={hip_arch}')
163
cmd.append('-DBUILD_CUTLASS=OFF')
164
+ cmd.append('-DCMAKE_HIP_FLAGS=-Wno-unused-result')
165
configure_args = os.getenv('CMAKE_CONFIGURE_ARGS')
166
if configure_args:
167
cmd.extend(shlex.split(configure_args))
0 commit comments