Skip to content

Commit c5f5d88

Browse files
tugsbayasgalanfacebook-github-bot
authored andcommitted
Support reexportability in torch.export
Summary: Title Differential Revision: D47890879 fbshipit-source-id: 880ed5495e866a0e7ec750cba9bcc69d946191dc
1 parent 1418b15 commit c5f5d88

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

backends/test/hta_partitioner_demo.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@ def forward(self, x_raw, h, c):
5757
input_x = torch.ones([1, 32])
5858
input_h = torch.ones([1, 32])
5959
input_c = torch.ones([1, 32])
60-
60+
# To avoid `operator torch._ops.aten.mkldnn_rnn_layer.default is not Aten Canonical.`
61+
# error
6162
pattern_lstm_conv_lifted = (
6263
exir.capture(
6364
LSTMConvPattern(),

0 commit comments

Comments
 (0)