Add CI for cuda builds

.github/workflows/rust.yml

@@ -60,39 +60,42 @@ jobs:
     - name: Build
       run: cargo build --verbose --features metal
 
-  # cuda:
-  #   name: Build with cuda support
-  #   strategy:
-  #   # Don't stop building if it fails on an OS
-  #     fail-fast: false
-  #     matrix:
-  #       os: [windows-latest, ubuntu-latest]
-  #   runs-on: ${{ matrix.os }}
-  #   steps: 
-  #     - uses: actions/checkout@v3
-  #       with:
-  #         submodules: recursive
-  #     - uses: Jimver/[email protected]
-  #       if: matrix.os == 'ubuntu-latest'
-  #       id: cuda-toolkit-linux
-  #       with:
-  #         cuda: '12.1.0'
-  #         method: 'network'
-  #         #See e.g. https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/
-  #         sub-packages: '["nvcc","compiler","libraries","libraries-dev","cudart","cudart-dev","libcublas","libcublas-dev"]'
-  #     - uses: Jimver/[email protected]
-  #       if: matrix.os == 'windows-latest'
-  #       id: cuda-toolkit-windows
-  #       with:
-  #         cuda: '12.1.0'
-  #         method: 'network'
-  #         #See https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#install-the-cuda-software
-  #         sub-packages: '["nvcc","cudart","visual_studio_integration","cublas_dev","cublas"]'
-  #     - uses: dtolnay/rust-toolchain@stable
-  #     - name: Check
-  #       run: cargo check --verbose
-  #     - name: Build
-  #       run: cargo build --verbose --features cublas
+  cuda:
+    name: Build with cuda support
+    strategy:
+    # Don't stop building if it fails on an OS
+      fail-fast: false
+      matrix:
+        os: [windows-latest, ubuntu-latest]
+    runs-on: ${{ matrix.os }}
+    steps: 
+      - uses: actions/checkout@v3
+        with:
+          submodules: recursive
+      - uses: Jimver/[email protected]
+        name: Install CUDA toolkit on Linux
+        if: matrix.os == 'ubuntu-latest'
+        id: cuda-toolkit-linux
+        with:
+          cuda: '12.2.0'
+          method: 'network'
+          #See e.g. https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/
+          non-cuda-sub-packages: '["libcublas","libcublas-dev"]'
+          sub-packages: '["nvcc","compiler","libraries","libraries-dev","cudart","cudart-dev"]'
+
+      - uses: Jimver/[email protected]
+        name: Install CUDA toolkit on Windows
+        if: matrix.os == 'windows-latest'
+        id: cuda-toolkit-windows
+        with:
+          cuda: '12.2.0'
+          #See https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#install-the-cuda-software
+          method: 'local'
+      - uses: dtolnay/rust-toolchain@stable
+      - name: Check
+        run: cargo check --verbose
+      - name: Build
+        run: cargo build --verbose --features cublas
 
   opencl:
     name: Build with opencl support