diff --git a/profiler/cli/compare_cli.py b/profiler/cli/compare_cli.py index 02470859ce2f3a6da70bda718cfee5bae8a5063d..b25fa039daa5f1ca4bd2b6c2d9becee8bc9abc2d 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)