fixups

NVIDIA · Aug 14, 2024 · 33f6185 · 33f6185
1 parent 5ae63ac
commit 33f6185
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/.github/container/Dockerfile.jax b/.github/container/Dockerfile.jax
@@ -60,7 +60,6 @@ RUN build-jax.sh \
 RUN <<"EOF" bash -ex -o pipefail
 pip install ninja && rm -rf ~/.cache/pip
 # TransformerEngine now needs JAX at build time
-pip install -e ${SRC_PATH_JAX}
 git-clone.sh ${URLREF_TRANSFORMER_ENGINE} ${SRC_PATH_TRANSFORMER_ENGINE}
 pushd ${SRC_PATH_TRANSFORMER_ENGINE}
 python setup.py bdist_wheel && rm -rf build

diff --git a/.github/container/Dockerfile.triton b/.github/container/Dockerfile.triton
@@ -38,6 +38,7 @@ EOF
 RUN <<"EOF" bash -ex
 mkdir /opt/llvm-build
 pushd /opt/llvm-build
+pip install ninja && rm -rf ~/.cache/pip
 cmake -G Ninja \
   -DCMAKE_BUILD_TYPE=Release \
   -DCMAKE_C_COMPILER=clang \