From f37e1292a1d8cbf52e969a2b72d357ece594e793 Mon Sep 17 00:00:00 2001
From: liqunfu <liqfu@microsoft.com>
Date: Thu, 17 Sep 2020 17:21:47 -0700
Subject: [PATCH] --shm-size=1024m to fix nccl shared memory issue (#5214)

* --shm-size=256m to fix nccl shared memory issue

Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
---
 tools/ci_build/github/linux/run_dockerbuild.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/ci_build/github/linux/run_dockerbuild.sh b/tools/ci_build/github/linux/run_dockerbuild.sh
index f792825aba..4f0f10cf0c 100755
--- a/tools/ci_build/github/linux/run_dockerbuild.sh
+++ b/tools/ci_build/github/linux/run_dockerbuild.sh
@@ -107,6 +107,8 @@ fi
 
 if [ $BUILD_DEVICE = "cpu" ] || [ $BUILD_DEVICE = "ngraph" ] || [ $BUILD_DEVICE = "openvino" ] || [ $BUILD_DEVICE = "nnapi" ] || [ $BUILD_DEVICE = "arm" ]; then
     RUNTIME=
+elif [[ $BUILD_EXTR_PAR = *--enable_training_python_frontend_e2e_tests* ]]; then
+     RUNTIME="--gpus all --shm-size=256m"
 else
     RUNTIME="--gpus all"
 fi