Skip to content

Commit 1da314a

Browse files
committed
fixup! Initialize the CUDA architectures defaults based on env vars checked by CMake
1 parent b9c7b03 commit 1da314a

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

src/aedifix/packages/cuda.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -131,10 +131,7 @@ class CUDA(Package):
131131
spec=ArgSpec(
132132
dest="cuda_arch",
133133
required=False,
134-
default=os.environ.get(
135-
"CUDAARCHS",
136-
os.environ.get("CMAKE_CUDA_ARCHITECTURES", "all-major"),
137-
),
134+
default=_guess_cuda_architecture(),
138135
action=CudaArchAction,
139136
help=(
140137
"Specify the target GPU architecture. Available choices are: "

0 commit comments

Comments
 (0)