From 5696cba38d52e44b8f355dd607444f83e634be9a Mon Sep 17 00:00:00 2001 From: vhjiang Date: Tue, 31 Oct 2023 04:18:29 +0000 Subject: [PATCH] fix: support lower cuda arch --- ec-gpu-gen/src/source.rs | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ec-gpu-gen/src/source.rs b/ec-gpu-gen/src/source.rs index 4813ccd..db3ac41 100644 --- a/ec-gpu-gen/src/source.rs +++ b/ec-gpu-gen/src/source.rs @@ -618,10 +618,11 @@ fn generate_cuda(source_builder: &SourceBuilder) -> PathBuf { // Compile with as many threads as CPUs are available. .arg("--threads=0") .arg("--fatbin") - .arg("--gpu-architecture=sm_86") - .arg("--generate-code=arch=compute_86,code=sm_86") + .arg("--gpu-architecture=sm_70") + .arg("--generate-code=arch=compute_70,code=sm_70") + .arg("--generate-code=arch=compute_72,code=sm_72") .arg("--generate-code=arch=compute_80,code=sm_80") - .arg("--generate-code=arch=compute_75,code=sm_75"); + .arg("--generate-code=arch=compute_86,code=sm_86"); command } };