File tree 2 files changed +8
-4
lines changed 2 files changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -54,8 +54,10 @@ struct ForestModel<rapids::HostMemory> {
54
54
try {
55
55
result = filex::import_from_treelite_model (
56
56
*tl_model_->base_tl_model (), filex::preferred_tree_layout,
57
- filex::index_type{}, std::nullopt,
58
- raft_proto::device_type::cpu);
57
+ filex::index_type{}, false , raft_proto::device_type::cpu);
58
+ // TODO(hcho3): Throw a warning when coercing a double-precision
59
+ // floats to single-precision
60
+ // TODO(hcho3): Support inferencing with double-precision floats
59
61
rapids::log_info (__FILE__, __LINE__)
60
62
<< " Loaded model to new FIL format" ;
61
63
}
Original file line number Diff line number Diff line change @@ -73,8 +73,10 @@ struct ForestModel<rapids::DeviceMemory> {
73
73
try {
74
74
result = filex::import_from_treelite_model (
75
75
*tl_model_->base_tl_model (), filex::preferred_tree_layout,
76
- filex::index_type{}, std::nullopt,
77
- raft_proto::device_type::gpu);
76
+ filex::index_type{}, false , raft_proto::device_type::gpu);
77
+ // TODO(hcho3): Throw a warning when coercing a double-precision
78
+ // floats to single-precision
79
+ // TODO(hcho3): Support inferencing with double-precision floats
78
80
rapids::log_info (__FILE__, __LINE__)
79
81
<< " Loaded model to new FIL format" ;
80
82
}
You can’t perform that action at this time.
0 commit comments