summaryrefslogtreecommitdiff
path: root/wscript
diff options
context:
space:
mode:
authorCarl Hetherington <cth@carlh.net>2025-09-19 15:04:25 +0200
committerCarl Hetherington <cth@carlh.net>2025-09-30 00:19:53 +0200
commit24fc7b5d990044c7e9f2c95ea82ce12f024b1bfc (patch)
treec9eb168fa682c0487e91d95b2ddc77147a87ff4e /wscript
parent19a0537345d9c39962f70420299a07293fe6a975 (diff)
wip: CUDA with nvjpeg2k
Diffstat (limited to 'wscript')
-rw-r--r--wscript7
1 files changed, 7 insertions, 0 deletions
diff --git a/wscript b/wscript
index b9ba179bf..a44f25dca 100644
--- a/wscript
+++ b/wscript
@@ -81,6 +81,7 @@ def options(opt):
opt.add_option('--disable-more-warnings', action='store_true', default=False, help='disable some warnings raised by Xcode 15 with the 2.16 branch')
opt.add_option('--c++17', action='store_true', default=False, help='build with C++17 and libxml++-4.0')
opt.add_option('--variant', help="build with variant")
+ opt.add_option('--cuda-path', help='path to directory containing include/cuda_runtime_api.h etc.', default='/usr/local/cuda')
def configure(conf):
conf.load('compiler_cxx')
@@ -193,6 +194,9 @@ def configure(conf):
except conf.errors.ConfigurationError:
pass
+ conf.env.append_value('INCLUDES', os.path.join(conf.options.cuda_path, 'include'))
+ conf.env.append_value('LIBPATH', os.path.join(conf.options.cuda_path, 'targets', 'x86_64-linux', 'lib'))
+
#
# Windows/Linux/macOS specific
#
@@ -680,6 +684,9 @@ def configure(conf):
define_name="DCPOMATIC_HAVE_SQLITE3_PREPARE_V3",
mandatory=False)
+ if conf.check(header_name="nvjpeg2k.h", mandatory=False, define_name="DCPOMATIC_HAVE_NVJPEG2K"):
+ conf.env.LIB_CUDA = ['cudart', 'nvjpeg2k']
+
# Other stuff