#7069: Zonescope all run op variants

tenstorrent · Apr 4, 2024 · d5aeb8e · d5aeb8e
1 parent 19ad08e
commit d5aeb8e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tt_eager/tt_dnn/op_library/run_operation.cpp b/tt_eager/tt_dnn/op_library/run_operation.cpp
@@ -313,6 +313,7 @@ std::vector<Tensor> run_without_autoformat(
     const std::vector<Tensor>& input_tensors,
     const std::vector<std::optional<const Tensor>>& optional_input_tensors
 ) {
+    ZoneScoped;
     Device* device = detail::get_device(input_tensors, optional_input_tensors);
 
     std::vector<Tensor> input_tensors_on_dev;
@@ -342,6 +343,7 @@ std::vector<Tensor> run_without_autoformat(
     const std::vector<std::optional<const Tensor>>& optional_input_tensors,
     const std::vector<std::optional<Tensor>>& optional_output_tensors
 ) {
+    ZoneScoped;
     Device* device = detail::get_device(input_tensors, optional_input_tensors);
 
     std::vector<Tensor> input_tensors_on_dev;
@@ -373,6 +375,7 @@ std::vector<Tensor> run_with_autoformat(
     const float pad_value,
     const bool pad_c
 ) {
+    ZoneScoped;
     if (detail::any_tensor_on_multi_device(input_tensors)) {
         return run(operation, input_tensors, optional_input_tensors);
     }
@@ -430,6 +433,7 @@ std::vector<Tensor> run_with_autoformat(
     const std::vector<std::optional<const Tensor>>& optional_input_tensors,
     const std::vector<std::optional<FormatParams>>& optional_input_formatting
 ) {
+    ZoneScoped;
     if (detail::any_tensor_on_multi_device(input_tensors)) {
         return run(operation, input_tensors, optional_input_tensors);
     }