Skip to content

Commit d7d976b

Browse files
authored
fix cuda error 700 for softmax (#74298)
1 parent 337bb42 commit d7d976b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/phi/kernels/gpudnn/softmax_gpudnn.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1375,7 +1375,7 @@ void SoftmaxBackwardCUDAKernelDriverImpl(const GPUContext& dev_ctx,
13751375
GetSoftmaxTensorDims<IndexType>(out.dims(), axis);
13761376
IndexType N = tensor_dims[0];
13771377
IndexType dim = tensor_dims[1];
1378-
int D = tensor_dims[2];
1378+
IndexType D = tensor_dims[2];
13791379

13801380
if (D == 1) {
13811381
if (!UseCudnnSoftmax<T>(dev_ctx, dim, true) ||

0 commit comments

Comments
 (0)