feat: Modularize Dockerfile dependency and ROCm SDK installations into dedicated scripts and add a GitHub Actions workflow to build and consume a custom RCCL library.
This commit is contained in:
@@ -59,6 +59,14 @@ jobs:
|
||||
with:
|
||||
buildkitd-flags: --config /tmp/buildkitd.toml
|
||||
|
||||
- name: Download custom RCCL artifact
|
||||
uses: dawidd6/action-download-artifact@v6
|
||||
with:
|
||||
workflow: build-rccl.yml
|
||||
name: librccl-gfx1151
|
||||
path: custom_libs
|
||||
if_no_artifact_found: warn
|
||||
|
||||
- name: Log in to Docker Hub
|
||||
uses: docker/login-action@v3
|
||||
with:
|
||||
|
||||
@@ -0,0 +1,40 @@
|
||||
name: build-rccl
|
||||
|
||||
on:
|
||||
workflow_dispatch:
|
||||
|
||||
env:
|
||||
ROCM_MAJOR_VER: 7
|
||||
GFX: gfx1151
|
||||
|
||||
jobs:
|
||||
build-rccl:
|
||||
runs-on: ubuntu-latest
|
||||
container: registry.fedoraproject.org/fedora:43
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Install Dependencies
|
||||
run: bash scripts/install_deps.sh
|
||||
|
||||
- name: Install ROCm SDK
|
||||
run: bash scripts/install_rocm_sdk.sh
|
||||
|
||||
- name: Build RCCL
|
||||
shell: bash
|
||||
run: |
|
||||
source /etc/profile.d/rocm-sdk.sh
|
||||
bash scripts/build_rccl_gfx1151.sh
|
||||
|
||||
- name: Compress Artifact
|
||||
run: |
|
||||
# Path determined from script logic: rocm-systems/projects/rccl/build_gfx1151/librccl.so.1
|
||||
ls -lh rocm-systems/projects/rccl/build_gfx1151/librccl.so.1
|
||||
gzip -c rocm-systems/projects/rccl/build_gfx1151/librccl.so.1 > librccl.so.1.gz
|
||||
ls -lh librccl.so.1.gz
|
||||
|
||||
- name: Upload Artifact
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: librccl-gfx1151
|
||||
path: librccl.so.1.gz
|
||||
Reference in New Issue
Block a user