@@ -73,88 +73,13 @@ jobs:
7373 echo "is MagiAttention modified: ${{ steps.filter.outputs.MagiAttention }}"
7474 echo "is MagiAttention csrc modified: ${{ steps.filter.outputs.MagiAttentionCsrc }}"
7575
76- install_MagiAttention_ngc2505_cuda12 :
77- needs : [detect_changes]
78- if : |
79- always() &&
80- needs.detect_changes.outputs.MagiAttentionCsrc == 'true'
81- environment : ${{ (github.event_name == 'pull_request_target') && 'ci-internal' || '' }}
82- runs-on : [self-hosted]
83- container :
84- image : registry.cn-sh-01.sensecore.cn/sandai-ccr/magi-base:25.05.4
85- options : --gpus all --ipc host
86- credentials :
87- username : ${{ secrets.DOCKER_USER_NAME }}
88- password : ${{ secrets.DOCKER_PASSWORD }}
89- env :
90- http_proxy : ${{ secrets.HTTP_PROXY }}
91- https_proxy : ${{ secrets.HTTP_PROXY }}
92- GITHUB_SHA : ${{ github.event.pull_request.head.sha || github.sha }}
93- volumes :
94- - /mnt/afs/littsk/ci_workspace_magi_attention/:/workspace
95- steps :
96- - name : echo proxy
97- run : |
98- echo "http_proxy: $http_proxy"
99- echo "https_proxy: $https_proxy"
100- echo "secret.HTTP_PROXY: ${{ secrets.HTTP_PROXY }}"
101- echo "secret.HTTPS_PROXY: ${{ secrets.HTTP_PROXY }}"
102- - name : curl google
103- run : |
104- curl www.google.com
105- - uses : actions/checkout@v4
106- with :
107- submodules : ' recursive'
108- ref : ${{ github.event.pull_request.head.sha || github.sha }}
109- repository : ${{ github.event.pull_request.head.repo.full_name || github.repository }}
110- fetch-depth : 0
111- - name : manually merge latest main
112- if : github.event_name == 'pull_request_target'
113- run : |
114- git config --global --add safe.directory $GITHUB_WORKSPACE
115- git config user.name "github-actions"
116- git config user.email "github-actions@github.com"
117-
118- echo "Fetching latest main from base repository..."
119- git fetch https://github.com/${{ github.repository }}.git ${{ github.event.pull_request.base.ref }}
120-
121- echo "Merging latest ${{ github.event.pull_request.base.ref }} into PR branch..."
122- git merge FETCH_HEAD
123- - name : install requirements
124- run : |
125- bash -x .github/scripts/install_requirements.sh
126- - name : clear MagiAttention cache
127- run : |
128- rm -rf /github/home/.cache/magi_attention/
129- - name : install MagiAttention
130- run : |
131- pip install -e . --no-build-isolation -vvv
132- - name : verify MagiAttention import
133- run : |
134- cd /github/home
135- python -c "import magi_attention; print('Pip Installation and Import Success')"
136- # ONLY clean the dist folder to ensure we install the correct newly-built wheel.
137- # We KEEP the 'build/' directory to reuse the compilation cache from the previous 'pip install -e' step.
138- - name : build MagiAttention wheel
139- run : |
140- rm -rf dist/
141- python -m build --wheel --no-isolation -v
142- - name : install MagiAttention wheel
143- run : |
144- pip uninstall -y magi_attention
145- pip install dist/*.whl
146- - name : verify MagiAttention wheel
147- run : |
148- cd /github/home
149- python -c "import magi_attention; print('Wheel Installation and Import Success')"
150-
15176 test_MagiAttention_ngc2510_cuda13 :
152- needs : [detect_changes, install_MagiAttention_ngc2505_cuda12 ]
77+ needs : [detect_changes]
15378 if : |
15479 always() &&
15580 (
15681 needs.detect_changes.outputs.MagiAttention == 'true' &&
157- (needs.detect_changes.outputs.MagiAttentionCsrc != 'true' || needs.install_MagiAttention_ngc2505_cuda12.result == 'success' )
82+ (needs.detect_changes.outputs.MagiAttentionCsrc != 'true')
15883 )
15984 environment : ${{ (github.event_name == 'pull_request_target') && 'ci-internal' || '' }}
16085 runs-on : [self-hosted]
0 commit comments