Skip to content

Commit fc004d4

Browse files
authored
fix: Fix TRT-LLM container build when using a custom pip wheel (#1825)
1 parent 3c6fc6f commit fc004d4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

container/build.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,13 +424,15 @@ else
424424
fi
425425
fi
426426

427-
cd "$NIXL_DIR" || exit
427+
pushd "$NIXL_DIR" || exit
428428
if ! git checkout ${NIXL_COMMIT}; then
429429
echo "ERROR: Failed to checkout NIXL commit ${NIXL_COMMIT}. The cached directory may be out of date."
430430
echo "Please delete $NIXL_DIR and re-run the build script."
431431
exit 1
432432
fi
433433

434+
popd
435+
434436
BUILD_CONTEXT_ARG+=" --build-context nixl=$NIXL_DIR"
435437

436438
# Add NIXL_COMMIT as a build argument to enable caching

0 commit comments

Comments
 (0)