Add lit dependency (#9)

2022-07-24 19:14:52 -07:00
parent a633d2b403
commit 7eda373a12
1 changed files with 1 additions and 0 deletions
--- a/python/setup.py
+++ b/python/setup.py
@@ -140,6 +140,7 @@ setup(
        "cmake",
        "filelock",
        "torch",
+        "lit",
    ],
    package_data={"triton/ops": ["*.c"], "triton/ops/blocksparse": ["*.c"]},
    include_package_data=True,