Skip to content

Commit c8e73d3

Browse files
authored
Define missed CUDA architectures for non x86_64 builds. (#318) (#319)
* Enable all-major CUDA architectures for non x86_64 builds. * Hardcoding the value
1 parent e0c13ae commit c8e73d3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,9 @@ def dockerfile_for_linux(output_file):
323323
else:
324324
cuda_archs = "87"
325325
else:
326-
if os.getenv("CUDA_ARCH_LIST") is not None:
326+
if os.uname().machine != "x86_64":
327+
cuda_archs = "80;86;90;100;110;120;121"
328+
elif os.getenv("CUDA_ARCH_LIST") is not None:
327329
print(f"[INFO] Defined CUDA_ARCH_LIST: {os.getenv('CUDA_ARCH_LIST')}")
328330
cuda_archs = (
329331
os.getenv("CUDA_ARCH_LIST")

0 commit comments

Comments
 (0)