From 9728a7c92851c11646091db14e71f73adf3fb7c5 Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Sat, 19 Oct 2024 14:18:02 -0700 Subject: [PATCH] test --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index e54762ca63..544af663fc 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,6 +2,7 @@ ARG BASE_IMAGE FROM $BASE_IMAGE # Install TransformerEngine +ARG TE_COMMIT RUN NVTE_FRAMEWORK=pytorch CMAKE_BUILD_PARALLEL_LEVEL=4 MAX_JOBS=4 pip install git+https://github.com/NVIDIA/TransformerEngine.git@$TE_COMMIT ARG BRANCH_NAME