fix three_interpolate bug (#133)

Signed-off-by: Zheng Qin <clavichord93@hotmail.com>
NVIDIAGameWorks · Jan 20, 2020 · d1ca062 · d1ca062
1 parent f1154b2
commit d1ca062
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/kaolin/cuda/three_nn.cpp b/kaolin/cuda/three_nn.cpp
@@ -88,10 +88,10 @@ at::Tensor three_interpolate_grad(at::Tensor grad_out, at::Tensor idx,
       torch::zeros({grad_out.size(0), grad_out.size(1), m},
                    at::device(grad_out.device()).dtype(at::ScalarType::Float));
 
-  three_interpolate_kernel_launcher(grad_out.size(0), grad_out.size(1),
-                                    grad_out.size(2), m, grad_out.data<float>(),
-                                    idx.data<int>(), weight.data<float>(),
-                                    output.data<float>());
+  three_interpolate_grad_kernel_launcher(grad_out.size(0), grad_out.size(1),
+                                         grad_out.size(2), m, grad_out.data<float>(),
+                                         idx.data<int>(), weight.data<float>(),
+                                         output.data<float>());
 
   return output;
 }