update topk runtime params to uint

tenstorrent · Mar 19, 2024 · aca11d9 · aca11d9
1 parent e83e838
commit aca11d9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/common/inc/sfpu/ckernel_sfpu_topk.h b/common/inc/sfpu/ckernel_sfpu_topk.h
@@ -243,7 +243,7 @@ inline void bitonic_topk_inc_x4_dest(uint inc, bool cr) {
 
 template <bool APPROXIMATION_MODE, int ITERATIONS>
 inline void _bitonic_topk_phases_steps(
-    const int idir, const int i_end_phase, const int i_start_phase, const int i_end_step, const int i_start_step)
+    const uint idir, const uint i_end_phase, const uint i_start_phase, const uint i_end_step, const uint i_start_step)
 {
     // If more than 1 phase is requested, do all the steps from all phases
     // If 1 phase is requested, use i_start_step/i_end_step parameters
@@ -385,7 +385,7 @@ inline void _bitonic_topk_phases_steps(
 }
 
 template <bool APPROXIMATION_MODE, int ITERATIONS>
-inline void _bitonic_topk_merge(const int m_iter, const int k)
+inline void _bitonic_topk_merge(const uint m_iter, const uint k)
 {
     uint dst_addr_offset = 0;
     for (int face=0; face<2; face++) {
@@ -423,7 +423,7 @@ inline void _bitonic_topk_merge(const int m_iter, const int k)
 }
 
 template <bool APPROXIMATION_MODE, int ITERATIONS>
-inline void _bitonic_topk_rebuild(const bool idir, const int m_iter, const int k, const int logk, const int skip_second)
+inline void _bitonic_topk_rebuild(const uint idir, const uint m_iter, const uint k, const uint logk, const uint skip_second)
 {
     // init replay buffer for rebuild interation 'm_iter' if uninitialized
     bool init_rebuild = (topk_replay_init != m_iter+1) ? true : false;