File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -17,9 +17,9 @@ before_install:
1717 - export CC="gcc-4.9"
1818 - export CXX="g++-4.9"
1919install :
20- - if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.0 .0-cp27-cp27mu-linux_x86_64.whl; fi
21- - if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.0 .0-cp35-cp35m-linux_x86_64.whl; fi
22- - if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.0 .0-cp36-cp36m-linux_x86_64.whl; fi
20+ - if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.1 .0-cp27-cp27mu-linux_x86_64.whl; fi
21+ - if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.1 .0-cp35-cp35m-linux_x86_64.whl; fi
22+ - if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install https://download.pytorch.org/whl/cpu/torch-1.1 .0-cp36-cp36m-linux_x86_64.whl; fi
2323 - pip install pycodestyle
2424 - pip install flake8
2525 - pip install codecov
Original file line number Diff line number Diff line change 22from torch .testing import get_all_dtypes
33
44dtypes = get_all_dtypes ()
5+
56dtypes .remove (torch .half )
67dtypes .remove (torch .short ) # PyTorch scatter does not work on short types.
8+ dtypes .remove (torch .bool )
79
810grad_dtypes = [torch .float , torch .double ]
911
You can’t perform that action at this time.
0 commit comments