From ab0fd5b87014de6ba60f74696c1e2f7c6d36ea9d Mon Sep 17 00:00:00 2001 From: zhouxianqi <13165993773@163.com> Date: Mon, 13 May 2024 15:54:52 +0800 Subject: [PATCH] bug_fix_for_op_name_map_default --- profiler/cli/compare_cli.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/profiler/cli/compare_cli.py b/profiler/cli/compare_cli.py index 02470859c..b25fa039d 100644 --- a/profiler/cli/compare_cli.py +++ b/profiler/cli/compare_cli.py @@ -38,7 +38,7 @@ from compare_backend.comparison_generator import ComparisonGenerator @click.option('--enable_communication_compare', is_flag=True) @click.option('--output_path', '-o', 'output_path', type=click.Path()) @click.option('--max_kernel_num', 'max_kernel_num', type=int, help="The number of kernels per torch op is limited.") -@click.option('--op_name_map', type=ast.literal_eval, default={}, +@click.option('--op_name_map', type=ast.literal_eval, default='{}', help="The mapping of operator names equivalent to GPUs and NPUs in the form of dictionaries.", required=False) @click.option('--use_input_shape', is_flag=True) -- Gitee