-
Notifications
You must be signed in to change notification settings - Fork 111
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
#0: Add support for all dynamic noc barriers, fix fw logic
- Loading branch information
Showing
25 changed files
with
1,027 additions
and
252 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
114 changes: 114 additions & 0 deletions
114
tests/tt_metal/tt_metal/test_kernels/dataflow/dedicated_noc_writer.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,114 @@ | ||
| // SPDX-FileCopyrightText: © 2025 Tenstorrent Inc. | ||
| // | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| #include <cstdint> | ||
|
|
||
| #include "dataflow_api.h" | ||
| #include "debug/dprint.h" | ||
|
|
||
| void kernel_main() { | ||
| constexpr std::uint32_t iteration = get_compile_time_arg_val(0); | ||
| constexpr std::uint32_t page_size = get_compile_time_arg_val(1); | ||
|
|
||
| std::uint32_t noc_x = get_arg_val<uint32_t>(0); | ||
| std::uint32_t noc_y = get_arg_val<uint32_t>(1); | ||
| bool mcast = get_arg_val<uint32_t>(2); | ||
| std::uint32_t top_left_core_x = get_arg_val<uint32_t>(3); | ||
| std::uint32_t top_left_core_y = get_arg_val<uint32_t>(4); | ||
| std::uint32_t bottom_right_core_x = get_arg_val<uint32_t>(5); | ||
| std::uint32_t bottom_right_core_y = get_arg_val<uint32_t>(6); | ||
| std::uint32_t num_dests = get_arg_val<uint32_t>(7); | ||
|
|
||
| constexpr uint32_t cb_id = 0; | ||
| uint32_t l1_read_addr = get_read_ptr(cb_id); | ||
|
|
||
| uint64_t mcast_addr_self_noc; | ||
| if (noc_index == 0) { | ||
| mcast_addr_self_noc = get_noc_multicast_addr( | ||
| top_left_core_x, top_left_core_y, bottom_right_core_x, bottom_right_core_y, l1_read_addr, noc_index); | ||
| } else { | ||
| mcast_addr_self_noc = get_noc_multicast_addr( | ||
| bottom_right_core_x, bottom_right_core_y, top_left_core_x, top_left_core_y, l1_read_addr, noc_index); | ||
| } | ||
|
|
||
| uint64_t addr_self_noc = get_noc_addr(noc_x, noc_y, l1_read_addr, noc_index); | ||
|
|
||
| DPRINT << "Start" << ENDL(); | ||
|
|
||
| // Test stateful read API | ||
| noc_async_read_set_state(addr_self_noc, noc_index); | ||
| for (uint32_t i = 0; i < iteration; i++) { | ||
| noc_async_read_with_state(l1_read_addr, l1_read_addr, page_size, noc_index); | ||
| } | ||
|
|
||
| // Test stateful read one packet API | ||
| noc_async_read_one_packet_set_state(addr_self_noc, page_size, noc_index); | ||
| for (uint32_t i = 0; i < iteration; i++) { | ||
| noc_async_read_one_packet_with_state(l1_read_addr, l1_read_addr, noc_index); | ||
| } | ||
|
|
||
| // Test stateful write one packet API | ||
| noc_async_write_one_packet_set_state(addr_self_noc, page_size, noc_index); | ||
| for (uint32_t i = 0; i < iteration; i++) { | ||
| noc_async_write_one_packet_with_state(l1_read_addr, l1_read_addr, noc_index); | ||
| } | ||
|
|
||
| // Test gen_fast | ||
| const InterleavedAddrGenFast<false> s0 = { | ||
| .bank_base_address = l1_read_addr, .page_size = page_size, .data_format = DataFormat::Float16_b}; | ||
|
|
||
| for (uint32_t i = 0; i < iteration; i++) { | ||
| uint32_t noc = noc_index; | ||
|
|
||
| // uint32_t noc = noc_index; | ||
| uint64_t noc_addr = get_noc_addr(noc_x, noc_y, l1_read_addr, noc); | ||
|
|
||
| // Test read | ||
| noc_async_read_one_packet(noc_addr, l1_read_addr, page_size, noc); | ||
| noc_async_read(noc_addr, l1_read_addr, page_size, noc); | ||
| // interleaved read | ||
| noc_async_read_tile(i % 1024, s0, l1_read_addr, 0, noc); | ||
|
|
||
| // Test semaphore | ||
| noc_semaphore_inc(noc_addr, 1, noc); | ||
| noc_semaphore_set_remote(l1_read_addr, noc_addr, noc); | ||
|
|
||
| // Test write | ||
| noc_async_write(l1_read_addr, noc_addr, page_size, noc); | ||
| noc_async_write_one_packet(l1_read_addr, noc_addr, page_size, noc); | ||
| // interleaved write | ||
| noc_async_write_tile(i % 1024, s0, l1_read_addr, noc); | ||
|
|
||
| // Test mcast | ||
| if (mcast) { | ||
| // write mcast | ||
| noc_async_write_multicast_one_packet( | ||
| l1_read_addr, mcast_addr_self_noc, page_size, num_dests - 1, false, true, noc); | ||
| noc_async_write_multicast(l1_read_addr, mcast_addr_self_noc, page_size, num_dests - 1, false, true, noc); | ||
| noc_async_write_multicast_loopback_src( | ||
| l1_read_addr, mcast_addr_self_noc, page_size, num_dests, false, true, noc); | ||
| // semaphore mcast | ||
| noc_semaphore_set_multicast(l1_read_addr, mcast_addr_self_noc, num_dests - 1, false, true, noc); | ||
| noc_semaphore_set_multicast_loopback_src(l1_read_addr, mcast_addr_self_noc, num_dests, false, true, noc); | ||
| } | ||
|
|
||
| // dw_write skip BH since there's HW issue | ||
| #ifndef ARCH_BLACKHOLE | ||
| noc_inline_dw_write(noc_addr, 1, 0xF, noc); | ||
| #endif | ||
| } | ||
|
|
||
| DPRINT << "END" << ENDL(); | ||
| DPRINT << "noc_mode " << (uint)noc_mode << ENDL(); | ||
|
|
||
| // Barrier test - test barrier itself working properly | ||
| for (int noc = 0; noc < NUM_NOCS; noc++) { | ||
| noc_async_read_barrier(); | ||
| noc_async_write_barrier(); | ||
| noc_async_writes_flushed(); | ||
| noc_async_posted_writes_flushed(); | ||
| noc_async_atomic_barrier(); | ||
| noc_async_full_barrier(); | ||
| } | ||
| } |
Oops, something went wrong.