Dixin Zhou

[Fx Importer] fix mutation importer with non persistent buffer

2

A non-persistent buffer will not be a part of this module’s `state_dict`. Hence when setting `experimental_support_mutation=True` and have non-persistent buffer, the current fx importer will fail to retrieve a value...

[Torch] add aten.bilinear op decomposing

7

This PR adds support for `aten.bilinear` op. The `aten.bilinear` op is decomposed to `aten._trilinear` and `aten.add` according to https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/Linear.cpp#L712. Additionally, this PR fixes a bug in `aten._trilinear` op decomposition about...

Given a program `model` that mutates its inputs, ``` class Basic(torch.nn.Module): def forward(self, x): x.mul_(2) return x ``` the fx_importer `fx.export_and_import(model, output_type=output = OutputType.TORCH, experimental_support_mutation=True)` produces the following mlir. ```...