Update softmax.cu

linjames0 · Sep 20, 2023 · 6f5173b · 6f5173b
1 parent 7fdb167
commit 6f5173b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/softmax.cu b/softmax.cu
@@ -35,9 +35,9 @@ __global__ void softmax(float *d_in, float *d_out, float *expArr, float *redArr,
 	}
 
     // calculate e^(x) / sum(e^(x)) = softmax
-  	if(col == 0) {
-    	float sum = redArr[0];
-    	for(int i = 0; i < N; ++i) {
+	if(col == 0) {
+		float sum = redArr[0];
+		for(int i = 0; i < N; ++i) {
   			d_out[i] = expArr[i] / sum;
 		}
 	}