fix GPU build for R package.

microsoft · May 9, 2017 · e2f3fc5 · e2f3fc5
1 parent 62b4691
commit e2f3fc5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/R-package/src/lightgbm-all.cpp b/R-package/src/lightgbm-all.cpp
@@ -31,6 +31,7 @@
 #include "../../src/treelearner/data_parallel_tree_learner.cpp"
 #include "../../src/treelearner/feature_parallel_tree_learner.cpp"
 #include "../../src/treelearner/serial_tree_learner.cpp"
+#include "../../src/treelearner/gpu_tree_learner.cpp"
 #include "../../src/treelearner/tree_learner.cpp"
 #include "../../src/treelearner/voting_parallel_tree_learner.cpp"
 

diff --git a/R-package/src/lightgbm-fullcode.cpp b/R-package/src/lightgbm-fullcode.cpp
@@ -31,6 +31,7 @@
 #include "./src/treelearner/data_parallel_tree_learner.cpp"
 #include "./src/treelearner/feature_parallel_tree_learner.cpp"
 #include "./src/treelearner/serial_tree_learner.cpp"
+#include "./src/treelearner/gpu_tree_learner.cpp"
 #include "./src/treelearner/tree_learner.cpp"
 #include "./src/treelearner/voting_parallel_tree_learner.cpp"