[torchdynamo] Add default decomposition for ops in the dynamo backend

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
pull/1830/head snapshot-20230123.727
Vivek Khandelwal 2023-01-23 12:22:44 +05:30
parent bf4c77cbc5
commit 23aa6903f7
1 changed files with 9 additions and 0 deletions

View File

@ -52,6 +52,15 @@ def _get_decomposition_table():
aten.native_group_norm,
aten.split.Tensor,
aten.split_with_sizes,
aten.norm.ScalarOpt_dim,
aten.embedding_dense_backward,
aten.native_layer_norm_backward,
aten.slice_backward,
aten.select_backward,
aten.upsample_bilinear2d.vec,
aten.mse_loss_backward,
aten.native_group_norm_backward,
aten.sigmoid_backward,
])