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(): ...@@ -539,7 +539,7 @@ if _build_custom_ops():
ext_modules.append(CMakeExtension(name="vllm._C")) ext_modules.append(CMakeExtension(name="vllm._C"))
package_data = { 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: if envs.VLLM_USE_PRECOMPILED:
ext_modules = [] 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