[python] modularized triton package

2019-09-04 21:55:47 -04:00
parent f6e9c24fe8
commit 945b5d0de9
12 changed files with 527 additions and 528 deletions
--- a/python/setup.py
+++ b/python/setup.py
@@ -82,7 +82,8 @@ setup(
    author_email='ptillet@g.harvard.edu',
    description='A language and compiler for custom Deep Learning operations',
    long_description='',
-    packages=['triton'],
+    packages=['triton',
+              'triton/ops'],
    ext_modules=[CMakeExtension('triton')],
    cmdclass=dict(build_ext=CMakeBuild),
    zip_safe=False,