Update gpu atomics include path

[peterbell10]

THC is being removed, and this header now lives in ATen.
See pytorch/pytorch#65470.

[datumbox]

@peterbell10 Thanks for the PR.

The failures on CI seem related:

FAILED: /root/project/build/temp.linux-x86_64-3.6/root/project/torchvision/csrc/ops/hip/ps_roi_pool_kernel.o 
/opt/rocm-4.1.0/bin/hipcc  -DWITH_HIP -I/root/project/torchvision/csrc -I/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include -I/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/torch/csrc/api/include -I/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/TH -I/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/THC -I/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/THH -I/opt/rocm-4.1.0/include -I/opt/rocm-4.1.0/miopen/include -I/root/project/ext_libraries/include -I/opt/python/cp36-cp36m/include/python3.6m -c -c /root/project/torchvision/csrc/ops/hip/ps_roi_pool_kernel.hip -o /root/project/build/temp.linux-x86_64-3.6/root/project/torchvision/csrc/ops/hip/ps_roi_pool_kernel.o -fPIC -D__HIP_PLATFORM_HCC__=1 -DCUDA_HAS_FP16=1 -D__HIP_NO_HALF_OPERATORS__=1 -D__HIP_NO_HALF_CONVERSIONS__=1 -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 --amdgpu-target=gfx803 --amdgpu-target=gfx900 --amdgpu-target=gfx906 --amdgpu-target=gfx908 -fno-gpu-rdc -std=c++14
In file included from /root/project/torchvision/csrc/ops/hip/ps_roi_pool_kernel.hip:5:
/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.h:102:5: warning: 'deviceCount' has a non-throwing exception specification but can still throw [-Wexceptions]
    C10_HIP_CHECK(hipGetDeviceCount(&deviceCnt));
    ^
/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/c10/hip/HIPException.h:44:7: note: expanded from macro 'C10_HIP_CHECK'
      throw c10::HIPError(                                         \
      ^
/opt/python/cp36-cp36m/lib/python3.6/site-packages/torch/include/ATen/hip/impl/HIPGuardImplMasqueradingAsCUDA.h:100:15: note: function declared non-throwing here
  DeviceIndex deviceCount() const noexcept override {
              ^                   ~~~~~~~~
/root/project/torchvision/csrc/ops/hip/ps_roi_pool_kernel.hip:7:10: fatal error: 'ATen/hip/Atomic.cuh' file not found
#include <ATen/hip/Atomic.cuh>
         ^~~~~~~~~~~~~~~~~~~~~
1 warning and 1 error generated when compiling for gfx803.

[peterbell10]

I think we just need to wait for a new nightly release.

[datumbox]

LGTM, thanks!