diff --git a/oneflow/core/kernel/softmax_kernel.cpp b/oneflow/core/kernel/softmax_kernel.cpp
index e242fcc87f5f5275b51c0dc37519a99efb295ea6..896a7d5ccf094961d1b547231fa0a81189777fc2 100644
--- a/oneflow/core/kernel/softmax_kernel.cpp
+++ b/oneflow/core/kernel/softmax_kernel.cpp
@@ -90,7 +90,7 @@ struct SoftmaxKernelUtil<DeviceType::kCPU, T> {
 
   static void Div(DeviceCtx* ctx, const int64_t n, const int64_t w, T* matrix, const T* vector) {
     for (int64_t i = 0; i < n; ++i) {
-      KernelUtil<DeviceType::kCPU, T>::Div(ctx, n, matrix + i * w, vector + i);
+      KernelUtil<DeviceType::kCPU, T>::Div(ctx, w, matrix + i * w, vector + i);
     }
   }
 };