-
Notifications
You must be signed in to change notification settings - Fork 27.4k
[torchbench] Missing meta function for aten::_cudnn_rnn_flatten_weight #144989
Copy link
Copy link
Closed
Labels
oncall: pt2pt2-pass-rate-regressionTrack regression of PT2 dashboard pass rateTrack regression of PT2 dashboard pass ratetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
🚀 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
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
oncall: pt2pt2-pass-rate-regressionTrack regression of PT2 dashboard pass rateTrack regression of PT2 dashboard pass ratetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module