Skip to content
Snippets Groups Projects
  • Cao Yuhang's avatar
    e5921ca8
    Fix pytorch1.5 compile error (#2524) · e5921ca8
    Cao Yuhang authored
    * 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
    Fix pytorch1.5 compile error (#2524)
    Cao Yuhang authored
    * 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
Code owners
Assign users and groups as approvers for specific file changes. Learn more.
setup.py 6.13 KiB