Skip to content

Commit 112a8fc

Browse files
authored
Enable numactl install for wide and deep (#293)
Signed-off-by: Abolfazl Shahbazi <[email protected]>
1 parent 39ac094 commit 112a8fc

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

benchmarks/common/tensorflow/start.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,9 @@ if [[ $NUMA_CORES_PER_INSTANCE != "None" || $SOCKET_ID != "-1" || $NUM_CORES !=
166166
elif [[ $MODEL_NAME == "bert_large" && $MODE == "training" && $MPI_NUM_PROCESSES != "None" ]]; then
167167
# BERT large training with MPI uses numactl
168168
INSTALL_NUMACTL="True"
169+
elif [[ $MODEL_NAME == "wide_deep" ]]; then
170+
# TODO: Why Wide & Deep uses numactl always
171+
INSTALL_NUMACTL="True"
169172
fi
170173

171174
# If we are running in a container, call the container_init.sh files

0 commit comments

Comments
 (0)