Handle the build type

JDAI-CV · May 22, 2019 · 6d04b45 · 6d04b45
1 parent eaeb3a4
commit 6d04b45
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tools/onnx2daq/python/setup.py b/tools/onnx2daq/python/setup.py
@@ -99,6 +99,7 @@ def build_extension(self, ext):
             os.makedirs(CMAKE_BUILD_DIR)
 
         with cd(CMAKE_BUILD_DIR):
+            build_type = 'Release'
             # configure
             cmake_args = [
                 CMAKE,
@@ -112,7 +113,8 @@ def build_extension(self, ext):
             if COVERAGE or DEBUG:
                 # in order to get accurate coverage information, the
                 # build needs to turn off optimizations
-                cmake_args.append('-DCMAKE_BUILD_TYPE=Debug')
+                build_type = 'Debug'
+            cmake_args.append('-DCMAKE_BUILD_TYPE={}'.format(build_type))
             if WINDOWS:
                 cmake_args.extend([
                     # we need to link with libpython on windows, so
@@ -136,6 +138,7 @@ def build_extension(self, ext):
 
             build_args = [CMAKE, '--build', os.curdir]
             if WINDOWS:
+                build_args.extend(['--config', build_type])
                 build_args.extend(['--', '/maxcpucount:{}'.format(multiprocessing.cpu_count())])
             else:
                 build_args.extend(['--', '-j', str(multiprocessing.cpu_count())])