Turi Create  4.0
training_utils.hpp
1 /* Copyright © 2019 Apple Inc. All rights reserved.
2  *
3  * Use of this source code is governed by a BSD-3-clause license that can
4  * be found in the LICENSE.txt file or at
5  * https://opensource.org/licenses/BSD-3-Clause
6  */
7 #ifndef TURI_UTIL_TRAINING_H_
8 #define TURI_UTIL_TRAINING_H_
9 
10 #include <string>
11 #include <vector>
12 
13 #include <core/logging/assertions.hpp>
14 
15 namespace turi {
16 
17 /**
18  * This prints the verbose statements based on GPU names
19  * it gets from the respective compute contexts.
20  */
21 void print_training_device(std::vector<std::string> gpu_names);
22 
23 } // namespace turi
24 
25 #endif /* TURI_UTIL_TRAINING_H_ */
void print_training_device(std::vector< std::string > gpu_names)