gpu tensor init with coordinate key none coords error, fix #240

NVIDIA · Dec 15, 2020 · d38ad33 · d38ad33
1 parent 0485901
commit d38ad33
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/MinkowskiEngine/MinkowskiTensor.py b/MinkowskiEngine/MinkowskiTensor.py
@@ -264,7 +264,9 @@ def __init__(
         # To device
         if device is not None:
             features = features.to(device)
-            coordinates = coordinates.to(device)
+            if coordinates is not None:
+                # assertion check for the map key done later
+                coordinates = coordinates.to(device)
 
         # Coordinate Management
         self._D = 0  # coordinate size - 1