[ONNX] Unsupported FX nodes: {'call_function': ['aten\ .glu.default']}

[titaiwangms]

🐛 Describe the bug

import torch

model = torch.nn.GLU().eval().cuda()
sample_input = (torch.randn(8, 4).cuda(),)

# This works fine                                                                                                           
torch.onnx.export(model, sample_input, "glu_legacy.onnx")

model = torch.export.export(model, sample_input)
                                                   
# torch.onnx._internal.diagnostics.infra.context.RuntimeErrorWithDiagnostic: Unsupported FX nodes: {'call_function': ['aten\
.glu.default']}.                                                                                                                                                                                             
options = torch.onnx.ExportOptions(dynamic_shapes=True)
out = torch.onnx.dynamo_export(model,
                               *sample_input,
                               export_options=options
                               )

Versions

Main

[titaiwangms]

cc @justinchuby