Files
mi300a-xnack/run.sh
2025-01-08 08:22:09 -08:00

59 lines
1.4 KiB
Bash
Executable File

#! /bin/bash
set -eou pipefail
if ! module is-loaded rocm; then
echo module load rocm
exit 1
fi
echo "==================="
echo "= HSA_XNACK unset ="
echo "==================="
unset HSA_XNACK
rocminfo | grep xnack
echo "==================="
echo "= HSA_XNACK=1 ="
echo "==================="
export HSA_XNACK=1
rocminfo | grep xnack
echo "==================="
echo "= HSA_XNACK=0 ="
echo "==================="
export HSA_XNACK=0
rocminfo | grep xnack
set +eou pipefail
for arch in "" "--offload-arch=gfx942:xnack+" "--offload-arch=gfx942:xnack-" "--offload-arch=gfx942"; do
rm -rf "build-$arch"
cmake -S . -B "build-$arch" -DCMAKE_CXX_COMPILER=hipcc -DCMAKE_BUILD_TYPE=Debug -DCMAKE_CXX_FLAGS="-Wall -Wextra $arch"
VERBOSE=1 make -C "build-$arch"
echo "================================"
echo "= HSA_XNACK unset $arch ="
echo "================================"
unset HSA_XNACK
ctest --test-dir "build-$arch"
"build-$arch"/can_map_host_memory
"build-$arch"/gcn_arch_name
echo "================================"
echo "= HSA_XNACK=1 $arch ="
echo "================================"
export HSA_XNACK=1
ctest --test-dir "build-$arch"
"build-$arch"/can_map_host_memory
"build-$arch"/gcn_arch_name
echo "================================"
echo "= HSA_XNACK=0 $arch ="
echo "================================"
export HSA_XNACK=0
ctest --test-dir "build-$arch"
"build-$arch"/can_map_host_memory
"build-$arch"/gcn_arch_name
done;