Commit 5b62725e authored by zhuwenwen's avatar zhuwenwen
Browse files

add benchmarks to vllm whl

parent 77a13b1b
......@@ -539,7 +539,7 @@ if _build_custom_ops():
ext_modules.append(CMakeExtension(name="vllm._C"))
package_data = {
"vllm": ["py.typed", "model_executor/layers/fused_moe/configs/*.json"]
"vllm": ["py.typed", "model_executor/layers/fused_moe/configs/*.json", "benchmarks/*.py"]
}
if envs.VLLM_USE_PRECOMPILED:
ext_modules = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment