#11649: update tt_lib with ttnn support for non working folder (#11654)

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_backward_sqrt.py

@@ -4,12 +4,12 @@
 
 import torch
 import pytest
-import tt_lib
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import (
     data_gen_with_range,
     compare_pcc,
     data_gen_with_val,
 )
+import ttnn
 
 
 @pytest.mark.parametrize(
@@ -26,7 +26,7 @@ def test_bw_sqrt(input_shapes, device):
 
     pyt_y = torch.sqrt(in_data)
 
-    tt_output_tensor_on_device = tt_lib.tensor.sqrt_bw(grad_tensor, input_tensor)
+    tt_output_tensor_on_device = ttnn.sqrt_bw(grad_tensor, input_tensor)
 
     in_data.retain_grad()
 

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_bfloat8_eltwise_logaddexp.py

@@ -6,7 +6,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_clamp_max_bw.py

@@ -6,8 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
-
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc
 from tests.tt_eager.python_api_testing.sweep_tests import tt_lib_ops, pytorch_ops

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_clamp_min_bw.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_complex_polar.py

@@ -8,7 +8,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_addcdiv_stuck_1.py

@@ -49,7 +49,7 @@ def run_addcdiv(input_shape, dtype, dlayout, buffer_type, output_mem_config, dat
             ttnn.MemoryConfig(ttnn.TensorMemoryLayout.INTERLEAVED, buffer_type[2]),
             dtype[2],
         )
-        t3 = ttnn.addcdiv(t0, t1, t2, scalar, output_mem_config)
+        t3 = ttnn.addcdiv(t0, t1, t2, value=scalar, memory_config=output_mem_config)
 
         y = tt2torch_tensor(t3)
 

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_addcdiv_stuck_2.py

@@ -47,7 +47,7 @@ def run_addcdiv(input_shape, dtype, dlayout, buffer_type, output_mem_config, dat
             ttnn.MemoryConfig(ttnn.TensorMemoryLayout.INTERLEAVED, buffer_type[2]),
             dtype[2],
         )
-        t3 = ttnn.addcdiv(t0, t1, t2, scalar, output_mem_config)
+        t3 = ttnn.addcdiv(t0, t1, t2, value=scalar, memory_config=output_mem_config)
 
         y = tt2torch_tensor(t3)
     except Exception as exc:

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_log_sigmoid.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_logical_and_.py

@@ -6,7 +6,6 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
 import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_ne.py

@@ -6,7 +6,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc
 from tests.tt_eager.python_api_testing.sweep_tests.pytorch_ops import ne as pt_ne

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_recip.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_rpow.py

@@ -6,7 +6,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_rsqrt_in_depth.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc
 from tests.tt_eager.python_api_testing.sweep_tests.tt_lib_ops import eltwise_rsqrt as tt_eltwise_rsqrt

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_scale_mask_softmax_in_place.py

@@ -6,7 +6,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_eltwise_softmax_in_place.py

@@ -5,7 +5,6 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_permute.py

@@ -8,7 +8,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_equal

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_pow.py

@@ -6,7 +6,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_reduce_max_h.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc, comp_equal

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_reduce_sum_h.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc, comp_equal

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_reshape.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops, tt_lib_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_mean_hw.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_std_hw.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc, comp_allclose

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_stats_var_hw.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc, comp_allclose

tests/tt_eager/python_api_testing/non_working_unit_tests/grayskull/test_sum_2.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_and.py

@@ -8,7 +8,6 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_backward_embedding.py

@@ -4,11 +4,11 @@
 
 import torch
 import pytest
-import tt_lib
 from tests.tt_eager.python_api_testing.sweep_tests import (
     comparison_funcs,
 )
 from loguru import logger
+import ttnn
 
 
 @pytest.mark.parametrize(
@@ -40,7 +40,7 @@ def test_embedding_bw(input_shapes, device):
 
     weights_tensor = ttnn.Tensor(weights, ttnn.bfloat16).to(ttnn.ROW_MAJOR_LAYOUT).to(device)
 
-    tt_output_tensor_on_device = tt_lib.tensor.embedding_bw(grad_tensor, input_tensor, weights_tensor)
+    tt_output_tensor_on_device = ttnn.embedding_bw(grad_tensor, input_tensor, weights_tensor)
     tt_output_tensor_a = tt_output_tensor_on_device[0].cpu().to(ttnn.ROW_MAJOR_LAYOUT).to_torch()
 
     weights.retain_grad()

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_backward_fill.py

@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_pt_tt, compare_results
 
 
@@ -29,7 +28,7 @@ def test_bw_fill(input_shapes, device):
     grad_sum = grad_data.sum()
     pyt_y.fill_(grad_sum)
 
-    tt_output_tensor_on_device = tt_lib.tensor.fill_bw(grad_tensor)
+    tt_output_tensor_on_device = ttnn.fill_bw(grad_tensor)
 
     golden_tensor = [pyt_y]
     comp_pass = compare_results(tt_output_tensor_on_device, golden_tensor)

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_clamp_max_bw.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_clamp_min_bw.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_dispatch_mode_transpose_nh.py

@@ -9,7 +9,6 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
 import random
 import ttnn
 

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_isfinite.py

@@ -6,7 +6,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops, tt_lib_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_relu_min.py

@@ -5,7 +5,7 @@
 from loguru import logger
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc
 from tests.tt_eager.python_api_testing.sweep_tests.pytorch_ops import relu_min as pt_relu_min

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_rpow.py

@@ -13,7 +13,7 @@
 import pytest
 import torch
 
-import tt_lib as ttl
+import ttnn
 
 from tt_lib.utils import (
     pad_weight,
@@ -32,7 +32,6 @@ def run_rpow_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config, d
     torch.manual_seed(data_seed)
 
     # Initialize the device
-    tensor = ttl.tensor
     dev = device
 
     test_dims = (input_shape,)
@@ -48,15 +47,15 @@ def run_rpow_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config, d
                 x = x.reshape(-1).tolist()
 
             if in_mem_config == "SYSTEM_MEMORY":
-                ttx = tensor.Tensor(
+                ttx = ttnn.Tensor(
                     x,
                     [N, C, H, W],
                     dtype,
                     dlayout,
                     dev,
                 ).cpu()
             else:
-                ttx = tensor.Tensor(
+                ttx = ttnn.Tensor(
                     x,
                     [N, C, H, W],
                     dtype,
@@ -66,7 +65,7 @@ def run_rpow_tests(input_shape, dtype, dlayout, in_mem_config, out_mem_config, d
                 )
 
             logger.info("Running rpow test")
-            ttz = tensor.rpow(ttx, factor, output_mem_config=out_mem_config)
+            ttz = ttnn.rpow(ttx, factor, memory_config=out_mem_config)
 
             logger.info("Done")
 

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_eltwise_threshold.py

@@ -8,7 +8,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_equal

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_embeddings.py

@@ -6,7 +6,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_pcc

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_isinf.py

@@ -8,7 +8,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_equal

tests/tt_eager/python_api_testing/non_working_unit_tests/wormhole/test_isnan.py

@@ -8,7 +8,7 @@
 import random
 import pytest
 import torch
-import tt_lib as ttl
+import ttnn
 
 from tests.tt_eager.python_api_testing.sweep_tests import pytorch_ops
 from tests.tt_eager.python_api_testing.sweep_tests.comparison_funcs import comp_equal