fix - change runfiles

Author: Nancheng-11

rtp_llm/test/utils/BUILD

@@ -27,7 +27,10 @@ py_binary(
     name = "gpu_lock",
     deps = [":torch"],
     main = "device_resource.py",
-    srcs = ["device_resource.py"],
+    srcs = [
+        "device_resource.py",
+        "jit_sys_path_setup.py",
+    ],
 )
 
 py_library(

rtp_llm/test/utils/bootstrap_runner.py

@@ -128,23 +128,10 @@ def __exit__(self, *args: Any):
 
         sys.exit(result.returncode)
     else:
-        if "H20" in cuda_info[0]:
-            # Setup JIT cache and create Bootstrap script
-            try:
-                from jit_sys_path_setup import setup_jit_cache_and_create_bootstrap
-
-                bootstrap_script = setup_jit_cache_and_create_bootstrap()
-            except Exception as e:
-                logging.warning(f"JIT setup failed: {e}, will run without Bootstrap")
-                bootstrap_script = None
-
-        # Prepare test command
-        if bootstrap_script:
-            # Use Bootstrap wrapper
-            test_command = [sys.executable, bootstrap_script] + sys.argv[1:]
-        else:
-            # Fallback to direct execution
-            test_command = sys.argv[1:]
+        from jit_sys_path_setup import setup_jit_cache
+
+        setup_jit_cache()
+
         device_name, _ = cuda_info
         require_count = int(
             os.environ.get("WORLD_SIZE", os.environ.get("GPU_COUNT", "1"))
@@ -155,11 +142,6 @@ def __exit__(self, *args: Any):
             else:
                 env_name = "CUDA_VISIBLE_DEVICES"
             os.environ[env_name] = ",".join(gpu_resource.gpu_ids)
-            result = subprocess.run(test_command)
+            result = subprocess.run(sys.argv[1:])
             logging.info("exitcode: %d", result.returncode)
-
-            # Cleanup Bootstrap script
-            if bootstrap_script and os.path.exists(bootstrap_script):
-                os.unlink(bootstrap_script)
-
             sys.exit(result.returncode)