Skip to content

Commit 7d46282

Browse files
committed
Grab stableivalue_conversions.h from torch main
1 parent 7e2d405 commit 7d46282

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

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

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,11 @@ printf "Installing TorchCodec\n"
5050
pip install --pre torchcodec --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/cpu
5151
python -c "import torchcodec; print(torchcodec.__version__)"
5252

53+
printf "Patching Torch\n"
54+
TORCH_INCLUDE_DIR="$(python -c "import torch,os;print(os.path.join(os.path.dirname(torch.__file__),'include'))")"
55+
printf " TORCH_INCLUDE_DIR=$(TORCH_INCLUDE_DIR)\n"
56+
curl -L --silent --output TORCH_INCLUDE_DIR/torch/csrc/stable/stableivalue_conversions.h https://raw.githubusercontent.com/pytorch/pytorch/refs/heads/main/torch/csrc/stable/stableivalue_conversions.h
57+
5358
# 2. Install torchaudio
5459
printf "* Installing fsspec\n" # TODO: is this required for torchaudio??
5560
pip install --trusted-host pypi.org --trusted-host files.pythonhosted.org fsspec

0 commit comments

Comments
 (0)