We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e005de4 commit 3317036Copy full SHA for 3317036
src/libkernelbot/launchers/github.py
@@ -105,6 +105,10 @@ async def run_submission( # noqa: C901
105
logger.info("Waiting for workflow to start...")
106
107
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
112
logger.info(f"Waiting for workflow to complete... (timeout: {timeout} minutes)")
113
await run.wait_for_completion(
114
lambda x: self.wait_callback(x, status), timeout_minutes=timeout
0 commit comments