summaryrefslogtreecommitdiff
path: root/ext/onnx/meson.build
blob: 90636422480a5f4f5b605ad194f38bc8822dffa7 (plain)
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
30
31
32
if get_option('onnx').disabled()
  subdir_done()
endif


onnxrt_dep = dependency('libonnxruntime',required : get_option('onnx'))

if onnxrt_dep.found()
	onnxrt_include_root = onnxrt_dep.get_pkgconfig_variable('includedir')
	onnxrt_includes = [onnxrt_include_root / 'core/session', onnxrt_include_root / 'core']
	onnxrt_dep_args = []
	
	compiler = meson.get_compiler('cpp')
	if compiler.has_header(onnxrt_include_root / 'core/providers/cuda/cuda_provider_factory.h')
	  onnxrt_dep_args = ['-DGST_ML_ONNX_RUNTIME_HAVE_CUDA']
	endif
    gstonnx = library('gstonnx',
    'gstonnx.c',
    'gstonnxelement.c',
    'gstonnxobjectdetector.cpp',
    'gstonnxclient.cpp',
    c_args : gst_plugins_bad_args,
    cpp_args: onnxrt_dep_args,
    link_args : noseh_link_args,
    include_directories : [configinc, libsinc, onnxrt_includes],
    dependencies : [gstbase_dep, gstvideo_dep, onnxrt_dep, libm],
    install : true,
    install_dir : plugins_install_dir,
  )
  pkgconfig.generate(gstonnx, install_dir : plugins_pkgconfig_install_dir)
  plugins += [gstonnx]
 endif