Skip to content

Commit fd42ff0

Browse files
committed
fix
1 parent b9d3369 commit fd42ff0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

script/torch.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,5 @@ fi
1212
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${TORCH_VERSION}" = "1.6.0" ]; then
1313
echo "Fix nvcc for PyTorch 1.6.0"
1414
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
15+
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
1516
fi

0 commit comments

Comments
 (0)