@@ -66,15 +66,29 @@ jobs:
66
66
python -m pip install --upgrade pip
67
67
pip install packaging setuptools wheel
68
68
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
69
75
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)
72
89
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
75
90
cd aiter
76
- python -m pip install -r requirements.txt
77
- python setup.py develop
91
+ python3 setup.py develop
78
92
79
93
- name : Check API functions
80
94
if : env.HASH_CHANGED == 'true'
0 commit comments