cuda feature

argumentcomputer · Jan 8, 2024 · 319da55 · 319da55
1 parent 311ec73
commit 319da55
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/build.rs b/build.rs
@@ -29,6 +29,8 @@ fn main() {
         compile_cuda("cuda/pallas.cu", "pallas_msm_cuda", implement_sort);
         compile_cuda("cuda/vesta.cu", "vesta_msm_cuda", implement_sort);
         println!("cargo:rerun-if-changed=cuda");
+        println!("cargo:rerun-if-env-changed=CXXFLAGS");
+        println!("cargo:rustc-cfg=feature=\"cuda\"");
     }
     println!("cargo:rerun-if-env-changed=NVCC");
 }