[Fix](mlu-ops): modify common func.

Cambricon · Dec 2, 2024 · d0bd4f6 · d0bd4f6
1 parent 273ccdc
commit d0bd4f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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);