[Fix] Fix spconv unit test by releasing cache on GPU (#1872)

open-mmlab · Apr 14, 2022 · 51b7a6d · 51b7a6d
1 parent 4826a9b
commit 51b7a6d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_ops/test_spconv.py b/tests/test_ops/test_spconv.py
@@ -76,6 +76,7 @@ def make_sparse_convmodule(in_channels,
 @pytest.mark.skipif(
     not torch.cuda.is_available(), reason='requires CUDA support')
 def test_make_sparse_convmodule():
+    torch.cuda.empty_cache()
     voxel_features = torch.tensor([[6.56126, 0.9648336, -1.7339306, 0.315],
                                    [6.8162713, -2.480431, -1.3616394, 0.36],
                                    [11.643568, -4.744306, -1.3580885, 0.16],