diff --git a/tests/layers/test_trtllm_allreduce_rms_fusion.py b/tests/layers/test_trtllm_allreduce_rms_fusion.py index 5e704a51d1d..31080b7745c 100644 --- a/tests/layers/test_trtllm_allreduce_rms_fusion.py +++ b/tests/layers/test_trtllm_allreduce_rms_fusion.py @@ -21,12 +21,6 @@ def test_run_distributed(): """Launch multi-GPU distributed test via paddle.distributed.launch as subprocess""" - # clearn flashinfer cache directory - flashinfer_cache_dir = os.path.join(os.sep, "root", ".cache", "flashinfer") - if os.path.exists(flashinfer_cache_dir): - print(f"=== Clearing flashinfer cache directory: {flashinfer_cache_dir} ===") - subprocess.run(["rm", "-rf", flashinfer_cache_dir], check=True) - current_dir = os.path.dirname(os.path.abspath(__file__)) run_script = os.path.join(current_dir, "trtllm_allreduce_rms_fusion.py") os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"