diff --git a/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst b/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst index c8e6ee480bb..1e15cba376f 100644 --- a/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst +++ b/circle-mlir/circle-mlir/tools-test/circle-impexp-test/test.lst @@ -34,6 +34,7 @@ AddModel(Conv2d_F32_R4) AddModel(Conv2d_F32_R4_nobias) AddModel(Conv2d_F32_R4_k33) AddModel(Conv2d_F32_R4_p11) +# AddModel(Conv2d_F32_R4_p11_g2) AddModel(Conv2d_F32_R4_p10) AddModel(Conv2d_F32_R4_g2) AddModel(Conv2d_F32_R4_g5) diff --git a/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst b/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst index a219120f000..e162da09082 100644 --- a/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst +++ b/circle-mlir/circle-mlir/tools-test/onnx2circle-models/test.lst @@ -34,6 +34,7 @@ AddModel(Conv2d_F32_R4) AddModel(Conv2d_F32_R4_nobias) AddModel(Conv2d_F32_R4_k33) AddModel(Conv2d_F32_R4_p11) +# AddModel(Conv2d_F32_R4_p11_g2) AddModel(Conv2d_F32_R4_p10) AddModel(Conv2d_F32_R4_g2) AddModel(Conv2d_F32_R4_g4_1) diff --git a/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst b/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst index d7e2f8551d1..6a809c9558e 100644 --- a/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst +++ b/circle-mlir/circle-mlir/tools-test/onnx2circle-value-test/test.lst @@ -34,6 +34,7 @@ AddModel(Conv2d_F32_R4) AddModel(Conv2d_F32_R4_nobias) AddModel(Conv2d_F32_R4_k33) AddModel(Conv2d_F32_R4_p11) +# AddModel(Conv2d_F32_R4_p11_g2) AddModel(Conv2d_F32_R4_p10) AddModel(Conv2d_F32_R4_g2) AddModel(Conv2d_F32_R4_g4_1) diff --git a/circle-mlir/models/unit/Conv2d_F32_R4_p11_g2/__init__.py b/circle-mlir/models/unit/Conv2d_F32_R4_p11_g2/__init__.py new file mode 100644 index 00000000000..4b33b772574 --- /dev/null +++ b/circle-mlir/models/unit/Conv2d_F32_R4_p11_g2/__init__.py @@ -0,0 +1,20 @@ +import torch + + +# Generate Conv2d operator with Float32, Rank-4 with groups +class net_Conv2d(torch.nn.Module): + def __init__(self): + super().__init__() + self.op = torch.nn.Conv2d(4, 4, 1, padding=(1, 1), groups=2) + + def forward(self, input): + return self.op(input) + + def onnx_opset_version(self): + # TODO set to appropriate value + return 14 + + +_model_ = net_Conv2d() + +_inputs_ = torch.randn(1, 4, 3, 3)