[Refactor] Share offline profiler argument parser
Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
This commit is contained in:
1 parent
ff806fd4b3
commit
d97bc28c68
4 files changed
+44
-37
No files matched your search
@@ -18,6 +18,6 @@ _EXAMPLES = (
|
||||
def test_offline_examples_use_typed_profiler_config(relative_path: str):
|
||||
source = (_REPO_ROOT / relative_path).read_text(encoding="utf-8")
|
||||
|
||||
assert "--profiler-config" in source
|
||||
assert "add_profiler_config_arg(parser)" in source
|
||||
assert "profiler_enabled = args.profiler_config is not None" in source
|
||||
assert "VLLM_TORCH_PROFILER_DIR" not in source
|
||||
Reference in new issue
Block a user