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 b885198 commit 73bdb33Copy full SHA for 73bdb33
olive/passes/onnx/onnxscript_fusion.py
@@ -38,6 +38,5 @@ def _run_for_config(
38
# TODO(exporter team): Different fusions support different devices
39
model_ir, function_stats = ort_fusions.optimize_for_ort(model_ir)
40
logger.debug("Function stats: %s", function_stats)
41
- model_proto = ir.to_proto(model_ir)
42
# save the model to the output path and return the model
43
- return model_proto_to_olive_model(model_proto, output_model_path, config)
+ return model_ir_to_olive_model(model_ir, output_model_path)
0 commit comments