Skip to content
Snippets Groups Projects
Unverified Commit e5921ca8 authored by Cao Yuhang's avatar Cao Yuhang Committed by GitHub
Browse files

Fix pytorch1.5 compile error (#2524)

* replace AT_CHECK with TORCH_CHECK to support pytorch1.5

* fix compile warning, delete grid sample op

* fix ci pytorch version

* delete python3.5

* update version of pytorch and torchvision in ci

* change is_cuda to device().is_cuda()

* specify TORCH_CUDA_ARCH_LIST=6.0,7.0

* fix typo

* delete affine grid

* fix typos

* fix setup.py

* remove redundant comment
parent 2e802ce3
No related branches found
No related tags found
No related merge requests found
Showing
with 132 additions and 429 deletions
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment