Skip to content

Commit

Permalink
Try enable_channels_padding=false to solve compile error
Browse files Browse the repository at this point in the history
 - due to tt-metal changes from tenstorrent/tt-metal#15171

(cherry picked from commit 14cbe04)
  • Loading branch information
kmabeeTT committed Nov 28, 2024
1 parent 7fab36e commit 2e2ecfd
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion runtime/lib/ttnn/operations/pool/maxpool2d.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,15 @@ preshardForMaxPool2d(const ::tt::target::ttnn::MaxPool2dOp *op,
op->dilation_width() * (op->kernel_width() - 1) - 1) /
op->stride_width();

// KCM FIXME - Is this the right thing to use?
bool enable_channels_padding = false;

auto parallel_config =
::ttnn::operations::conv::conv2d::determine_parallel_config(
::ttnn::TensorMemoryLayout::HEIGHT_SHARDED, op->batch_size(),
op->channels(), output_height, output_width, op->channels(),
device.compute_with_storage_grid_size(), ShardOrientation::ROW_MAJOR);
device.compute_with_storage_grid_size(), ShardOrientation::ROW_MAJOR,
enable_channels_padding);
auto sharded_memory_config = ::ttnn::operations::conv::conv2d::
create_sharded_memory_config_from_parallel_config(inputShape,
parallel_config, 1);
Expand Down

0 comments on commit 2e2ecfd

Please sign in to comment.