Skip to content

Commit bb5f2ee

Browse files
author
Mark Saroufim
committed
Update AMD_REQUIREMENTS to use ROCm 7.2 nightly index
1 parent bdc4523 commit bb5f2ee

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/libkernelbot/consts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ class RankCriterion(Enum):
142142
]
143143
MODAL_CUDA_INCLUDE_DIRS = ["/ThunderKittens/include"]
144144

145-
DEFAULT_GITHUB_TIMEOUT_MINUTES = 10 # Default timeout for GitHub launcher in minutes
145+
DEFAULT_GITHUB_TIMEOUT_MINUTES = 30 # Default timeout for GitHub launcher in minutes
146146

147147
NVIDIA_REQUIREMENTS = """
148148
numpy
@@ -153,7 +153,7 @@ class RankCriterion(Enum):
153153
"""
154154

155155
AMD_REQUIREMENTS = """
156-
--index-url https://download.pytorch.org/whl/rocm6.2.4
156+
--index-url https://download.pytorch.org/whl/nightly/rocm7.2
157157
torch
158158
numpy
159159
"""

0 commit comments

Comments
 (0)