Skip to content
This repository has been archived by the owner on Dec 11, 2024. It is now read-only.

Commit

Permalink
Fix exit condition check for mlperf-loadgen-build
Browse files Browse the repository at this point in the history
  • Loading branch information
arjunsuresh committed Dec 9, 2024
1 parent c135e8b commit b4c93b0
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions script/get-mlperf-inference-loadgen/run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -25,14 +25,14 @@ cmake \
-DCMAKE_INSTALL_PREFIX="${INSTALL_DIR}" \
"${CM_MLPERF_INFERENCE_SOURCE}/loadgen" \
-DPYTHON_EXECUTABLE:FILEPATH="${CM_PYTHON_BIN_WITH_PATH}" -B .
if [ ${?} -ne 0 ]; then exit $?; fi
test $? -eq 0 || exit $?

echo "******************************************************"
CM_MAKE_CORES=${CM_MAKE_CORES:-${CM_HOST_CPU_TOTAL_CORES}}
CM_MAKE_CORES=${CM_MAKE_CORES:-2}

cmake --build . --target install -j "${CM_MAKE_CORES}"
if [ ${?} -ne 0 ]; then exit $?; fi
test $? -eq 0 || exit $?

# Clean build directory (too large)
cd "${CUR_DIR}"
Expand All @@ -43,8 +43,7 @@ fi

cd "${CM_MLPERF_INFERENCE_SOURCE}/loadgen"
${CM_PYTHON_BIN_WITH_PATH} -m pip install . --target="${MLPERF_INFERENCE_PYTHON_SITE_BASE}"

if [ ${?} -ne 0 ]; then exit $?; fi
test $? -eq 0 || exit $?

# Clean the built wheel
#find . -name 'mlcommons_loadgen*.whl' | xargs rm
Expand Down

0 comments on commit b4c93b0

Please sign in to comment.