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
Showing
- .travis.yml 12 additions, 4 deletions.travis.yml
- mmdet/models/mask_heads/fcn_mask_head.py 2 additions, 2 deletionsmmdet/models/mask_heads/fcn_mask_head.py
- mmdet/ops/affine_grid/__init__.py 0 additions, 3 deletionsmmdet/ops/affine_grid/__init__.py
- mmdet/ops/affine_grid/affine_grid.py 0 additions, 68 deletionsmmdet/ops/affine_grid/affine_grid.py
- mmdet/ops/affine_grid/src/affine_grid_ext.cpp 0 additions, 23 deletionsmmdet/ops/affine_grid/src/affine_grid_ext.cpp
- mmdet/ops/affine_grid/src/cpu/affine_grid_cpu.cpp 0 additions, 108 deletionsmmdet/ops/affine_grid/src/cpu/affine_grid_cpu.cpp
- mmdet/ops/carafe/src/carafe_ext.cpp 2 additions, 2 deletionsmmdet/ops/carafe/src/carafe_ext.cpp
- mmdet/ops/carafe/src/carafe_naive_ext.cpp 2 additions, 2 deletionsmmdet/ops/carafe/src/carafe_naive_ext.cpp
- mmdet/ops/carafe/src/cuda/carafe_cuda.cpp 2 additions, 2 deletionsmmdet/ops/carafe/src/cuda/carafe_cuda.cpp
- mmdet/ops/carafe/src/cuda/carafe_cuda_kernel.cu 33 additions, 33 deletionsmmdet/ops/carafe/src/cuda/carafe_cuda_kernel.cu
- mmdet/ops/carafe/src/cuda/carafe_naive_cuda.cpp 2 additions, 2 deletionsmmdet/ops/carafe/src/cuda/carafe_naive_cuda.cpp
- mmdet/ops/carafe/src/cuda/carafe_naive_cuda_kernel.cu 10 additions, 10 deletionsmmdet/ops/carafe/src/cuda/carafe_naive_cuda_kernel.cu
- mmdet/ops/dcn/src/cuda/deform_conv_cuda.cpp 21 additions, 21 deletionsmmdet/ops/dcn/src/cuda/deform_conv_cuda.cpp
- mmdet/ops/dcn/src/cuda/deform_conv_cuda_kernel.cu 24 additions, 24 deletionsmmdet/ops/dcn/src/cuda/deform_conv_cuda_kernel.cu
- mmdet/ops/dcn/src/cuda/deform_pool_cuda.cpp 3 additions, 3 deletionsmmdet/ops/dcn/src/cuda/deform_pool_cuda.cpp
- mmdet/ops/dcn/src/cuda/deform_pool_cuda_kernel.cu 12 additions, 12 deletionsmmdet/ops/dcn/src/cuda/deform_pool_cuda_kernel.cu
- mmdet/ops/dcn/src/deform_conv_ext.cpp 5 additions, 5 deletionsmmdet/ops/dcn/src/deform_conv_ext.cpp
- mmdet/ops/dcn/src/deform_pool_ext.cpp 2 additions, 2 deletionsmmdet/ops/dcn/src/deform_pool_ext.cpp
- mmdet/ops/grid_sampler/__init__.py 0 additions, 3 deletionsmmdet/ops/grid_sampler/__init__.py
- mmdet/ops/grid_sampler/grid_sampler.py 0 additions, 100 deletionsmmdet/ops/grid_sampler/grid_sampler.py
Loading
Please register or sign in to comment