From ed48fcb94060633196db67d99202448f8bfb50ab Mon Sep 17 00:00:00 2001
From: RaulPPealez <raulppelaez@gmail.com>
Date: Fri, 17 Nov 2023 11:47:58 +0100
Subject: [PATCH] Set TORCH_CUDA_ARCH_LIST

---
 .github/workflows/ci-cuda12.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/ci-cuda12.yml b/.github/workflows/ci-cuda12.yml
index f8d6fee..e3b0305 100644
--- a/.github/workflows/ci-cuda12.yml
+++ b/.github/workflows/ci-cuda12.yml
@@ -72,7 +72,7 @@ jobs:
               -DENABLE_CUDA=${{ matrix.enable_cuda }} \
               -DTorch_DIR=$(python -c 'import torch.utils; print(torch.utils.cmake_prefix_path)')/Torch \
               -DCMAKE_INSTALL_PREFIX=$CONDA_PREFIX \
-              -DCMAKE_CUDA_ARCHITECTURES=OFF
+              -DTORCH_CUDA_ARCH_LIST=8.9
         make install
 
     - name: Test