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

[python] more robust way to add triton includes to python package

Browse Source
This commit is contained in:
Philippe Tillet
2019-09-05 16:01:56 -04:00
parent 945593e847
commit 0a6329ea7d
3 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/triton/kernel.py
View File

@@ -57,11 +57,13 @@ def _write_bindings(src, root):
return (cpp, so)
def _build(src, path):
ccdir = os.path.join(libtriton.__file__, os.path.pardir)
ccdir = os.path.realpath(ccdir)
# include directories
triton_include_dirs = [os.path.realpath(os.path.join(libtriton.__file__, 'include'))]
triton_include_dirs = [os.path.join(ccdir, 'include')]
include_dirs = triton_include_dirs
# library directories
triton_library_dirs = [os.path.realpath(os.path.join(libtriton.__file__, os.path.pardir))]
triton_library_dirs = [ccdir]
library_dirs = triton_library_dirs
# libraries
libraries = ['triton']