Loading [MathJax]/extensions/tex2jax.js
cuML C++ API  23.12
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
gpu.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2023, NVIDIA CORPORATION.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #pragma once
21 #include <type_traits>
22 namespace ML {
23 namespace experimental {
24 namespace fil {
25 namespace detail {
26 namespace device_initialization {
27 
28 /* Non-CUDA header declaration of the GPU specialization for device
29  * initialization
30  */
31 template <typename forest_t, raft_proto::device_type D>
32 std::enable_if_t<std::conjunction_v<std::bool_constant<raft_proto::GPU_ENABLED>,
33  std::bool_constant<D == raft_proto::device_type::gpu>>,
34  void>
36 
37 } // namespace device_initialization
38 } // namespace detail
39 } // namespace fil
40 
41 } // namespace experimental
42 } // namespace ML
std::enable_if_t< std::disjunction_v< std::bool_constant<!raft_proto::GPU_ENABLED >, std::bool_constant< D==raft_proto::device_type::cpu > >, void > initialize_device(raft_proto::device_id< D > device)
Definition: cpu.hpp:36
Definition: dbscan.hpp:27
Definition: base.hpp:22