Skip to content

Commit fb05135

Browse files
committed
Optimize configuration record in .json
1 parent a9c71ab commit fb05135

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

graph_net/torch/test_compiler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def test_single_model(args):
162162
"device": args.device,
163163
"hardware": None,
164164
"compiler": args.compiler,
165-
"compiler_framework_version": None,
165+
"compile_framework_version": None,
166166
"warmup": args.warmup,
167167
"trials": args.trials,
168168
},
@@ -182,10 +182,10 @@ def test_single_model(args):
182182
result_data["configuration"]["hardware"] = "unknown"
183183

184184
if args.compiler == "inductor":
185-
result_data["configuration"]["compiler_framework_version"] = torch.__version__
185+
result_data["configuration"]["compile_framework_version"] = torch.__version__
186186
elif args.compiler == "tensorrt":
187187
result_data["configuration"][
188-
"compiler_framework_version"
188+
"compile_framework_version"
189189
] = f"TensorRT {torch_tensorrt.version}"
190190
else:
191191
result_data["configuration"]["compiler_version"] = "unknown"

0 commit comments

Comments
 (0)