Skip to content

[torchbench] Missing meta function for aten::_cudnn_rnn_flatten_weight #144989

@IvanKobzarev

Description

@IvanKobzarev

🚀 The feature, motivation and pitch

Torchbench for export-aot-inductor for tts_angular model fails due to missing Meta fn.

python benchmarks/dynamo/torchbench.py --only tts_angular --accuracy --no-translation-validation --inference --bfloat16 --export-aot-inductor --disable-cudagraphs --device cuda
NotImplementedError: aten::_cudnn_rnn_flatten_weight: attempted to run this operator with Meta tensors, but there was no fake impl or Meta kernel registered. You may have run into this message while using an operator with PT2 compilation APIs (torch.compile/torch.export); in order to use this operator with those APIs you'll need to add a fake impl. Please see the following for next steps:  https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html
TorchDynamo optimized model failed to run because of following error
fail_to_run

Alternatives

No response

Additional context

No response

cc @chauhang @penguinwu

Metadata

Metadata

Assignees

Labels

oncall: pt2pt2-pass-rate-regressionTrack regression of PT2 dashboard pass ratetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions