[Fix](mluOpLogspace): Fix logspace bug. (#1153)

Cambricon · Nov 22, 2024 · 3af1925 · 3af1925
1 parent 1725280
commit 3af1925
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kernels/logspace/logspace_block.mlu b/kernels/logspace/logspace_block.mlu
@@ -72,7 +72,7 @@ __mlu_func__ void float2DifferentType(float *result_float, T *result,
                 (int16_t *)result_ge_half_max, num);
   }
   if (std::is_same<T, int>::value) {
-    __bang_float2int32_tz((int *)result, result_float, num, 0);
+    __cn_vector_cast_f32_to_s32(num, (int *)result, result_float);
   }
 }