Skip to content

[torchlib] aten_div needs rounding mode #2144

Closed
@justinchuby

Description

@justinchuby

ERR_export: Failed to convert the exported program to an ONNX model. [96mThis is step 3/3[0m of exporting the model to ONNX. Next steps:- If there is a missing ONNX function implement it and register it to the registry.- If there is an internal error during ONNX conversion debug the error and summit a PR to PyTorch.- Create an error report with torch.onnx.export(... report=True) and save the ExportedProgram as a pt2 file. Create an issue in the PyTorch GitHub repository against the [96m*onnx*[0m component. Attach the error report and the pt2 model.## Exception summary<class 'TypeError'>: aten_div() got an unexpected keyword argument 'rounding_mode'<class 'torch.onnx._internal.exporter._errors.GraphConstructionError'>: Error when calling function 'TracedOnnxFunction(<function aten_div at 0x7fd48b5220e0>)' with args '[SymbolicTensor('_to_copy_2' type=Tensor(FLOAT) shape=[32] producer=node_Cast_131 index=0) SymbolicTensor('scalar_tensor_default' type=Tensor(FLOAT) shape=[] producer=node_Cast_132 index=0)]' and kwargs '{'rounding_mode': 'floor'}'<class 'torch.onnx._internal.exporter._errors.ConversionError'>: Error when translating node %div_2 : [num_users=1] = call_function[target=torch.ops.aten.div.Tensor_mode](args = (%_to_copy_2 %scalar_tensor_default) kwargs = {rounding_mode: floor}). See the stack trace for more information.(Refer to the full stack trace above for more information.)

Metadata

Metadata

Assignees

Labels

contribution welcomeWe welcome code contributions for thismodule: torchlibRelated to the torch/aten function lib in development

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions