Unverified Commit 51b7a6d7 authored by VVsssssk's avatar VVsssssk Committed by GitHub
Browse files

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

parent 4826a9b7
......@@ -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],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment