Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-scatter
Commits
ca3f6c7f
Commit
ca3f6c7f
authored
Feb 13, 2020
by
rusty1s
Browse files
potential windows fix
parent
01ab13d0
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
2 deletions
+9
-2
.travis.yml
.travis.yml
+2
-2
script/cuda.sh
script/cuda.sh
+7
-0
No files found.
.travis.yml
View file @
ca3f6c7f
...
@@ -27,7 +27,7 @@ env:
...
@@ -27,7 +27,7 @@ env:
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu101
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.5 IDX=cu101
jobs
:
jobs
:
exclude
:
# Exclude all macOS CUDA jobs and Windows
cu
100 job.
exclude
:
# Exclude all macOS CUDA jobs and Windows
CUDA
10
.
0 job
s
.
-
os
:
osx
-
os
:
osx
env
:
TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
env
:
TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
-
os
:
osx
-
os
:
osx
...
@@ -63,8 +63,8 @@ jobs:
...
@@ -63,8 +63,8 @@ jobs:
install
:
install
:
-
source script/cuda.sh
-
source script/conda.sh
-
source script/conda.sh
-
source script/cuda.sh
-
conda create --yes -n test python=${PYTHON_VERSION}
-
conda create --yes -n test python=${PYTHON_VERSION}
-
source activate test
-
source activate test
-
if [ ${TRAVIS_OS_NAME} = "windows" ] || [ $IDX != "cu92" ]; then conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch -c defaults -c numba/label/dev --yes; else conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes; fi
-
if [ ${TRAVIS_OS_NAME} = "windows" ] || [ $IDX != "cu92" ]; then conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch -c defaults -c numba/label/dev --yes; else conda install pytorch=${TORCH_VERSION} ${TOOLKIT} -c pytorch --yes; fi
...
...
script/cuda.sh
View file @
ca3f6c7f
...
@@ -102,3 +102,10 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" = "cu92" ]; then
...
@@ -102,3 +102,10 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" = "cu92" ]; then
sed
-i
.bak
-e
'129,141d'
"
${
CUDA_PATH
}
/include/crt/host_config.h"
sed
-i
.bak
-e
'129,141d'
"
${
CUDA_PATH
}
/include/crt/host_config.h"
cat
-n
"
${
CUDA_HOME
}
/include/crt/host_config.h"
cat
-n
"
${
CUDA_HOME
}
/include/crt/host_config.h"
fi
fi
# Fix "member may not be initialized" error on Windows: https://github.com/pytorch/pytorch/issues/27958
if
[
"
${
TRAVIS_OS_NAME
}
"
=
"windows"
]
&&
[
"
${
IDX
}
"
!=
"cpu"
]
;
then
sed
-i
.bak
-e
's/constexpr/const/g'
/c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/script/module.h
sed
-i
.bak
-e
's/constexpr/const/g'
/c/tools/miniconda3/envs/test/lib/site-packages/torch/include/torch/csrc/jit/script/argument_spec.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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment