Skip to content

Commit a526af3

Browse files
Gasoonjiafacebook-github-bot
authored andcommitted
mitigate llama2 export issue (#2682)
Summary: Pull Request resolved: #2682 put full like op into verifier to migitate llama2 export issue. Reviewed By: mergennachin, cccclai Differential Revision: D55354975 fbshipit-source-id: a1417b5ed92955be01bb673d52f41225ae573a7f
1 parent 5e23c33 commit a526af3

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

exir/emit/test/test_emit.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -521,8 +521,6 @@ def forward(self, x: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]:
521521
program = to_edge(export(f, x)).to_executorch().executorch_program
522522
self._assertCallLength(program, 0, 8)
523523

524-
# Non contiguous tensors are not supported in ExecuTorch
525-
@unittest.expectedFailure
526524
def test_emit_layout(self) -> None:
527525
class Foo(torch.nn.Module):
528526
def forward(self, x: torch.Tensor) -> torch.Tensor:

exir/verification/verifier.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ def check_valid_op(self, op):
7878
torch.ops.aten.quantize_per_tensor.default,
7979
torch.ops.aten.dequantize.self,
8080
torch.ops.aten.max.default,
81+
torch.ops.aten.full_like.default, # TODO(T183507359)
8182
):
8283
return
8384
if torch.Tag.core not in op.tags and torch.Tag.view_copy not in op.tags:

0 commit comments

Comments
 (0)