Revert "Extend the capabilities of the graph tooling to include extra…

… information of the ttnn operations (#18380)" This reverts commit a33d64a, which broke ttnn unit tests 1.
tenstorrent · Mar 6, 2025 · 87ab998 · 87ab998
1 parent 0c0def2
commit 87ab998
Show file tree

Hide file tree

Showing 15 changed files with 22 additions and 713 deletions.
diff --git a/tests/ttnn/unit_tests/gtests/CMakeLists.txt b/tests/ttnn/unit_tests/gtests/CMakeLists.txt
@@ -5,8 +5,6 @@ set(TTNN_UNIT_TESTS_SRC
     ${CMAKE_CURRENT_SOURCE_DIR}/test_async_runtime.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/test_multiprod_queue.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/test_multi_cq_multi_dev.cpp
-    ${CMAKE_CURRENT_SOURCE_DIR}/test_graph_capture_arguments_morehdot.cpp
-    ${CMAKE_CURRENT_SOURCE_DIR}/test_graph_capture_arguments_transpose.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/test_graph_query_op_constraints.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/test_graph_query_op_runtime.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/test_reflect.cpp

diff --git a/tests/ttnn/unit_tests/gtests/test_graph_capture_arguments_morehdot.cpp b/tests/ttnn/unit_tests/gtests/test_graph_capture_arguments_morehdot.cpp
diff --git a/tests/ttnn/unit_tests/gtests/test_graph_capture_arguments_transpose.cpp b/tests/ttnn/unit_tests/gtests/test_graph_capture_arguments_transpose.cpp
diff --git a/tests/ttnn/unit_tests/gtests/ttnn_test_fixtures.hpp b/tests/ttnn/unit_tests/gtests/ttnn_test_fixtures.hpp
@@ -13,8 +13,6 @@
 
 #include "ttnn/device.hpp"
 #include "ttnn/types.hpp"
-#include "ttnn/tensor/tensor.hpp"
-#include "ttnn/tensor/tensor_impl.hpp"
 #include "tests/tt_metal/test_utils/env_vars.hpp"
 #include <tt-metalium/host_api.hpp>
 #include "hostdevcommon/common_values.hpp"
@@ -55,26 +53,6 @@ class TTNNFixtureWithDevice : public TTNNFixture {
     tt::tt_metal::IDevice& getDevice() { return *device_; }
 };
 
-struct CreateTensorParameters {
-    ttnn::Shape input_shape;
-    DataType dtype;
-    Layout layout;
-    MemoryConfig mem_cfg;
-};
-
-class TTNNFixtureWithTensor : public TTNNFixtureWithDevice, public testing::WithParamInterface<CreateTensorParameters> {
-protected:
-    [[nodiscard]] const Tensor CreateTensor() {
-        CreateTensorParameters params = GetParam();
-        TensorSpec tensor_spec(
-            params.input_shape, TensorLayout(params.dtype, PageConfig(params.layout), params.mem_cfg));
-        auto input_buffer = tt::tt_metal::tensor_impl::allocate_buffer_on_device(device_, tensor_spec);
-        auto input_storage = tt::tt_metal::DeviceStorage{input_buffer};
-        Tensor input_tensor = Tensor(input_storage, params.input_shape, params.dtype, params.layout);
-        return std::move(input_tensor);
-    }
-};
-
 }  // namespace ttnn
 
 namespace ttnn::distributed::test {