diff --git a/Fuser/dispatch_kernel_agent.py b/Fuser/dispatch_kernel_agent.py index c041b64..167dffe 100644 --- a/Fuser/dispatch_kernel_agent.py +++ b/Fuser/dispatch_kernel_agent.py @@ -456,7 +456,7 @@ def main(argv: list[str] | None = None) -> int: help="Max concurrent subgraphs to dispatch (default: 2); use 'auto' to match subgraph count", ) p.add_argument( - "test-timeout-s", + "--test-timeout-s", type=int, default=30, help="Timeout for each test (default: 30s)", @@ -501,7 +501,7 @@ def main(argv: list[str] | None = None) -> int: jobs=jobs_val, target_platform=args.target_platform, no_cusolver=args.no_cusolver, - test_timeout_s=30, + test_timeout_s=args.test_timeout_s, ) print(str(summary_path)) return 0