Add aiter install to amd docker. (#261)

* add aiter

* Update amd-docker.Dockerfile

* move aiter install

* try sudo

* Update amd-docker.Dockerfile

* Update amd-docker.Dockerfile

* sudo everything

* Update amd-docker.Dockerfile

* Update amd_workflow.yml

* Update amd_workflow.yml

* Update amd-docker.Dockerfile

* Update amd-docker.Dockerfile

* Update amd-docker.Dockerfile

* Update amd_workflow.yml

* aiter hash

Author: saienduri

.github/workflows/amd_workflow.yml

@@ -24,11 +24,6 @@ jobs:
     timeout-minutes: 10
     steps:
     - uses: actions/checkout@v3
-    - name: Setup Python
-      uses: actions/setup-python@v5
-      with:
-        python-version: '3.10'
-
     - name: Create input files
       shell: bash
       run: |

docker/amd-docker.Dockerfile

@@ -23,25 +23,32 @@ RUN sudo apt-get update -y \
     python3.10-venv \
     && sudo rm -rf /var/lib/apt/lists/*
 
+RUN sudo apt-get update && sudo apt-get install -y python3.10 python3-pip python-is-python3 python3-setuptools python3-wheel libpython3.10
+
 RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash && \
     sudo apt-get install git-lfs
 
 RUN sudo groupadd -g 109 render
 
 RUN sudo apt update -y \
     && sudo apt install -y "linux-headers-$(uname -r)" "linux-modules-extra-$(uname -r)" \
-    && sudo apt install python3-setuptools python3-wheel libpython3.10 \
     && sudo usermod -a -G render,video runner \
     && wget https://repo.radeon.com/amdgpu-install/6.3.1/ubuntu/jammy/amdgpu-install_6.3.60301-1_all.deb \
     && sudo apt install -y ./amdgpu-install_6.3.60301-1_all.deb \
     && sudo apt update -y \
     && sudo apt install -y rocm
 
-RUN pip install --upgrade pip
+RUN sudo pip install --upgrade pip
+
+RUN sudo pip install --no-cache-dir pytorch-triton-rocm torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
 
-RUN pip install --no-cache-dir pytorch-triton-rocm torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
+RUN git clone --recursive https://github.com/ROCm/aiter.git \
+    && cd aiter \
+    && git checkout 1d88633958236e942cba3c283864282f7af3ebc5 \
+    && sudo pip install -r requirements.txt \
+    && sudo python3 setup.py develop
 
-RUN pip install \
+RUN sudo pip install \
     ninja \
     numpy \
     packaging \