[Softmax][Bugfix] fixed softmax compile error (#63)

DefTruth · web-flow · commit bbec7b5626e4 · 2024-10-08T11:43:26.000+08:00
diff --git a/softmax/softmax.cu b/softmax/softmax.cu
@@ -38,6 +38,7 @@ __device__ __forceinline__ MD warp_reduce_md_op(MD value) {
     value.m = bigger_m.m;
   }
   return value;
+}
 
 // Warp Reduce Sum
 template<const int kWarpSize = WARP_SIZE>

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,7 @@ __device__ __forceinline__ MD warp_reduce_md_op(MD value) {`
`38`	`38`	`value.m = bigger_m.m;`
`39`	`39`	`}`
`40`	`40`	`return value;`
	`41`	`+}`
`41`	`42`
`42`	`43`	`// Warp Reduce Sum`
`43`	`44`	`template<const int kWarpSize = WARP_SIZE>`