Skip to content

Add Windows cuda 12.9.1 build #156630

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
atalman wants to merge 11 commits into from
Closed

Add Windows cuda 12.9.1 build #156630

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
f8cf9a7
Add Windows cuda 12.9.1 build
tinglvv Jun 11, 2025
0391d7f
Set MEMORY_LIMIT_MAX_JOBS 12->4
atalman Jun 17, 2025
0169f75
Fix OOM error due to path typo
tinglvv Jun 17, 2025
53ba783
commend_segment_reduce
atalman Jun 23, 2025
57305e8
fix
atalman Jun 23, 2025
289fc7a
fix
atalman Jun 23, 2025
fdbd9ba
fix
atalman Jun 23, 2025
e104649
lint
atalman Jun 23, 2025
188a8fe
tune
atalman Jun 23, 2025
174fa8c
try_fix
atalman Jun 23, 2025
6031650
Update aten/src/ATen/native/cuda/SegmentReduce.cu
atalman Jun 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .ci/pytorch/windows/cuda129.bat
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ set CMAKE_GENERATOR=Visual Studio 15 2017 Win64

IF "%CUDA_PATH_V129%"=="" (
IF EXIST "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9\bin\nvcc.exe" (
set "CUDA_PATH_V128=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9"
set "CUDA_PATH_V129=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9"
) ELSE (
echo CUDA 12.9 not found, failing
exit /b 1
Expand Down
5 changes: 0 additions & 5 deletions .github/scripts/generate_binary_build_matrix.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -224,8 +224,6 @@ def generate_libtorch_matrix(
arches += ROCM_ARCHES
elif os == "windows":
arches += CUDA_ARCHES
if "12.9" in arches:
arches.remove("12.9")
if libtorch_variants is None:
libtorch_variants = [
"shared-with-deps",
Expand Down Expand Up @@ -291,9 +289,6 @@ def generate_wheels_matrix(
arches += CUDA_ARCHES + ROCM_ARCHES + XPU_ARCHES
elif os == "windows":
arches += CUDA_ARCHES + XPU_ARCHES
# skip CUDA 12.9 builds on Windows
if "12.9" in arches:
arches.remove("12.9")
elif os == "linux-aarch64":
# Separate new if as the CPU type is different and
# uses different build/test scripts
Expand Down
250 changes: 250 additions & 0 deletions .github/workflows/generated-windows-binary-libtorch-debug-nightly.yml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading