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

python/setup.py

@@ -84,9 +84,7 @@ directories = [x[0] for x in os.walk(os.path.join(os.path.pardir, 'include'))]
 data = []
 for d in directories:
     files = glob.glob(os.path.join(d, '*.h'), recursive=False)
-    dest = os.path.relpath(d, os.path.pardir)
+    data += [os.path.relpath(f, os.path.pardir) for f in files]
-    dest = os.path.join('triton', '_C', dest)
-    data += [(dest, files)]
 
 setup(
     name='triton',
@@ -95,8 +93,8 @@ setup(
     author_email='ptillet@g.harvard.edu',
     description='A language and compiler for custom Deep Learning operations',
     long_description='',
-    packages=['triton', 'triton/ops'],
+    packages=['triton', 'triton/_C', 'triton/ops'],
-    data_files=data,
+    package_data={'': data},
     ext_modules=[CMakeExtension('triton', 'triton/_C/')],
     cmdclass=dict(build_ext=CMakeBuild),
     zip_safe=False,

python/triton/_C/include

@@ -0,0 +1 @@
+../../../include/

python/triton/kernel.py

@@ -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']