Skip to content

Commit

Permalink
Ignore error when cleaning cache
Browse files Browse the repository at this point in the history
  • Loading branch information
pbchekin committed Dec 17, 2024
1 parent 41437e9 commit fb95dc5
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion python/test/unit/runtime/test_cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -513,7 +513,7 @@ def cache_hook(*args, **kwargs):
assert specialization_data is not None

# clear the cache
shutil.rmtree(fresh_triton_cache)
shutil.rmtree(fresh_triton_cache, ignore_errors=True)
kernel_add.device_caches[device][0].clear()

# preload the kernel
Expand Down
2 changes: 1 addition & 1 deletion python/test/unit/runtime/test_subproc.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def test_compile_in_forked_subproc_with_forced_gc(fresh_triton_cache) -> None:
compile_empty_kernel_with_gc(config)

# stage 2.p
shutil.rmtree(fresh_triton_cache)
shutil.rmtree(fresh_triton_cache, ignore_errors=True)
mp_ctx = multiprocessing.get_context(start_method)
proc = mp_ctx.Process(target=compile_empty_kernel_with_gc, args=(config, ))

Expand Down

0 comments on commit fb95dc5

Please sign in to comment.