remove deprecation warnings

d8a70d93 · rusty1s · 743e46e1 · d8a70d93 · d8a70d93
Commit d8a70d93 authored Feb 05, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

cuda/basis.cpp cuda/basis.cpp +2 -1

cuda/weighting.cpp cuda/weighting.cpp +2 -1

No files found.
--- a/cuda/basis.cpp
+++ b/cuda/basis.cpp
 #include <torch/extension.h>
-#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
+#define CHECK_CUDA(x)                                                          \
+  AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
 std::tuple<at::Tensor, at::Tensor> linear_fw_cuda(at::Tensor pseudo,
                                                  at::Tensor kernel_size,

--- a/cuda/weighting.cpp
+++ b/cuda/weighting.cpp
 #include <torch/extension.h>
-#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be CUDA tensor")
+#define CHECK_CUDA(x)                                                          \
+  AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
 at::Tensor weighting_fw_cuda(at::Tensor x, at::Tensor weight, at::Tensor basis,
                             at::Tensor weight_index);