gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PACKAGING] Fixed import error

Browse Source
This commit is contained in:
Philippe Tillet
2020-05-04 10:14:34 -04:00
committed by Philippe Tillet
parent 435acbf585
commit 769c1180c5
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/setup.py
View File

@@ -35,6 +35,8 @@ class CMakeExtension(Extension):
self.path = path self.path = path
prefix = os.pardir
class CMakeBuild(build_ext): class CMakeBuild(build_ext):
def run(self): def run(self):
@@ -82,19 +84,19 @@ class CMakeBuild(build_ext):
self.distribution.get_version()) self.distribution.get_version())
if not os.path.exists(self.build_temp): if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp) os.makedirs(self.build_temp)
sourcedir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')) sourcedir = os.path.abspath(os.path.join(os.path.dirname(__file__), prefix))
subprocess.check_call(['cmake', sourcedir] + cmake_args, cwd=self.build_temp, env=env) subprocess.check_call(['cmake', sourcedir] + cmake_args, cwd=self.build_temp, env=env)
subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp) subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp)
find_llvm() find_llvm()
directories = [x[0] for x in os.walk(os.path.join('src', 'include'))] directories = [x[0] for x in os.walk(os.path.join(prefix, 'include'))]
data = [] data = []
for d in directories: for d in directories:
for htype in ['h', 'hpp']: for htype in ['h', 'hpp']:
files = glob.glob(os.path.join(d, f'*.{htype}'), recursive=False) files = glob.glob(os.path.join(d, f'*.{htype}'), recursive=False)
data += [os.path.relpath(f, 'src') for f in files] data += [os.path.relpath(f, prefix) for f in files]
setup( setup(
name='triton', name='triton',