Skip to content

[torchlib] Implement aten.embedding_bag.padding_idx #2219

Open
@justinchuby

Description

@justinchuby

Model dlrm

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 [96monnx[0m component. Attach the error report and the pt2 model.## Exception summary<class 'AssertionError'>: padding_idx must not be None. This is likely a dispatcher error<class 'torch.onnx._internal.exporter._errors.GraphConstructionError'>: Error when calling function 'TracedOnnxFunction(<function aten_embedding_bag_padding_idx at 0x7f265502ff40>)' with args '[SymbolicTensor('p_model_emb_l_0_weight' type=Tensor(FLOAT16) shape=[100000064] producer=None index=None) SymbolicTensor('args_2_0' type=Tensor(INT64) shape=[100] producer=None index=None) SymbolicTensor('select' type=Tensor(INT64) shape=[1] producer=node_Gather_5 index=0) False 0 True None False None]' and kwargs '{}'<class 'torch.onnx._internal.exporter._errors.ConversionError'>: Error when translating node %embedding_bag : [num_users=1] = call_function[target=torch.ops.aten.embedding_bag.padding_idx](args = (%p_model_emb_l_0_weight %args_2_0 %select False 0 True None False None) kwargs = {}). 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 thisgood first issueGood for newcomersmodule: 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