Imported Upstream version 1.25.0

[platform/core/ml/nnfw.git] / infra / command / build-docker-image

diff --git a/infra/command/build-docker-image b/infra/command/build-docker-image
index 9a04438..f4e2069 100644 (file)
--- a/infra/command/build-docker-image
+++ b/infra/command/build-docker-image
@@ -45,6 +45,12 @@ done
 
 DOCKER_IMAGE_NAME=${DOCKER_IMAGE_NAME:-nnfw/one-devtools:$DOCKER_TAG}
 DOCKER_FILE_RPATH=$DOCKER_FILE_RPATH_BASE/$UBUNTU_CODENAME/Dockerfile
+
+HOST_ARCH=$(uname -m)
+if [[ -n $HOST_ARCH  && $HOST_ARCH != "x86_64" ]]; then
+  DOCKER_FILE_RPATH=$DOCKER_FILE_RPATH.$HOST_ARCH
+fi
+
 DOCKER_BUILD_ARGS+=("-t ${DOCKER_IMAGE_NAME}")
 
 docker build --build-arg http_proxy="${http_proxy}" \