From bba1a103eef702d2a91b85d722c641139296b984 Mon Sep 17 00:00:00 2001 From: Sankar Manoj Date: Fri, 19 Jul 2024 12:51:56 +0000 Subject: [PATCH] #9758: Fixed kernel paths --- .../python_api_testing/unit_testing/misc/test_max_pool.py | 4 ++-- .../operations/pool/maxpool/device/max_pool_single_core.cpp | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/tt_eager/python_api_testing/unit_testing/misc/test_max_pool.py b/tests/tt_eager/python_api_testing/unit_testing/misc/test_max_pool.py index ef3c11fe5d0..661d0a4588a 100644 --- a/tests/tt_eager/python_api_testing/unit_testing/misc/test_max_pool.py +++ b/tests/tt_eager/python_api_testing/unit_testing/misc/test_max_pool.py @@ -171,8 +171,8 @@ def test_run_max_pool( f"Skipping over Resnet specific config where parallelization does not fit on core grid {compute_grid_size}" ) - if (compute_grid_size.x * compute_grid_size.y) == ncores_on_n300: - pytest.skip(f"Skipping on N300 (8x7 core grid) due to bug https://github.com/tenstorrent/tt-metal/issues/5458") + # if (compute_grid_size.x * compute_grid_size.y) == ncores_on_n300: + # pytest.skip(f"Skipping on N300 (8x7 core grid) due to bug https://github.com/tenstorrent/tt-metal/issues/5458") torch.set_printoptions(precision=3, sci_mode=False, linewidth=500, threshold=10000, edgeitems=32) diff --git a/ttnn/cpp/ttnn/operations/pool/maxpool/device/max_pool_single_core.cpp b/ttnn/cpp/ttnn/operations/pool/maxpool/device/max_pool_single_core.cpp index 4054102ed1f..d8fce28c364 100644 --- a/ttnn/cpp/ttnn/operations/pool/maxpool/device/max_pool_single_core.cpp +++ b/ttnn/cpp/ttnn/operations/pool/maxpool/device/max_pool_single_core.cpp @@ -151,7 +151,7 @@ operation::ProgramWithCallbacks max_pool_2d_single_core(const Tensor &input, Ten (in_cb_page_nelems_padded * out_nelems * 2) >> 5 // TODO: generalize num rows to fill in in_cb }; auto reader_config = ReaderDataMovementConfig(reader_ct_args); - std::string reader_kernel_fname("ttnn/cpp/ttnn/operations/pool/device/kernels/dataflow/reader_max_pool_2d_single_core.cpp"); + std::string reader_kernel_fname("ttnn/cpp/ttnn/operations/pool/maxpool/device/kernels/dataflow/reader_max_pool_2d_single_core.cpp"); auto reader_kernel = CreateKernel(program, reader_kernel_fname, cores, @@ -200,7 +200,7 @@ operation::ProgramWithCallbacks max_pool_2d_single_core(const Tensor &input, Ten std::vector writer_ct_args = reader_ct_args; std::vector writer_rt_args = reader_rt_args; auto writer_config = WriterDataMovementConfig(writer_ct_args); - std::string writer_kernel_fname("ttnn/cpp/ttnn/operations/pool/device/kernels/dataflow/writer_max_pool_2d_single_core.cpp"); + std::string writer_kernel_fname("ttnn/cpp/ttnn/operations/pool/maxpool/device/kernels/dataflow/writer_max_pool_2d_single_core.cpp"); auto writer_kernel = CreateKernel(program, writer_kernel_fname, cores, @@ -228,7 +228,7 @@ operation::ProgramWithCallbacks max_pool_2d_single_core(const Tensor &input, Ten nbatch, out_h}, // out_h_per_core .defines = reduce_op_utils::get_defines(reduce_op, reduce_dim)}; - std::string compute_kernel_fname("ttnn/cpp/ttnn/operations/pool/device/kernels/compute/max_pool.cpp"); + std::string compute_kernel_fname("ttnn/cpp/ttnn/operations/pool/maxpool/device/kernels/compute/max_pool.cpp"); auto compute_kernel = CreateKernel(program, compute_kernel_fname, cores,