Use build arg instead of env variable to enable dev features

Signed-off-by: Greg Bonik <gbonik@nvidia.com>

Author: gbonik

CMakeLists.txt

@@ -30,6 +30,8 @@ else()
     endif()
 endif()
 
+option(ENABLE_DEV_FEATURES "Enable development-only features" OFF)
+
 option(ENABLE_COVERAGE_FOR_TESTS "Enable code coverage for tests" OFF)
 set(test_coverage_options)
 

cext/CMakeLists.txt

@@ -52,7 +52,7 @@ target_link_options(_cext PUBLIC ${cext_link_flags} ${nostdlib_flags})
 
 set_target_properties(_cext PROPERTIES SKIP_BUILD_RPATH TRUE)
 
-if ("$ENV{CUDA_TILE_ENABLE_DEV_FEATURES}")
+if (ENABLE_DEV_FEATURES)
     target_compile_definitions(_cext PRIVATE CUDA_TILE_ENABLE_DEV_FEATURES)
 endif()
 

setup.py

@@ -16,11 +16,13 @@
 class BuildExtWithCmake(build_ext):
     user_options = build_ext.user_options + [
         ('disable-internal', None, 'Disable building internal extension'),
+        ('enable-dev-features', None, 'Enable development-only features'),
     ]
 
     def initialize_options(self):
         super().initialize_options()
         self.disable_internal = False
+        self.enable_dev_features = False
 
     def finalize_options(self):
         super().finalize_options()
@@ -43,6 +45,8 @@ def _cmake(self, build_dir: str, build_type: str, dlpack_path: str):
                      "-DCMAKE_POLICY_VERSION_MINIMUM=3.5"]
         if self.disable_internal:
             cmake_cmd.append("-DDISABLE_INTERNAL=1")
+        if self.enable_dev_features:
+            cmake_cmd.append("-DENABLE_DEV_FEATURES=1")
         self.spawn(cmake_cmd)
 
     def run(self):