From 33d586364292f65fb41aa596d8eb90797b37a819 Mon Sep 17 00:00:00 2001
From: mahxn0 <1262384588@qq.com>
Date: Mon, 2 Dec 2024 20:43:05 +0800
Subject: [PATCH] [Fix](mlu-ops): modify common func.

---
 kernels/utils/common.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kernels/utils/common.h b/kernels/utils/common.h
index bceb8ccd4..c6bd1aead 100644
--- a/kernels/utils/common.h
+++ b/kernels/utils/common.h
@@ -419,7 +419,7 @@ __mlu_func__ void __mluop_store_str_3D(T *dst, T *src, int size, int seg_num_in,
  *      dst_nram only support nram.
  * ****************************************************************************/
 __mlu_func__ void __mluop_get_stage_indices_tfuse(int *dst_nram, int length) {
-#if (__BANG_ARCH__ == 372 || __BANG_ARCH__ == 592)
+#if __BANG_ARCH__ >= 372
   int align_num = 128;
   int repeat = (int)(logf(length / align_num) / logf(2));
   int remain = length / align_num - powf(2, repeat);