adding cmake_flags to CUDA build GA workflow

flatironinstitute · Nov 26, 2023 · 7268e33 · 7268e33
1 parent b2ab169
commit 7268e33
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/gpu-build.yml b/.github/workflows/gpu-build.yml
@@ -34,6 +34,8 @@ jobs:
 
       - name: Compile the extension
         run: python -m pip install -v .
+        env:
+          CMAKE_FLAGS: "-DJAX_FINUFFT_USE_CUDA=ON"
 
       - name: Check that the GPU extension was built
         run: python -c "import jax_finufft.jax_finufft_gpu"