clear global coords man

MinkowskiEngine/SparseTensor.py

@@ -58,6 +58,16 @@ def sparse_tensor_operation_mode():
     return copy.deepcopy(_sparse_tensor_operation_mode)
 
 
+def clear_global_coords_man():
+    r"""
+    When using the operation mode:
+    `SparseTensorOperationMode.SHARE_COORDS_MANAGER`, you must explicitly clear
+    the coordinate manager when done using it.
+    """
+    global _global_coords_man
+    _global_coords_man = None
+
+
 class SparseTensor():
     r"""A sparse tensor class. Can be accessed via
     :attr:`MinkowskiEngine.SparseTensor`.

MinkowskiEngine/__init__.py

@@ -33,7 +33,7 @@
 import torch
 
 from SparseTensor import SparseTensor, SparseTensorOperationMode, \
-    set_sparse_tensor_operation_mode, sparse_tensor_operation_mode
+    set_sparse_tensor_operation_mode, sparse_tensor_operation_mode, clear_global_coords_man
 
 from Common import RegionType, convert_to_int_tensor, convert_region_type, \
     MinkowskiModuleBase, KernelGenerator, GlobalPoolingMode

examples/sparse_tensor_basic.py

@@ -104,15 +104,19 @@ def operation_mode():
     new_coords = torch.IntTensor([[0, 1], [2, 3], [4, 5]])
     new_feats = torch.rand(len(new_coords), feats.size(1))
 
-    # sparse tensors
-    A = ME.SparseTensor(coords=coords, feats=feats)
-    B = ME.SparseTensor(
-        coords=new_coords,
-        feats=new_feats,
-        #coords_manager=A.coords_man,  No need to feed the coords_man
-        force_creation=True)
+    for _ in range(2):
+        # sparse tensors
+        A = ME.SparseTensor(coords=coords, feats=feats)
+        B = ME.SparseTensor(
+            coords=new_coords,
+            feats=new_feats,
+            # coords_manager=A.coords_man,  No need to feed the coords_man
+            force_creation=True)
 
-    C = A + B
+        C = A + B
+
+        # When done using it for forward and backward, you must cleanup the coords man
+        ME.clear_global_coords_man()
 
 
 if __name__ == '__main__':