Unverified Commit d724546c authored by Hongtao Zhang's avatar Hongtao Zhang Committed by GitHub
Browse files

Bugfix: deepseek benchmark names conflict (#725)



Summary
Deepseek currently registers two benchmarks using the same identifier,
causing a naming conflict.

Change
Rename each benchmark so that both have unique, descriptive names.
Co-authored-by: default avatarHongtao Zhang <hongtaozhang@microsoft.com>
parent 44e35cda
......@@ -749,7 +749,7 @@ def _cal_params_count(self):
'megatron-deepseek-v2', MegatronGPT, parameters='--model=deepseek', platform=Platform.ROCM
)
BenchmarkRegistry.register_benchmark(
'megatron-deepseek-v2',
'megatron-deepseek-v2-lite',
MegatronGPT,
parameters=(
'--model=deepseek '
......
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