#9758: Fixed kernel paths

tenstorrent · Jul 20, 2024 · bba1a10 · bba1a10
1 parent 5cfcb90
commit bba1a10
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 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
@@ -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
@@ -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<uint32_t> writer_ct_args = reader_ct_args;
     std::vector<uint32_t> 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,