Skip to content

Commit 0dbaf3e

Browse files
committed
fixes
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 24e173f commit 0dbaf3e

File tree

2 files changed

+0
-4
lines changed

2 files changed

+0
-4
lines changed

python/test/unit/runtime/test_cache.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import importlib.util
22
import itertools
33
import os
4-
import shutil
54
import pathlib
65

76
import pytest
@@ -495,7 +494,6 @@ def cache_hook(*args, **kwargs):
495494
assert specialization_data is not None
496495

497496
# clear the cache
498-
shutil.rmtree(fresh_triton_cache)
499497
kernel_add.device_caches[device][0].clear()
500498

501499
# preload the kernel

python/test/unit/runtime/test_subproc.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import multiprocessing
2-
import shutil
32

43
import triton
54
import triton.language as tl
@@ -87,7 +86,6 @@ def test_compile_in_forked_subproc_with_forced_gc(fresh_triton_cache) -> None:
8786
compile_empty_kernel_with_gc()
8887

8988
# stage 2.p
90-
shutil.rmtree(fresh_triton_cache)
9189
mp_ctx = multiprocessing.get_context(start_method)
9290
proc = mp_ctx.Process(target=compile_empty_kernel_with_gc)
9391

0 commit comments

Comments
 (0)