add debug inf

core/tensorBase.py

@@ -240,7 +240,10 @@ class TensorBase(torch.nn.Module):
         print("aabb", self.aabb.view(-1))
         print("grid size", gridSize)
         self.aabbSize = self.aabb[1] - self.aabb[0]
+        print(self.aabbSize.device)
         self.invaabbSize = 2.0/self.aabbSize
+        print(self.invaabbSize.device)
+        self.invaabbSize = self.invaabbSize.to(self.aabb.device)
         self.gridSize= gridSize.float()
         self.units=self.aabbSize / (self.gridSize-1)
         self.stepSize=torch.mean(self.units)*self.step_ratio  # TBD step_ratio? why so small 0.5