From 221ccb8281d9bd8a34b28bec4fa5dccf8e49be5b Mon Sep 17 00:00:00 2001 From: rusty1s Date: Thu, 4 Apr 2024 20:04:40 +0000 Subject: [PATCH] update --- setup.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 4ea063ff2..67993157b 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ class CMakeExtension(Extension): def __init__(self, name, sourcedir=''): Extension.__init__(self, name, sources=[]) - self.sourcedir = os.path.abspath(sourcedir) + self.sourcedir = osp.abspath(sourcedir) class CMakeBuild(build_ext): @@ -29,6 +29,7 @@ def check_env_flag(name: str, default: str = "") -> bool: return value in ["1", "ON", "YES", "TRUE", "Y"] def get_ext_filename(self, ext_name): + print("GET EXT FILENAME") # Remove Python ABI suffix: ext_filename = super().get_ext_filename(ext_name) ext_filename_parts = ext_filename.split('.') @@ -40,7 +41,7 @@ def build_extension(self, ext): import torch - extdir = os.path.abspath(osp.dirname(self.get_ext_fullpath(ext.name))) + extdir = osp.abspath(osp.dirname(self.get_ext_fullpath(ext.name))) self.build_type = "DEBUG" if self.debug else "RELEASE" if self.debug is None: if CMakeBuild.check_env_flag("DEBUG"): @@ -79,10 +80,13 @@ def build_extension(self, ext): build_args = [] + print("1111 ---------------") subprocess.check_call(['cmake', ext.sourcedir] + cmake_args, cwd=self.build_temp) + print("2222 ---------------") subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp) + print("3333 ---------------") def maybe_append_with_mkl(dependencies):