File tree Expand file tree Collapse file tree 2 files changed +4
-8
lines changed
Expand file tree Collapse file tree 2 files changed +4
-8
lines changed Original file line number Diff line number Diff line change @@ -16,11 +16,11 @@ CUDA_LINK=${CUDA_LINKS[$CUDA_VERSION]}
1616CUDA_SHORT=${CUDA_VERSION: 2: 2} .${CUDA_VERSION: 4: 1}
1717curl -k -L $CUDA_LINK --output _cuda_installer.exe
1818
19- # crt component is separate in CUDA 13.0+
20- CRT_COMPONENT =" "
19+ # crt and nvvm components are separate in CUDA 13.0+
20+ EXTRA_COMPONENTS =" "
2121if [[ ${CUDA_VERSION: 2: 2} -ge 13 ]]; then
22- CRT_COMPONENT =" crt_${CUDA_SHORT} "
22+ EXTRA_COMPONENTS =" crt_${CUDA_SHORT} nvvm_ ${CUDA_SHORT} "
2323fi
2424
25- PowerShell -Command " Start-Process -FilePath \" _cuda_installer.exe\" -ArgumentList \" -s nvcc_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cudart_${CUDA_SHORT} cufft_dev_${CUDA_SHORT} curand_dev_${CUDA_SHORT} cusolver_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} thrust_${CUDA_SHORT} npp_dev_${CUDA_SHORT} nvrtc_dev_${CUDA_SHORT} nvml_dev_${CUDA_SHORT} ${CRT_COMPONENT } \" -Wait -NoNewWindow"
25+ PowerShell -Command " Start-Process -FilePath \" _cuda_installer.exe\" -ArgumentList \" -s nvcc_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cudart_${CUDA_SHORT} cufft_dev_${CUDA_SHORT} curand_dev_${CUDA_SHORT} cusolver_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} thrust_${CUDA_SHORT} npp_dev_${CUDA_SHORT} nvrtc_dev_${CUDA_SHORT} nvml_dev_${CUDA_SHORT} ${EXTRA_COMPONENTS } \" -Wait -NoNewWindow"
2626rm _cuda_installer.exe
Original file line number Diff line number Diff line change @@ -80,10 +80,6 @@ def main():
8080 if os_name == MACOS_ARM64 and pv [1 ] <= 9 :
8181 continue
8282
83- # skip CUDA 13.0 on Windows for now
84- if os_name == WINDOWS_X64 and compute_platform == "cu130" :
85- continue
86-
8783 jobs .append ({
8884 "os" : os_name ,
8985 "torch-version" : torch_version ,
You can’t perform that action at this time.
0 commit comments