fix precision issue #125

Xtra-Computing · Feb 15, 2019 · a824a1a · a824a1a
1 parent 640b7b0
commit a824a1a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/thundersvm/kernel/kernelmatrix_kernel.cpp b/src/thundersvm/kernel/kernelmatrix_kernel.cpp
@@ -130,7 +130,7 @@ namespace svm_kernel {
                     const float_type *coef1 = &coef_data[(j - 1) * total_sv];
                     const float_type *coef2 = &coef_data[i * total_sv];
                     const kernel_type *k_values = &k_mat_data[idx * total_sv];
-                    kernel_type sum = 0;
+                    double sum = 0;
 #pragma omp parallel for reduction(+:sum)
                     for (int l = 0; l < ci; ++l) {
                         sum += coef1[si + l] * k_values[si + l];

diff --git a/src/thundersvm/kernel/kernelmatrix_kernel.cu b/src/thundersvm/kernel/kernelmatrix_kernel.cu
@@ -58,7 +58,7 @@ namespace svm_kernel {
                     const float_type *coef1 = &coef[(j - 1) * total_sv];
                     const float_type *coef2 = &coef[i * total_sv];
                     const kernel_type *k_values = &k_mat[idx * total_sv];
-                    kernel_type sum = 0;
+                    double sum = 0;
                     for (int l = 0; l < ci; ++l) {
                         sum += coef1[si + l] * k_values[si + l];
                     }