Skip to content

Commit

Permalink
Track latest TVM (#680)
Browse files Browse the repository at this point in the history
  • Loading branch information
ashokkumarkannan1 authored Nov 13, 2024
1 parent e9c0c52 commit 42a2b48
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 1 deletion.
7 changes: 7 additions & 0 deletions forge/test/mlir/test_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -1502,6 +1502,13 @@ def forward(self, x):
x = nn.functional.pad(x, self.padding, mode="constant", value=0)
return self.conv(x)

pad_top, pad_bottom, pad_left, pad_right = padding
if pad_top != pad_bottom or pad_left != pad_right:
pytest.xfail(
"TTNN only supports padding height/width attributes. Thus, padding_top "
"must equal padding_bottom for the op to execute as expected."
)

framework_model = PaddingAndConv2d(padding=padding)

inputs = [torch.rand(shape)]
Expand Down
2 changes: 1 addition & 1 deletion third_party/tvm

0 comments on commit 42a2b48

Please sign in to comment.