Make setup.py PyPI-compatible by separating git dependencies (ROCm#1653)

* trying to fix pypi won't handle git dependencies

* fix format

* fix comments typo

Author: xiaohuguo2023

README.md

@@ -25,6 +25,7 @@ If you happen to forget the `--recursive` during `clone`, you can use the follow
 git submodule sync && git submodule update --init --recursive
 ```
 
+
 ### Triton-based Communication (Iris)
 
 AITER supports GPU-initiated communication using the [Iris library](https://github.com/ROCm/iris). This enables high-performance Triton-based communication primitives like reduce-scatter and all-gather.
@@ -34,12 +35,10 @@ AITER supports GPU-initiated communication using the [Iris library](https://gith
 Install with Triton communication support:
 
 ```bash
-# Option 1: Install via extras
-pip install -e ".[triton_comms]"
-
-# Option 2: Install all optional dependencies
-pip install -e ".[all]"
-```
+# Install AITER with Triton communication dependencies
+pip install -e .
+pip install -r requirements-triton-comms.txt
+ ```
 
 For more details, see [docs/triton_comms.md](docs/triton_comms.md).
 

requirements-triton-comms.txt

@@ -0,0 +1,5 @@
+# Optional dependencies for Triton-based GPU communication
+# Iris library for GPU-initiated communication primitives
+# Pinned to commit 905ec1c (Nov 18, 2024) for reproducibility and API stability
+iris @ git+https://github.com/ROCm/iris.git@905ec1cea8f350211a70c7d0b2bc11a09a6f6429
+

setup.py

@@ -307,9 +307,6 @@ def has_ext_modules(self):
         return True
 
 
-# Define Iris dependency once to avoid duplication
-IRIS_DEP = "iris @ git+https://github.com/ROCm/iris.git@905ec1cea8f350211a70c7d0b2bc11a09a6f6429"
-
 setup(
     name=PACKAGE_NAME,
     use_scm_version=True,
@@ -334,14 +331,12 @@ def has_ext_modules(self):
     ],
     extras_require={
         # Triton-based communication using Iris
-        # Pinned to commit 905ec1c (Nov 18, 2024) for reproducibility and API stability
-        "triton_comms": [
-            IRIS_DEP,
-        ],
+        # Note: Iris is not available on PyPI and must be installed separately
+        # Install with: pip install -r requirements-triton-comms.txt
+        # (See requirements-triton-comms.txt for pinned Iris version)
+        "triton_comms": [],
         # Install all optional dependencies
-        "all": [
-            IRIS_DEP,
-        ],
+        "all": [],
     },
     setup_requires=setup_requires,
     distclass=ForcePlatlibDistribution,