LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
instantiate.hpp
Go to the documentation of this file.
1 // Copyright (c) 2014-2023, Lawrence Livermore National Security, LLC.
3 // Produced at the Lawrence Livermore National Laboratory.
4 // Written by the LBANN Research Team (B. Van Essen, et al.) listed in
5 // the CONTRIBUTORS file. <lbann-dev@llnl.gov>
6 //
7 // LLNL-CODE-697807.
8 // All rights reserved.
9 //
10 // This file is part of LBANN: Livermore Big Artificial Neural Network
11 // Toolkit. For details, see http://software.llnl.gov/LBANN or
12 // https://github.com/LLNL/LBANN.
13 //
14 // Licensed under the Apache License, Version 2.0 (the "Licensee"); you
15 // may not use this file except in compliance with the License. You may
16 // obtain a copy of the License at:
17 //
18 // http://www.apache.org/licenses/LICENSE-2.0
19 //
20 // Unless required by applicable law or agreed to in writing, software
21 // distributed under the License is distributed on an "AS IS" BASIS,
22 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
23 // implied. See the License for the specific language governing
24 // permissions and limitations under the license.
26 
27 PROTO(float);
28 PROTO(double);
29 
30 #ifdef LBANN_HAS_HALF
31 #ifdef LBANN_INSTANTIATE_CPU_HALF
32 PROTO(::lbann::cpu_fp16);
33 #endif // LBANN_INSTANTIATE_CPU_HALF
34 #endif // LBANN_HAS_HALF
35 
36 #ifdef LBANN_HAS_GPU_FP16
37 #ifdef LBANN_INSTANTIATE_GPU_HALF
38 PROTO(::lbann::fp16);
39 #endif // LBANN_INSTANTIATE_GPU_HALF
40 #endif // LBANN_HAS_GPU_FP16
PROTO
PROTO(float)