float64 support in FIL core

cpp/src/fil/fil.cu

@@ -504,9 +504,9 @@ struct sparse_forest : forest<typename node_t::real_type> {
 
   void free(const raft::handle_t& h) override
   {
-    forest<real_type>::free(h);
     trees_.release();
     nodes_.release();
+    forest<real_type>::free(h);
   }
 
   int num_nodes_ = 0;
@@ -616,13 +616,27 @@ template void init<dense_node<float>, float>(const raft::handle_t& h,
                                              const int* trees,
                                              const dense_node<float>* nodes,
                                              const forest_params_t* params);
+template void init<dense_node<double>, double>(const raft::handle_t& h,
+                                               forest_t<double>* pf,
+                                               const categorical_sets& cat_sets,
+                                               const std::vector<double>& vector_leaf,
+                                               const int* trees,
+                                               const dense_node<double>* nodes,
+                                               const forest_params_t* params);
 template void init<sparse_node16<float>, float>(const raft::handle_t& h,
                                                 forest_t<float>* pf,
                                                 const categorical_sets& cat_sets,
                                                 const std::vector<float>& vector_leaf,
                                                 const int* trees,
                                                 const sparse_node16<float>* nodes,
                                                 const forest_params_t* params);
+template void init<sparse_node16<double>, double>(const raft::handle_t& h,
+                                                  forest_t<double>* pf,
+                                                  const categorical_sets& cat_sets,
+                                                  const std::vector<double>& vector_leaf,
+                                                  const int* trees,
+                                                  const sparse_node16<double>* nodes,
+                                                  const forest_params_t* params);
 template void init<sparse_node8, float>(const raft::handle_t& h,
                                         forest_t<float>* pf,
                                         const categorical_sets& cat_sets,
@@ -639,6 +653,7 @@ void free(const raft::handle_t& h, forest_t<real_t> f)
 }
 
 template void free<float>(const raft::handle_t& h, forest_t<float> f);
+template void free<double>(const raft::handle_t& h, forest_t<double> f);
 
 template <typename real_t>
 void predict(const raft::handle_t& h,
@@ -657,6 +672,12 @@ template void predict<float>(const raft::handle_t& h,
                              const float* data,
                              size_t num_rows,
                              bool predict_proba);
+template void predict<double>(const raft::handle_t& h,
+                              forest_t<double> f,
+                              double* preds,
+                              const double* data,
+                              size_t num_rows,
+                              bool predict_proba);
 
 }  // namespace fil
 }  // namespace ML

cpp/src/fil/infer.cu

@@ -917,9 +917,15 @@ void infer(storage_type forest, predict_params params, cudaStream_t stream)
 template void infer<dense_storage_f32>(dense_storage_f32 forest,
                                        predict_params params,
                                        cudaStream_t stream);
+template void infer<dense_storage_f64>(dense_storage_f64 forest,
+                                       predict_params params,
+                                       cudaStream_t stream);
 template void infer<sparse_storage16_f32>(sparse_storage16_f32 forest,
                                           predict_params params,
                                           cudaStream_t stream);
+template void infer<sparse_storage16_f64>(sparse_storage16_f64 forest,
+                                          predict_params params,
+                                          cudaStream_t stream);
 template void infer<sparse_storage8>(sparse_storage8 forest,
                                      predict_params params,
                                      cudaStream_t stream);