Files
Flux/examples/dense-neural-network/run.sh
T
Bausager bd2edea8ef
Sync public mirror / sync (push) Failing after 24s
p.238 in NNFS
2025-12-23 14:47:40 +01:00

28 lines
626 B
Bash
Executable File

#!/bin/bash
set -e
TARGET="dense-neural-network"
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
ROOT_DIR="$(cd "$SCRIPT_DIR/../.." && pwd)"
BUILD_DIR="$ROOT_DIR/build"
mkdir -p "$BUILD_DIR"
cd "$BUILD_DIR"
cmake .. #-DCMAKE_BUILD_TYPE=Debug
cmake --build . --target "$TARGET"
# Load omp.cfg
if [ -f "$ROOT_DIR/omp.cfg" ]; then
export $(grep -v '^[[:space:]]*#' "$ROOT_DIR/omp.cfg" | grep -v '^[[:space:]]*$' | xargs)
fi
echo "=== CPU / OpenMP info ==="
echo "System cores (nproc): $(nproc)"
echo "OMP_NUM_THREADS=${OMP_NUM_THREADS:-"(not set)"}"
echo "========================="
"$BUILD_DIR/bin/$TARGET"