Skip to content

Commit 3c8c82d

Browse files
authored
[CI] Remove flashinfer cache cleanup to reduce unit test runtime (#7476)
1 parent 5e866e3 commit 3c8c82d

1 file changed

Lines changed: 0 additions & 6 deletions

File tree

tests/layers/test_trtllm_allreduce_rms_fusion.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,6 @@
2121

2222
def test_run_distributed():
2323
"""Launch multi-GPU distributed test via paddle.distributed.launch as subprocess"""
24-
# clearn flashinfer cache directory
25-
flashinfer_cache_dir = os.path.join(os.sep, "root", ".cache", "flashinfer")
26-
if os.path.exists(flashinfer_cache_dir):
27-
print(f"=== Clearing flashinfer cache directory: {flashinfer_cache_dir} ===")
28-
subprocess.run(["rm", "-rf", flashinfer_cache_dir], check=True)
29-
3024
current_dir = os.path.dirname(os.path.abspath(__file__))
3125
run_script = os.path.join(current_dir, "trtllm_allreduce_rms_fusion.py")
3226
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

0 commit comments

Comments
 (0)