vllm/tests/compile/piecewise/__init__.py