Skip to content

Commit

Permalink
Fixed build script for NVIDIA CUDA version of standard Zhivo library.
Browse files Browse the repository at this point in the history
  • Loading branch information
nthnn committed Oct 22, 2024
1 parent e628c36 commit 375bc69
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions build.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,15 +139,21 @@

cuda_lib_args = [
'nvcc', '--shared', '-Ilib', '-o',
os.path.join('dist', 'stdzhv1.0-nvidia'),
'--compiler-options \'-fPIC\'',
os.path.join('dist', 'stdzhv1.0-nvidia')
]

if PLATFORM == 'Windows':
cuda_lib_args[-1] += '.dll'
else:
cuda_lib_args[-1] += '.so'

cuda_lib_args += [
'--compiler-options', '-fPIC',
rt_bin
] + cc_files

if PLATFORM == 'Windows':
cuda_lib_args[4] += '.dll'
else:
cuda_lib_args[4] += '.so'
cuda_lib_args += ['--compiler-options', '/std:c++17']

try:
os.makedirs(OUT_DIR, exist_ok=True)
Expand Down

0 comments on commit 375bc69

Please sign in to comment.