Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit 40c0d00

Browse files
committed
Fix lint
1 parent da75ec1 commit 40c0d00

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/operator/nn/softmax-inl.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -415,7 +415,8 @@ __global__ void softmax_stride1_compute_kernel(const DType *in, OType *out, ITyp
415415

416416
for (index_t i = my_id; i < M; i += threads_per_row) {
417417
const DType val = negate ? -row[i] : row[i];
418-
row[i] = (i < len) ? DType(OP::Map((val - smax)/static_cast<DType>(temperature), ssum)) : DType(0.0f);
418+
row[i] = (i < len) ? DType(OP::Map((val - smax)/static_cast<DType>(temperature), ssum)) :
419+
DType(0.0f);
419420
}
420421
__syncthreads();
421422

0 commit comments

Comments
 (0)