From 7eda373a1253d7c7a94d99c529bebfa16d57b050 Mon Sep 17 00:00:00 2001
From: Keren Zhou <kerenzhou@openai.com>
Date: Sun, 24 Jul 2022 19:14:52 -0700
Subject: [PATCH] Add lit dependency (#9)

---
 python/setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/setup.py b/python/setup.py
index 05f008d2c..d8ddb327c 100644
--- 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,