mirror of https://github.com/llvm/torch-mlir
38 lines
951 B
Bash
38 lines
951 B
Bash
#!/bin/bash
|
|
# Runs lit-based tests by properly translating paths to the build directory.
|
|
# Example:
|
|
# ./tools/run_lit.sh test/Dialect -v
|
|
set -e
|
|
|
|
td="$(realpath $(dirname $0)/..)"
|
|
build_dir="$td/build"
|
|
install_mlir="$td/install-mlir"
|
|
build_mlir="$td/external/llvm-project/build"
|
|
|
|
lit_exe="$build_mlir/bin/llvm-lit"
|
|
if ! [ -f "$lit_exe" ]; then
|
|
echo "Could not find lit: $lit_exe"
|
|
exit 1
|
|
fi
|
|
|
|
declare -a lit_args
|
|
for i in "$@"; do
|
|
if [[ ${i:0:1} = "-" ]] || [[ ${i:0:1} = "/" ]]; then
|
|
lit_args+=("$i")
|
|
else
|
|
if ! [ -e "$i" ]; then
|
|
echo "Specified lit input does not exist: $i"
|
|
exit 1
|
|
fi
|
|
test_local_path="$(realpath $i)"
|
|
# Replace the src prefix with the build dir.
|
|
test_build_path="$build_dir/${test_local_path##$td/}"
|
|
lit_args+=("$test_build_path")
|
|
fi
|
|
done
|
|
|
|
set -x
|
|
cd $build_dir
|
|
ninja npcomp-opt npcomp-run-mlir NPCOMPCompilerRuntimeShlib NPCOMPNativePyExt
|
|
cd test && python3 "$lit_exe" ${lit_args[@]}
|