Skip to content

Commit cdb0af0

Browse files
committed
Test torch PR 169475
1 parent 62e7927 commit cdb0af0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/scripts/unittest-windows/install.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,10 @@ fi
3232
printf "Installing PyTorch\n"
3333
pip install --pre torch --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${wheel}
3434

35+
printf "Patching Torch\n"
36+
TORCH_INCLUDE_DIR="$(python -c "import torch,os;print(os.path.join(os.path.dirname(torch.__file__),'include'))")"
37+
curl -L --silent --output ${TORCH_INCLUDE_DIR}/torch/csrc/stable/stableivalue_conversions.h https://raw.githubusercontent.com/pytorch/pytorch/refs/heads/gh/mikaylagawarecki/392/head/torch/csrc/stable/stableivalue_conversions.h
38+
3539
python -c "import torch; print(torch.__version__)"
3640

3741
torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")

0 commit comments

Comments
 (0)