forked from AliceO2Group/AliceO2
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCMakeLists.txt
More file actions
29 lines (27 loc) · 1.05 KB
/
CMakeLists.txt
File metadata and controls
29 lines (27 loc) · 1.05 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# Copyright 2019-2020 CERN and copyright holders of ALICE O2.
# See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
# All rights not expressly granted are reserved.
#
# This software is distributed under the terms of the GNU General Public
# License v3 (GPL Version 3), copied verbatim in the file "COPYING".
#
# In applying this license CERN does not waive the privileges and immunities
# granted to it by virtue of its status as an Intergovernmental Organization
# or submit itself to any jurisdiction.
# Pass ORT variables as a preprocessor definition
if(ORT_ROCM_BUILD)
add_compile_definitions(ORT_ROCM_BUILD=1)
endif()
if(ORT_CUDA_BUILD)
add_compile_definitions(ORT_CUDA_BUILD=1)
endif()
if(ORT_MIGRAPHX_BUILD)
add_compile_definitions(ORT_MIGRAPHX_BUILD=1)
endif()
if(ORT_TENSORRT_BUILD)
add_compile_definitions(ORT_TENSORRT_BUILD=1)
endif()
o2_add_library(ML
SOURCES src/OrtInterface.cxx
TARGETVARNAME targetName
PRIVATE_LINK_LIBRARIES O2::Framework ONNXRuntime::ONNXRuntime)