49 lines
1.2 KiB
Bash
Executable file
49 lines
1.2 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
cd "$ROOT_DIR"
|
|
|
|
usage() {
|
|
cat <<'EOF'
|
|
usage: tools/flux_rust.sh <command> [options]
|
|
|
|
commands:
|
|
check
|
|
text-smoke
|
|
t5-smoke
|
|
t5-stage-compare
|
|
t5-debug-compare
|
|
transformer-smoke
|
|
transformer-ref-compare
|
|
transformer-ref-stage-compare
|
|
transformer-ref-compare-f32
|
|
native-generate
|
|
warm-bench
|
|
handoff-generate
|
|
|
|
This is the Rust/ggml-only wrapper around tools/flux_debug.sh.
|
|
Pass options such as `--workflow`, `--model-root`, `--width`, `--height`,
|
|
`--steps`, `--compare-cond-dir`, `--cond-dir`, `--native-output`, `--t5-mode`
|
|
after the command.
|
|
EOF
|
|
}
|
|
|
|
main() {
|
|
local command="${1:-}"
|
|
case "$command" in
|
|
check|text-smoke|t5-smoke|t5-stage-compare|t5-debug-compare|transformer-smoke|transformer-ref-compare|transformer-ref-stage-compare|transformer-ref-compare-f32|native-generate|warm-bench|handoff-generate)
|
|
exec tools/flux_debug.sh "$@"
|
|
;;
|
|
-h|--help|help|"")
|
|
usage
|
|
;;
|
|
*)
|
|
printf 'unknown command: %s\n\n' "$command" >&2
|
|
usage >&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
}
|
|
|
|
main "$@"
|