cuda feature (#8)

Co-authored-by: Hanting Zhang <[email protected]>
lurk-lab · Jan 11, 2024 · 12a0fdc · 12a0fdc
1 parent e30d10d
commit 12a0fdc
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");
 }