Skip to content

Commit 9df77ad

Browse files
Cleanup persistent cuBLASLt workspaces before test_mempool_ctx_multithread
Cherry-pick of #3372 Co-authored-by: Dmitry Nikolaev <139769634+dnikolaev-amd@users.noreply.github.com>
1 parent 712584b commit 9df77ad

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

test/test_cuda.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6713,6 +6713,7 @@ def test_mempool_expandable(self):
67136713

67146714
@serialTest()
67156715
def test_mempool_ctx_multithread(self):
6716+
torch._C._cuda_clearCublasWorkspaces()
67166717
torch.cuda.empty_cache()
67176718
segments = torch.cuda.memory._snapshot()["segments"]
67186719
self.assertEqual(len(segments), 0, "Expected empty pool in the beginning")

0 commit comments

Comments
 (0)