sync with cupy

cupy · Aug 20, 2024 · e300c82 · e300c82
1 parent 8e35273
commit e300c82
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.pfnci/wheel-windows/_flexci.ps1 b/.pfnci/wheel-windows/_flexci.ps1
@@ -56,8 +56,10 @@ function ActivateCUDA($version) {
         $Env:CUDA_PATH = $Env:CUDA_PATH_V12_4
     } elseif ($version -eq "12.5") {
         $Env:CUDA_PATH = $Env:CUDA_PATH_V12_5
+    } elseif ($version -eq "12.6") {
+        $Env:CUDA_PATH = $Env:CUDA_PATH_V12_6
     } elseif ($version -eq "12.x") {
-        $Env:CUDA_PATH = $Env:CUDA_PATH_V12_5
+        $Env:CUDA_PATH = $Env:CUDA_PATH_V12_6
     } else {
         throw "Unsupported CUDA version: $version"
     }