From fe64438f28703f4a807398fce2379e14d7f447f3 Mon Sep 17 00:00:00 2001 From: Olli Lupton Date: Fri, 17 May 2024 09:15:23 +0200 Subject: [PATCH] CUDA_MODULE_LOADING: use default lazy loading Prefer to set fewer magic variables. Eager loading cuDNN/cuBLAS during XLA compilation can also be noticably slow. --- .github/container/Dockerfile.jax | 2 -- 1 file changed, 2 deletions(-) diff --git a/.github/container/Dockerfile.jax b/.github/container/Dockerfile.jax index fcbdde518..d9fcb8ef8 100644 --- a/.github/container/Dockerfile.jax +++ b/.github/container/Dockerfile.jax @@ -67,8 +67,6 @@ ENV XLA_FLAGS="${XLA_FLAGS} --xla_gpu_enable_latency_hiding_scheduler=true" ENV XLA_FLAGS="${XLA_FLAGS} --xla_gpu_enable_triton_gemm=false" ENV CUDA_DEVICE_MAX_CONNECTIONS=1 ENV NCCL_NVLS_ENABLE=0 -ENV CUDA_MODULE_LOADING=EAGER - COPY --from=builder ${SRC_PATH_JAX} ${SRC_PATH_JAX} COPY --from=builder ${SRC_PATH_XLA} ${SRC_PATH_XLA}