error in setup.py

[FinallyKiKi]

src/deform_conv_cuda.cpp:559:3: error: ‘AT_CHECK’ was not declared in this scope; did you mean ‘DCHECK’?

  559 |   AT_CHECK(input.is_contiguous(), "input tensor has to be contiguous");
      |   ^~~~~~~~
      |   DCHECK

error: command '/usr/bin/gcc' failed with exit code 1

[FinallyKiKi]

AT_CHECK is deprecated in torch 1.5
replace all ‘AT_CHECK' as ‘TORCH_CHECK’