38 lines
557 B
Bash
Executable File
38 lines
557 B
Bash
Executable File
#!/bin/bash
|
|
set -euo pipefail
|
|
|
|
# Default working directory
|
|
WORKDIR="/sglang-checkout/test/srt"
|
|
ENV_ARGS=(
|
|
-e SGLANG_AMD_CI=1
|
|
-e SGLANG_IS_IN_CI=1
|
|
-e SGLANG_AITER_MOE=1
|
|
)
|
|
|
|
# Parse optional -w/--workdir and -e ENV=VAL flags
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-w|--workdir)
|
|
WORKDIR="$2"
|
|
shift 2
|
|
;;
|
|
-e)
|
|
ENV_ARGS+=("-e" "$2")
|
|
shift 2
|
|
;;
|
|
--)
|
|
shift
|
|
break
|
|
;;
|
|
*)
|
|
break
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Run docker exec
|
|
docker exec \
|
|
-w "$WORKDIR" \
|
|
"${ENV_ARGS[@]}" \
|
|
ci_sglang "$@"
|