#18325: missing resolutions

tenstorrent · Feb 26, 2025 · ad74270 · ad74270
1 parent e0a6930
commit ad74270
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 4 deletions.
diff --git a/ttnn/cpp/pybind11/core.hpp b/ttnn/cpp/pybind11/core.hpp
@@ -19,6 +19,9 @@ namespace core {
 void py_module_types(py::module& module) { py::class_<ttnn::Config>(module, "Config"); }
 
 void py_module(py::module& module) {
+    using tt::tt_metal::LightMetalBeginCapture;
+    using tt::tt_metal::LightMetalBinary;
+    using tt::tt_metal::LightMetalEndCapture;
     auto py_config = static_cast<py::class_<ttnn::Config>>(module.attr("Config"));
     py_config.def(py::init<const ttnn::Config&>()).def("__repr__", [](const ttnn::Config& config) {
         return fmt::format("{}", config);

diff --git a/ttnn/cpp/ttnn/operations/core/to_dtype/to_dtype_op.hpp b/ttnn/cpp/ttnn/operations/core/to_dtype/to_dtype_op.hpp
@@ -39,7 +39,7 @@ inline Tensor convert_to_cpp_supported_dtype(const Tensor& input_tensor) {
                     "Tensor MultiDeviceHostStorage cannot be converted to torch directly. Use composer(..) "
                     "functionality.");
             } else {
-                raise_unsupported_storage<T>();
+                tt::tt_metal::raise_unsupported_storage<T>();
             }
         },
         input_tensor.get_storage());

diff --git a/ttnn/cpp/ttnn/operations/trace.cpp b/ttnn/cpp/ttnn/operations/trace.cpp
@@ -11,6 +11,8 @@
 
 namespace ttnn::operations::trace {
 
+using namespace tt::tt_metal;
+
 uint32_t begin_trace_capture(IDevice* device, QueueId cq_id) {
     ZoneScoped;
     uint32_t trace_id = Trace::next_id();

diff --git a/ttnn/cpp/ttnn/tensor/flatbuffer/tensor_types_from_flatbuffer.hpp b/ttnn/cpp/ttnn/tensor/flatbuffer/tensor_types_from_flatbuffer.hpp
@@ -21,6 +21,6 @@ CoreRangeSet from_flatbuffer(const flatbuffer::CoreRangeSet* fb_coord);
 tt::tt_metal::ShardSpec from_flatbuffer(const flatbuffer::ShardSpec* spec);
 tt::tt_metal::MemoryConfig from_flatbuffer(const flatbuffer::MemoryConfig* config);
 tt::tt_metal::TensorLayout from_flatbuffer(const flatbuffer::TensorLayout* layout);
-TensorSpec from_flatbuffer(const flatbuffer::TensorSpec* spec);
+tt::tt_metal::TensorSpec from_flatbuffer(const flatbuffer::TensorSpec* spec);
 
 }  // namespace ttnn
diff --git a/ttnn/cpp/ttnn/tensor/flatbuffer/tensor_types_to_flatbuffer.cpp b/ttnn/cpp/ttnn/tensor/flatbuffer/tensor_types_to_flatbuffer.cpp
@@ -141,7 +141,7 @@ flatbuffers::Offset<flatbuffer::TensorLayout> to_flatbuffer(
 }
 
 flatbuffers::Offset<flatbuffer::TensorSpec> to_flatbuffer(
-    const TensorSpec& spec, flatbuffers::FlatBufferBuilder& builder) {
+    const tt::tt_metal::TensorSpec& spec, flatbuffers::FlatBufferBuilder& builder) {
     const auto& shape = spec.logical_shape();
     auto flat_shape = builder.CreateVector(shape.view().data(), shape.rank());
     return flatbuffer::CreateTensorSpec(builder, flat_shape, to_flatbuffer(spec.tensor_layout(), builder));

diff --git a/ttnn/cpp/ttnn/tensor/flatbuffer/tensor_types_to_flatbuffer.hpp b/ttnn/cpp/ttnn/tensor/flatbuffer/tensor_types_to_flatbuffer.hpp
@@ -32,6 +32,6 @@ flatbuffers::Offset<flatbuffer::MemoryConfig> to_flatbuffer(
 flatbuffers::Offset<flatbuffer::TensorLayout> to_flatbuffer(
     const tt::tt_metal::TensorLayout& layout, flatbuffers::FlatBufferBuilder& builder);
 flatbuffers::Offset<flatbuffer::TensorSpec> to_flatbuffer(
-    const TensorSpec& spec, flatbuffers::FlatBufferBuilder& builder);
+    const tt::tt_metal::TensorSpec& spec, flatbuffers::FlatBufferBuilder& builder);
 
 }  // namespace ttnn