in setup.py [0:0]
def build_extension(self, ext):
"""Build a specific CMakeExtension."""
extdir = os.path.abspath(
os.path.dirname(self.get_ext_fullpath(ext.name)))
cmake_args = ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=' + os.path.join(extdir, 'sagemaker_tensorflow')]
cfg = 'Debug' if self.debug else 'Release'
build_args = ['--config', cfg]
cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
build_args += ['--', '-j{}'.format(multiprocessing.cpu_count())]
env = os.environ.copy()
env['CXXFLAGS'] = '{} -DNDEBUG -DVERSION_INFO=\\"{}\\"'.format(
env.get('CXXFLAGS', ''),
self.distribution.get_version())
env['CXX'] = 'g++'
env['PYTHON_EXECUTABLE'] = sys.executable
if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp)
subprocess.check_call(['cmake', ext.sourcedir] + cmake_args,
cwd=self.build_temp, env=env)
subprocess.check_call(['cmake', '--build', '.'] + build_args,
cwd=self.build_temp, env=env)
subprocess.check_call(['ctest'], cwd=self.build_temp)
print()