[SETUP] Fixed dependencies

.github/workflows/integration-tests.yml

@@ -76,10 +76,12 @@ jobs:
       - name: Install Triton
         run: |
           cd python
+          pip3 install cmake==3.24
           TRITON_USE_ASSERT_ENABLED_LLVM=TRUE pip3 install -e '.[tests]'
 
       - name: Run lit tests
         run: |
+          pip3 install lit
           cd python
           LIT_TEST_DIR="build/$(ls build)/test"
           if [ ! -d "$LIT_TEST_DIR" ]; then

python/setup.py

@@ -211,19 +211,25 @@ def build_extension(self, ext):
 
 download_and_copy_ptxas()
 
+
 setup(
     name="triton",
     version="2.1.0",
     author="Philippe Tillet",
     author_email="[email protected]",
     description="A language and compiler for custom Deep Learning operations",
     long_description="",
-    packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/impl", "triton/ops", "triton/runtime", "triton/ops/blocksparse"],
+    packages=[
+        "triton",
+        "triton/_C",
+        "triton/language",
+        "triton/tools",
+        "triton/impl",
+        "triton/ops",
+        "triton/runtime",
+        "triton/ops/blocksparse"],
     install_requires=[
-        "cmake>=3.20",
         "filelock",
-        "torch",
-        "lit",
     ],
     package_data={"triton": ["third_party/**/*"]},
     include_package_data=True,
@@ -242,13 +248,18 @@ def build_extension(self, ext):
     ],
     test_suite="tests",
     extras_require={
+        "build": [
+            "cmake>=3.20",
+            "lit",
+        ],
         "tests": [
             "autopep8",
             "flake8",
             "isort",
             "numpy",
             "pytest",
             "scipy>=1.7.1",
+            "torch",
         ],
         "tutorials": [
             "matplotlib",