"""Override build_extension to run cmake."""
def build_extension(self, ext):
- cflags, libs = get_cflags_and_libs('../build/root')
- if len(libs) == 0:
- cflags, libs = get_cflags_and_libs('./bundled/root')
-
- if len(libs) == 0:
- if is_sentencepiece_installed():
- cflags = cflags + run_pkg_config('cflags')
- libs = run_pkg_config('libs')
- else:
- subprocess.check_call(['./build_bundled.sh', __version__])
- cflags, libs = get_cflags_and_libs('./bundled/root')
+ # cflags, libs = get_cflags_and_libs('../build/root')
+ # if len(libs) == 0:
+ # cflags, libs = get_cflags_and_libs('./bundled/root')
+
+ # if len(libs) == 0:
+ # if is_sentencepiece_installed():
+ # cflags = cflags + run_pkg_config('cflags')
+ # libs = run_pkg_config('libs')
+ # else:
+ # subprocess.check_call(['./build_bundled.sh', __version__])
+ # cflags, libs = get_cflags_and_libs('./bundled/root')
# Fix compile on some versions of Mac OSX
# See: https://github.com/neulab/xnmt/issues/199
- if sys.platform == 'darwin':
- cflags.append('-mmacosx-version-min=10.9')
- else:
- cflags.append('-Wl,-strip-all')
- libs.append('-Wl,-strip-all')
+ # if sys.platform == 'darwin':
+ # cflags.append('-mmacosx-version-min=10.9')
+ # else:
+ # cflags.append('-Wl,-strip-all')
+ # libs.append('-Wl,-strip-all')
+ cflags = ['-I../src']
+ cmd = "dpkg-architecture -q DEB_BUILD_GNU_TYPE"
+ arch = subprocess.check_output(cmd, shell=True).decode("utf-8").strip().split()[0]
+ libs = ["-L../obj-%s/src" % arch, "-lsentencepiece", "-lsentencepiece_train"]
print('## cflags={}'.format(' '.join(cflags)))
print('## libs={}'.format(' '.join(libs)))
ext.extra_compile_args = cflags