|
30 | 30 | name: Build TransformerEngine |
31 | 31 | # This is a build job, doesn't need GPUs |
32 | 32 | runs-on: [build-only] |
| 33 | + container: |
| 34 | + image: ${{ steps.select-image.outputs.image-tag }} |
| 35 | + options: --shm-size=16g |
33 | 36 |
|
34 | 37 | outputs: |
35 | 38 | docker-image: ${{ steps.select-image.outputs.image-tag }} |
|
41 | 44 | submodules: 'recursive' |
42 | 45 | fetch-depth: 0 # Needed to get all branches for version detection |
43 | 46 |
|
44 | | - - name: 🛠️ Set up Node.js |
45 | | - uses: actions/setup-node@v4 |
46 | | - with: |
47 | | - node-version: '20' |
48 | | - |
49 | 47 | - name: 🖼️ Select Docker Image |
50 | 48 | id: select-image |
51 | 49 | run: | |
@@ -81,26 +79,23 @@ jobs: |
81 | 79 | username: ${{ secrets.ARTIFACTORY_USER }} |
82 | 80 | password: ${{ secrets.ARTIFACTORY_PAT }} |
83 | 81 |
|
84 | | - - name: 📦 Build inside container and create wheel |
85 | | - uses: devcontainers/ci@v0.3 |
86 | | - with: |
87 | | - imageName: ${{ steps.select-image.outputs.image-tag }} |
88 | | - runCmd: | |
89 | | - set -ex |
90 | | - |
91 | | - # Install CMake |
92 | | - CMAKE_LINK="https://github.com/Kitware/CMake/releases/download/v3.30.0/cmake-3.30.0-linux-x86_64.sh" |
93 | | - wget -q -P /opt ${CMAKE_LINK} |
94 | | - chmod +x /opt/cmake-*.sh |
95 | | - bash /opt/cmake-*.sh --skip-license --prefix=/opt/cmake |
96 | | - ln -sf /opt/cmake/bin/* /usr/local/bin/ |
97 | | - cmake --version |
98 | | -
|
99 | | - # Build Wheel |
100 | | - export PYTORCH_ROCM_ARCH=gfx942 |
101 | | - export NVTE_ROCM_ARCH=gfx942 |
102 | | - pip install --upgrade pip ninja |
103 | | - pip install -v . |
| 82 | + - name: 📦 Build wheel inside container |
| 83 | + run: | |
| 84 | + set -ex |
| 85 | + |
| 86 | + # Install CMake |
| 87 | + CMAKE_LINK="https://github.com/Kitware/CMake/releases/download/v3.30.0/cmake-3.30.0-linux-x86_64.sh" |
| 88 | + wget -q -P /opt ${CMAKE_LINK} |
| 89 | + chmod +x /opt/cmake-*.sh |
| 90 | + bash /opt/cmake-*.sh --skip-license --prefix=/opt/cmake |
| 91 | + ln -sf /opt/cmake/bin/* /usr/local/bin/ |
| 92 | + cmake --version |
| 93 | + |
| 94 | + # Build Wheel |
| 95 | + export PYTORCH_ROCM_ARCH=gfx942 |
| 96 | + export NVTE_ROCM_ARCH=gfx942 |
| 97 | + pip install --upgrade pip ninja |
| 98 | + pip install -v . |
104 | 99 | |
105 | 100 | - name: ⬆️ Upload wheel artifact |
106 | 101 | uses: actions/upload-artifact@v4 |
|
0 commit comments