#9874: Update files

tenstorrent · Jul 9, 2024 · c025307 · c025307
1 parent 134bc89
commit c025307
Show file tree

Hide file tree

Showing 17 changed files with 13 additions and 33 deletions.
diff --git a/docs/source/ttnn/ttnn/ttnn/binary_eq_bw.rst b/docs/source/ttnn/ttnn/ttnn/binary_eq_bw.rst
diff --git a/docs/source/ttnn/ttnn/ttnn/eq_bw.rst b/docs/source/ttnn/ttnn/ttnn/eq_bw.rst
@@ -0,0 +1,6 @@
+.. _ttnn.eq_bw:
+
+ttnn.eq_bw
+#################
+
+.. autofunction:: ttnn.eq_bw
diff --git a/docs/source/ttnn/ttnn/ttnn/unary_eq_bw.rst b/docs/source/ttnn/ttnn/ttnn/unary_eq_bw.rst
diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_addalpha.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_addalpha.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_addcdiv.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_addcdiv.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_assign.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_assign.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 
@@ -50,9 +49,7 @@ def test_bw_binary_assign(input_shapes, device):
     grad_data, grad_tensor = data_gen_with_range(input_shapes, -100, 100, device)
 
     tt_output_tensor_on_device = ttnn.assign_bw(grad_tensor, input_tensor, other_tensor)
-    print(tt_output_tensor_on_device)
-    print(grad_tensor)
-    print(input_tensor, other_tensor)
+
     in_data.retain_grad()
 
     pyt_y = torch.clone(in_data)

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_atan2.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_atan2.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import (
     data_gen_with_range,

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_bias_gelu.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_bias_gelu.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import compare_pcc, data_gen_with_range
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_binary_ge.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_binary_ge.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_binary_gt.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_binary_gt.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_binary_ne.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_binary_ne.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_clamp.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_clamp.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_clamp_min.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_clamp_min.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_eq.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_eq.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_mvlgamma.py b/tests/tt_eager/python_api_testing/unit_testing/backward_ops/test_backward_mvlgamma.py
@@ -4,7 +4,6 @@
 
 import torch
 import pytest
-import tt_lib
 import ttnn
 from tests.tt_eager.python_api_testing.unit_testing.backward_ops.utility_funcs import data_gen_with_range, compare_pcc
 

diff --git a/ttnn/cpp/ttnn/operations/eltwise/unary_backward/device/unary_backward_op.cpp b/ttnn/cpp/ttnn/operations/eltwise/unary_backward/device/unary_backward_op.cpp
@@ -81,7 +81,7 @@ std::vector<Tensor> _add_bw(
     return grad_tensor;
 }
 
-std::vector<Tensor> _unary_comp_bw(const Tensor& grad, const Tensor& input, float other, const MemoryConfig& output_mem_config) {
+std::vector<Tensor> _unary_comp_bw(const Tensor& grad, const MemoryConfig& output_mem_config) {
     std::vector<Tensor> grad_tensor;
     Tensor zero_grad = tt::tt_metal::zeros_like(grad, output_mem_config);
     grad_tensor.emplace_back(zero_grad);
@@ -90,7 +90,7 @@ std::vector<Tensor> _unary_comp_bw(const Tensor& grad, const Tensor& input, floa
 
 std::vector<Tensor> _eq_bw(
     const Tensor& grad, const Tensor& input, float other, const MemoryConfig& output_mem_config) {
-    return _unary_comp_bw(grad, input, other, output_mem_config);
+    return _unary_comp_bw(grad, output_mem_config);
 }
 
 std::function<std::vector<ttnn::Tensor>(const Tensor&, const Tensor&, const MemoryConfig&)> get_function_type1(UnaryBackwardOpType OpType){

diff --git a/ttnn/cpp/ttnn/operations/eltwise/unary_backward/unary_backward_pybind.hpp b/ttnn/cpp/ttnn/operations/eltwise/unary_backward/unary_backward_pybind.hpp
@@ -160,7 +160,7 @@ void py_module(py::module& module) {
     detail::bind_unary_backward(
         module,
         ttnn::mul_bw,
-        R"doc(Performs backward operations for multiply on :attr:`input_tensor`, :attr:`alpha` with given :attr:`grad_tensor`.)doc");
+        R"doc(Performs backward operations for multiply on :attr:`input_tensor`, :attr:`alpha` or attr:`input_tensor_a`, attr:`input_tensor_b` with given :attr:`grad_tensor`.)doc");
 
     detail::bind_unary_backward(
         module,
@@ -175,7 +175,7 @@ void py_module(py::module& module) {
     detail::bind_unary_backward(
         module,
         ttnn::assign_bw,
-        R"doc(Performs backward operations for assign on :attr:`input_tensor` with given :attr:`grad_tensor`.)doc");
+        R"doc(Performs backward operations for assign on :attr:`input_tensor` or attr:`input_tensor_a`, attr:`input_tensor_b` with given :attr:`grad_tensor`.)doc");
 
     detail::bind_unary_backward(
         module,
@@ -187,12 +187,12 @@ void py_module(py::module& module) {
     detail::bind_unary_backward(
         module,
         ttnn::add_bw,
-        R"doc(Performs backward operations for addition on :attr:`input_tensor`, :attr:`alpha` with given :attr:`grad_tensor`.)doc");
+        R"doc(Performs backward operations for addition on :attr:`input_tensor`, :attr:`alpha` or attr:`input_tensor_a`, attr:`input_tensor_b` with given :attr:`grad_tensor`.)doc");
 
     detail::bind_unary_backward(
         module,
         ttnn::eq_bw,
-        R"doc(Performs backward operations for equal to comparison on :attr:`input_tensor`, :attr:`alpha` or attr:`input_tensor_b` with given :attr:`grad_tensor`.
+        R"doc(Performs backward operations for equal to comparison on :attr:`input_tensor`, :attr:`alpha` or attr:`input_tensor_a`, attr:`input_tensor_b` with given :attr:`grad_tensor`.
         Returns an tensor of zeros like input tensors.)doc");
 
 }