#14453: Fix moreh_getitem assert bug (#14454)

#14453: fix moreh_getitem assert bug
tenstorrent · Oct 30, 2024 · e05135b · e05135b
1 parent d4b7fc2
commit e05135b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/ttnn/cpp/ttnn/operations/moreh/moreh_getitem/device/moreh_getitem_rm_factory.cpp b/ttnn/cpp/ttnn/operations/moreh/moreh_getitem/device/moreh_getitem_rm_factory.cpp
@@ -242,8 +242,6 @@ void MorehGetItemOperation::MorehGetItemRmFactory::override_runtime_arguments(
     auto index_dims = cached_program.shared_variables.index_dims;
     auto input_dim_offset = cached_program.shared_variables.input_dim_offset;
 
-    TT_ASSERT(tensor_return_value.buffer()->size() == 1);
-
     auto src_buffer = tensor_args.input.buffer();
     auto dst_buffer = tensor_return_value.buffer();
     auto index_tensors = tensor_args.index_tensors;

diff --git a/ttnn/cpp/ttnn/operations/moreh/moreh_getitem/device/moreh_getitem_tilized_factory.cpp b/ttnn/cpp/ttnn/operations/moreh/moreh_getitem/device/moreh_getitem_tilized_factory.cpp
@@ -555,8 +555,6 @@ void MorehGetItemOperation::MorehGetItemTilizedFactory::override_runtime_argumen
     auto index_dims = cached_program.shared_variables.index_dims;
     auto input_dim_offset = cached_program.shared_variables.input_dim_offset;
 
-    TT_ASSERT(tensor_return_value.buffer()->size() == 1);
-
     auto src_buffer = tensor_args.input.buffer();
     auto dst_buffer = tensor_return_value.buffer();
     auto index_tensors = tensor_args.index_tensors;