From e05135b88931eba7816c7bbf0fb1b7a390e1e835 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Choi=20HyungSuk=28=EC=B5=9C=ED=98=95=EC=84=9D=29?= Date: Wed, 30 Oct 2024 10:29:49 +0900 Subject: [PATCH] #14453: Fix `moreh_getitem` assert bug (#14454) #14453: fix moreh_getitem assert bug --- .../moreh/moreh_getitem/device/moreh_getitem_rm_factory.cpp | 2 -- .../moreh_getitem/device/moreh_getitem_tilized_factory.cpp | 2 -- 2 files changed, 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 index 3cd4ccff68d..8f74b1cbe4f 100644 --- 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 index 480c6010841..74749bc4636 100644 --- 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;