Skip to content

Commit 3317036

Browse files
committed
update timeout
1 parent e005de4 commit 3317036

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/libkernelbot/launchers/github.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,10 @@ async def run_submission( # noqa: C901
105105
logger.info("Waiting for workflow to start...")
106106

107107
timeout = get_timeout(config) + TIMEOUT_BUFFER_MINUTES
108+
# AMD workflows need extra time for PyTorch ROCm installation
109+
# Add 10 more minutes
110+
if gpu_vendor == "AMD":
111+
timeout += 10
108112
logger.info(f"Waiting for workflow to complete... (timeout: {timeout} minutes)")
109113
await run.wait_for_completion(
110114
lambda x: self.wait_callback(x, status), timeout_minutes=timeout

0 commit comments

Comments
 (0)