numa_wrapper.sh

#!/bin/sh

if [ -z "${_VLLM_INTERNAL_NUMACTL_ARGS:-}" ]; then
    echo "_VLLM_INTERNAL_NUMACTL_ARGS is not set" >&2
    exit 1
fi

if [ -z "${_VLLM_INTERNAL_NUMACTL_PYTHON_EXECUTABLE:-}" ]; then
    echo "_VLLM_INTERNAL_NUMACTL_PYTHON_EXECUTABLE is not set" >&2
    exit 1
fi

if ! command -v numactl >/dev/null 2>&1; then
    echo "numactl is not available on PATH" >&2
    exit 1
fi

case "${_VLLM_INTERNAL_NUMACTL_ARGS}" in
    *[![:alnum:]\ \-\_=,./]*)
        echo "Invalid characters in _VLLM_INTERNAL_NUMACTL_ARGS" >&2
        exit 1
        ;;
esac

exec numactl ${_VLLM_INTERNAL_NUMACTL_ARGS} "${_VLLM_INTERNAL_NUMACTL_PYTHON_EXECUTABLE}" "$@"