#0: Fixed output size calc

tenstorrent · Dec 24, 2024 · b03a6b2 · b03a6b2
1 parent d6268b6
commit b03a6b2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ttnn/cpp/ttnn/operations/conv/conv2d/conv2d_utils.cpp b/ttnn/cpp/ttnn/operations/conv/conv2d/conv2d_utils.cpp
@@ -1136,8 +1136,9 @@ std::pair<uint32_t,uint32_t> conv2d::estimate_L1_usage(
     } else if (shard_layout == TensorMemoryLayout::HEIGHT_SHARDED) {
         uint32_t output_size = 0;
         if(use_non_tile_height){
-            uint32_t total_height = conv_output_h * conv_output_w * batch_size;
-            output_size = total_height / pconfig.num_cores_nhw * output_channels;
+            // uint32_t total_height = conv_output_h * conv_output_w * batch_size;
+            // output_size = total_height / pconfig.num_cores_nhw * output_channels;
+            output_size = pconfig.per_core_out_matrix_width * pconfig.per_core_out_matrix_height;
             if(output_dtype == DataType::BFLOAT16) {
                 output_size *= 2;
             } else if(output_dtype == DataType::FLOAT32) {