Update FitnessFunctionKernel.cu

Update indexing
buensons · Nov 16, 2020 · f367963 · f367963
1 parent d42be31
commit f367963
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/FitnessFunctionKernel.cu b/src/FitnessFunctionKernel.cu
@@ -36,7 +36,7 @@ __global__ void calculate_fitness(
 
             float x = 0.0f;
             for(int k = 0; k < node_number; ++k) {
-                x += mapWeights[j * node_number + k + node_number * node_number * index] * mapInput[k];
+                x += mapWeights[j * node_number + k + node_number * node_number * index] * mapInput[k + index * node_number];
             }
 
             float y = 1.0 / (1.0 + expf(-5 * x));