[BUILD] Fix setup.py (#45)

2022-08-13 16:38:31 -07:00
parent d5856435d7
commit e5ec8e16ea
1 changed files with 1 additions and 1 deletions
--- a/python/setup.py
+++ b/python/setup.py
@@ -128,7 +128,7 @@ setup(
    author_email="phil@openai.com",
    description="A language and compiler for custom Deep Learning operations",
    long_description="",
-    packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/ops", "triton/ops/blocksparse"],
+    packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/ops", "triton/runtime", "triton/ops/blocksparse"],
    install_requires=[
        "cmake",
        "filelock",