@@ -66,15 +66,29 @@ jobs:
6666 python -m pip install --upgrade pip
6767 pip install packaging setuptools wheel
6868 pip install torch --index-url https://download.pytorch.org/whl/rocm6.3
69+
70+ - name : Clone ROCm/aiter (shallow)
71+ run : |
72+ git clone --depth 1 --recursive https://github.com/ROCm/aiter.git
73+ cd aiter
74+ git log -1 --pretty=format:"%H %s" > ../last_commit.txt
6975
70- - name : Clone ROCm/aiter repository
71- if : env.HASH_CHANGED == 'true'
76+ - name : Patch cpp_extension.py to fake ROCm
77+ run : |
78+ cd aiter/aiter/jit/utils
79+ # Replace get_hip_version to return a fake version
80+ sed -i 's/raise RuntimeError("ROCm version file not found")/return "5.0.0"/' cpp_extension.py
81+ # Force IS_HIP_EXTENSION = True
82+ echo "IS_HIP_EXTENSION = True" >> cpp_extension.py
83+ echo "ROCM_HOME = \"/opt/rocm\"" >> cpp_extension.py
84+ echo "HIP_HOME = \"/opt/rocm/hip\"" >> cpp_extension.py
85+ echo "HIP_VERSION = \"5.0.0\"" >> cpp_extension.py
86+ echo "ROCM_VERSION = (5, 0)" >> cpp_extension.py
87+
88+ - name : Install aiter (setup.py develop)
7289 run : |
73- git clone --recursive https://github.com/ROCm/aiter.git
74- sed -i 's/raise RuntimeError("ROCm version file not found")/return "5.0.0"/' aiter/aiter/jit/utils/cpp_extension.py
7590 cd aiter
76- python -m pip install -r requirements.txt
77- python setup.py develop
91+ python3 setup.py develop
7892
7993 - name : Check API functions
8094 if : env.HASH_CHANGED == 'true'
0 commit comments