"vscode:/vscode.git/clone" did not exist on "56dd6dd4e9895841840f92ad93231672bd45449e"
torch.sh 1.13 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
#!/bin/bash

# Fix "member may not be initialized" error on Windows: https://github.com/pytorch/pytorch/issues/27958
rusty1s's avatar
rusty1s committed
4
5
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.5.0" ]; then
  echo "Fix nvcc for PyTorch 1.5.0"
rusty1s's avatar
rusty1s committed
6
7
  sed -i.bak -e 's/constexpr/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/api/module.h
  sed -i.bak -e 's/constexpr/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/runtime/argument_spec.h
rusty1s's avatar
rusty1s committed
8
9
10
  sed -i.bak -e 's/return \*(this->value)/return \*((type\*)this->value)/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/pybind11/cast.h
fi

rusty1s's avatar
rusty1s committed
11
12
13
14
15
16
# https://github.com/pytorch/pytorch/commit/d2e16dd888a9b5fd55bd475d4fcffb70f388d4f0
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.6.0" ]; then
  echo "Fix nvcc for PyTorch 1.6.0"
  sed -i.bak -e 's/CONSTEXPR_EXCEPT_WIN_CUDA/const/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/api/module.h
  sed -i.bak -e 's/return \*(this->value)/return \*((type\*)this->value)/g' /c/tools/miniconda3/envs/test/lib/site-packages/torch/include/pybind11/cast.h
fi