Skip to content

Commit

Permalink
Revert "#15565 Add unit test to show sharding ttnn.from_torch problems (
Browse files Browse the repository at this point in the history
#15827)"

This reverts commit 5683627.
  • Loading branch information
nemanjagrujic authored and ttmchiou committed Dec 17, 2024
1 parent 738f9f5 commit 8caba97
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 321 deletions.

This file was deleted.

Original file line number Diff line number Diff line change
Expand Up @@ -1049,51 +1049,6 @@ INSTANTIATE_TEST_SUITE_P(
CreateShardedTensorWithAlignmentExpected{
.physical_shape = Size{28, 9}
}
},
////////////////////////////////////////////////////////////////////
// EXAMPLE 4: Some of block sharding failurs
////////////////////////////////////////////////////////////////////
CreateShardedTensorWithAlignmentParams{
CreateShardedTensorWithAlignmentInputs{
.shape = SimpleShape{32, 4, 8, 768},
.data_type = DataType::BFLOAT16,
.page_config = PageConfig(Layout::TILE),
.memory_config =
MemoryConfig{
.memory_layout = TensorMemoryLayout::BLOCK_SHARDED,
.buffer_type = BufferType::L1,
.shard_spec = ShardSpec{
num_cores_to_corerangeset(64, CoreCoord{8, 8}, /*row_wise=*/true), // tt::div_up(32 * 4 * 8, 128) * tt::div_up(768, 96)
{128, 96},
ShardOrientation::ROW_MAJOR,
false,
ShardMode::PHYSICAL}
}
},
CreateShardedTensorWithAlignmentExpected{
.physical_size = Size{1024, 768}
}
},
CreateShardedTensorWithAlignmentParams{
CreateShardedTensorWithAlignmentInputs{
.shape = SimpleShape{32, 4, 8, 768},
.data_type = DataType::BFLOAT16,
.page_config = PageConfig(Layout::TILE),
.memory_config =
MemoryConfig{
.memory_layout = TensorMemoryLayout::BLOCK_SHARDED,
.buffer_type = BufferType::L1,
.shard_spec = ShardSpec{
num_cores_to_corerangeset(64, CoreCoord{8, 8}, /*row_wise=*/true), // tt::div_up(32 * 4 * 8, 128) * tt::div_up(768, 96)
{128, 96},
ShardOrientation::COL_MAJOR,
false,
ShardMode::PHYSICAL}
}
},
CreateShardedTensorWithAlignmentExpected{
.physical_size = Size{1024, 768}
}
}
) // Values
// clang-format on
Expand Down

0 comments on commit 8caba97

Please sign in to comment.