Remove guards of aten.full (#614)

Co-authored-by: Artem Yerofieiev <[email protected]>

Author: jerrysky3

tests/lowering/creation/test_full.py

@@ -1,7 +1,6 @@
 import torch
 import torch_ttnn
 import pytest
-import ttnn
 
 
 class FullModule(torch.nn.Module):
@@ -14,7 +13,11 @@ def forward(self, size, fill_value):
 
 @pytest.mark.parametrize(
     "input_shapes",
-    [[(64, 128)]],
+    [
+        [(64, 128)],
+        [(19, 19)],
+        [(59, 59)],
+    ],
 )
 def test_full(device, input_shapes):
     m = FullModule()
@@ -29,6 +32,7 @@ def test_full(device, input_shapes):
 
     # Check the graph has be rewritten and contain ttnn ops
     nodes = list(option._out_fx_graphs[0].nodes)
-    assert [node.target for node in nodes].count(ttnn.full) == 1
+    # Check the graph has be rewritten and aten ops are replaced
+    assert not any(node.target == torch.ops.aten.full.default for node in nodes)
     # Check inference result
     assert torch.allclose(result_before, result_after)

torch_ttnn/passes/lowering/to_tt_guard_autogen.py

@@ -62,46 +62,6 @@
     ["Tensor<[1, 16, 1, 60]> self = ?", "Tensor<[]> other = ?"],
 ]
 aten__log_softmax_default_blocklist = [["Tensor<[19, 256008]> self = ?", "int dim = 1", "bool half_to_float = False"]]
-aten_full_default_blocklist = [
-    [
-        "List[int] size = [19, 19]",
-        "number fill_value = -3.4028234663852886e+38",
-        "Optional[Device] device = cpu",
-        "Optional[bool] pin_memory = False",
-    ],
-    [
-        "List[int] size = [7, 7]",
-        "number fill_value = -3.3895313892515355e+38",
-        "Optional[Device] device = cpu",
-        "Optional[bool] pin_memory = False",
-    ],
-    [
-        "List[int] size = [45, 45]",
-        "number fill_value = -3.3895313892515355e+38",
-        "Optional[Device] device = cpu",
-        "Optional[bool] pin_memory = False",
-    ],
-    [
-        "List[int] size = [59, 59]",
-        "number fill_value = -3.3895313892515355e+38",
-        "Optional[Device] device = cpu",
-        "Optional[bool] pin_memory = False",
-    ],
-    [
-        "List[int] size = [19, 19]",
-        "number fill_value = -3.3895313892515355e+38",
-        "Optional[Device] device = cpu",
-        "Optional[bool] pin_memory = False",
-    ],
-]
-# TODO(#615): Dynamic shape is not supported yet
-aten_full_like_default_blocklist = [
-    [
-        "Tensor<[s0 + 1, s0 + 1]> self = ?",
-        "number fill_value = 31",
-        "Optional[bool] pin_memory = False",
-    ],
-]
 aten__scaled_dot_product_flash_attention_default_blocklist = [
     ["Tensor<[1, 16, 197, 64]> query = ?", "Tensor<[1, 16, 197, 64]> key = ?", "Tensor<[1, 16, 197, 64]> value = ?"],
     ["Tensor<[1, 12, 197, 64]> query = ?", "Tensor<[1, 12, 197, 64]> key = ?", "Tensor<[1, 12, 197, 64]> value = ?"],
@@ -1402,8 +1362,6 @@ def guard_aten(blocklist, node):
     torch.ops.aten.clamp.default: partial(guard_aten, aten_clamp_default_blocklist),
     torch.ops.aten.maximum.default: partial(guard_aten, aten_maximum_default_blocklist),
     torch.ops.aten._log_softmax.default: partial(guard_aten, aten__log_softmax_default_blocklist),
-    torch.ops.aten.full.default: partial(guard_aten, aten_full_default_blocklist),
-    torch.ops.aten.full_like.default: partial(guard_aten, aten_full_like_default_blocklist),
     torch.ops.aten._scaled_dot_product_flash_attention.default: partial(
         guard_aten, aten__scaled_dot_product_flash_attention_default_blocklist
     ),
@@ -1430,8 +1388,6 @@ def guard_aten(blocklist, node):
     "torch.ops.aten.clamp.default",
     "torch.ops.aten.maximum.default",
     "torch.ops.aten._log_softmax.default",
-    "torch.ops.aten.full.default",
-    "torch.ops.aten.full_like.default",
     "torch.ops.aten.rsub.Scalar",
     "torch.ops.aten._scaled_dot_product_flash_attention.default",
     "torch.ops.aten.transpose.int",