Skip to content

Commit ee6a120

Browse files
committed
fix(evo2_megatron): pin warp-lang<1.13.0 in CI and Dockerfile
subquadratic-ops is incompatible with warp 1.13+. Pin warp-lang<1.13.0 in both .ci_build.sh and the Dockerfile until upstream releases a fix. Signed-off-by: svc-bionemo <267129667+svc-bionemo@users.noreply.github.com>
1 parent 5497faa commit ee6a120

2 files changed

Lines changed: 6 additions & 0 deletions

File tree

bionemo-recipes/recipes/evo2_megatron/.ci_build.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,9 @@ uv venv --system-site-packages
1010
# 2. Activate the environment
1111
source .venv/bin/activate
1212

13+
# FIXME: Pin warp-lang<1.13.0 until subquadratic-ops is compatible with warp 1.13+
14+
uv pip install 'warp-lang<1.13.0'
15+
1316
# 3. Install build requirements and pin transformer_engine
1417
pip freeze | grep transformer_engine > pip-constraints.txt
1518
uv pip install -r build_requirements.txt --no-build-isolation

bionemo-recipes/recipes/evo2_megatron/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH"
2121
# We create it one level up (/workspace/.venv) to keep it out of the source dir
2222
RUN uv venv --system-site-packages --seed $VIRTUAL_ENV
2323

24+
# FIXME: Pin warp-lang<1.13.0 until subquadratic-ops is compatible with warp 1.13+
25+
RUN uv pip install 'warp-lang<1.13.0'
26+
2427
# 4. Pin transformer_engine to the version pre-installed in the base image.
2528
# Without this constraint uv may try to upgrade or rebuild TE, which breaks the build.
2629
RUN pip freeze | grep transformer_engine > pip-constraints.txt

0 commit comments

Comments
 (0)