Justin Ngo

`fx.export_and_import` hangs

8

I have a simple program: ``` class Conv2D(torch.nn.Module): def __init__( self, kernel_size=3, in_channels=8, out_channels=16, stride=1, padding=0, dilation=1, bias=True, ): super().__init__() self.conv = torch.nn.Conv2d( in_channels=in_channels, out_channels=out_channels, kernel_size=kernel_size, stride=stride, padding=padding, dilation=dilation, bias=bias,...

`BertModule` e2e test timed out because of TOSA's `TosaLayerwiseConstantFoldPass`. This is because it takes too long for this pass to go through and fold all constant layerwise ops in the...

Failed to legalize operation 'bufferization.dealloc' that was explicitly marked illegal

5

When run e2e tests for `tosa` and `onnx_tosa` configs, a lot of tests unexpectedly failed with the following error: ```error: failed to legalize operation 'bufferization.dealloc' that was explicitly marked illegal```...

[TOSA] Update TOSA's rounding mode, nan propagation and resize mode to enums

1

Context: TOSA's rounding mode, nan propagation, and resize mode have been updated from string attributes to enums. This commit updates the Torch to TOSA path to align with those changes.