#11838: Change Files

tenstorrent · Sep 3, 2024 · 61856c8 · 61856c8
1 parent 9086d3a
commit 61856c8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/tt_metal/common/work_split.hpp b/tt_metal/common/work_split.hpp
@@ -13,7 +13,9 @@
 #include "tt_metal/common/math.hpp"
 #include "tt_metal/host_api.hpp"
 
-namespace tt::tt_metal{
+namespace tt {
+namespace tt_metal {
+
 inline uint32_t merge_num_sticks_to_read(uint32_t num_sticks_to_read, uint32_t stick_size_bytes, uint32_t max_read_size) {
     uint32_t total_bytes = num_sticks_to_read * stick_size_bytes;
     uint32_t new_num_sticks_to_read = num_sticks_to_read;
@@ -236,4 +238,5 @@ inline std::tuple<uint32_t, CoreRangeSet, CoreRangeSet, CoreRangeSet, uint32_t,
         target_num_cores, all_cores, core_group_1, core_group_2, units_per_core_group_1, units_per_core_group_2);
 }
 
-} // namespace tt::tt_metal
+}  // namespace tt_metal
+}  // namespace tt
diff --git a/ttnn/ttnn/core.py b/ttnn/ttnn/core.py
@@ -55,9 +55,9 @@ def num_cores_to_core_range_set(
     Returns a CoreRangeSet from number of cores
     """
     return ttnn._ttnn.operations.core.num_cores_to_core_range_set(
-        target_num_cores=target_num_cores,
-        grid_size=grid_size,
-        row_wise=row_wise,
+        target_num_cores,
+        grid_size,
+        row_wise,
     )
 
 
@@ -70,9 +70,9 @@ def num_cores_to_corerange_set(
     Create a CoreRangeSet containing the specified number of cores
     """
     return ttnn._ttnn.operations.core.num_cores_to_corerange_set(
-        target_num_cores=target_num_cores,
-        grid_size=grid_size,
-        row_wise=row_wise,
+        target_num_cores,
+        grid_size,
+        row_wise,
     )