File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments