Skip to content

Commit 2cb5216

Browse files
authored
Remove devcontainer
1 parent 02aff26 commit 2cb5216

File tree

1 file changed

+20
-25
lines changed

1 file changed

+20
-25
lines changed

.github/workflows/transformer-engine-ci.yml

Lines changed: 20 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,9 @@ jobs:
3030
name: Build TransformerEngine
3131
# This is a build job, doesn't need GPUs
3232
runs-on: [build-only]
33+
container:
34+
image: ${{ steps.select-image.outputs.image-tag }}
35+
options: --shm-size=16g
3336

3437
outputs:
3538
docker-image: ${{ steps.select-image.outputs.image-tag }}
@@ -41,11 +44,6 @@ jobs:
4144
submodules: 'recursive'
4245
fetch-depth: 0 # Needed to get all branches for version detection
4346

44-
- name: 🛠️ Set up Node.js
45-
uses: actions/setup-node@v4
46-
with:
47-
node-version: '20'
48-
4947
- name: 🖼️ Select Docker Image
5048
id: select-image
5149
run: |
@@ -81,26 +79,23 @@ jobs:
8179
username: ${{ secrets.ARTIFACTORY_USER }}
8280
password: ${{ secrets.ARTIFACTORY_PAT }}
8381

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 .
10499
105100
- name: ⬆️ Upload wheel artifact
106101
uses: actions/upload-artifact@v4

0 commit comments

Comments
 (0)