Merge pull request #4 from CodingCat/patch-1

Fix typo in generated code of sigmod func
dmlc · Aug 30, 2017 · 5e22d6b · 5e22d6b
2 parents eab8e71 + f63891f
commit 5e22d6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/compiler/pred_transform.cc b/src/compiler/pred_transform.cc
@@ -40,7 +40,7 @@ sigmoid(const Model& model, bool batch) {
       "const float alpha = (float)") + treelite::common::ToString(alpha) + ";",
       "int64_t i;",
       "#pragma omp parallel for schedule(static) num_threads(nthread) \\",
-      "   default(none) firstprivate(alpha) shared(pred) private(i) \\",
+      "   default(none) firstprivate(alpha) shared(pred) private(i)",
       "for (i = 0; i < ndata; ++i) {",
       "  pred[i] = 1.0f / (1 + expf(-alpha * pred[i]));",
       "}",