Skip to content

Commit 517855b

Browse files
committed
Optimize configuration record in .json
1 parent a5f3171 commit 517855b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graph_net/torch/test_compiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def test_single_model(args):
186186
elif args.compiler == "tensorrt":
187187
result_data["configuration"][
188188
"compiler_framework_version"
189-
] = f"TensorRT {torch_tensorrt.__version__}"
189+
] = f"TensorRT {torch_tensorrt.version}"
190190
else:
191191
result_data["configuration"]["compiler_version"] = "unknown"
192192

0 commit comments

Comments
 (0)