We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a126f4 commit 1e237feCopy full SHA for 1e237fe
graph_net/torch/test_compiler.py
@@ -229,6 +229,10 @@ def test_single_model(args):
229
result_data["configuration"][
230
"compile_framework_version"
231
] = f"TensorRT {torch_tensorrt.version}"
232
+ elif args.compiler == "bladedisc":
233
+ result_data["configuration"][
234
+ "compile_framework_version"
235
+ ] = f"BladeDISC {torch_blade.version}"
236
else:
237
result_data["configuration"]["compiler_version"] = "unknown"
238
0 commit comments