wheel-5

punica-ai · Dec 22, 2023 · 1d21193 · 1d21193
1 parent 165e548
commit 1d21193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build_wheel.yml b/.github/workflows/build_wheel.yml
@@ -7,7 +7,7 @@ on:
       - "v[0-9]+*"
 
 env:
-  TORCH_CUDA_ARCH_LIST: "8.0 8.6 8.9 9.0+PTX"
+  TORCH_CUDA_ARCH_LIST: "8.0 8.6 8.9+PTX"  # Need fix for 9.0
   PUNICA_CI_TORCH_VERSION: "2.1.0"
 
 jobs: