Skip to content

Commit 587cb34

Browse files
committed
fix timeout
Signed-off-by: Jack Luar <[email protected]>
1 parent d27b280 commit 587cb34

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/AutoTuner/src/autotuner/distributed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -514,7 +514,7 @@ def parse_arguments():
514514

515515
# Calculate timeout based on cpu_budget
516516
if args.cpu_budget is not None:
517-
args.timeout = args.cpu_budget / os.cpu_count()
517+
args.timeout = round(args.cpu_budget / os.cpu_count() * 3600)
518518
template = calculate_expected_numbers(args.timeout, args.samples)
519519
print(template)
520520
if not args.yes:

0 commit comments

Comments
 (0)