multidevice: enable NCCL Copy Engine allgather via CTAPolicy=ZERO #26783
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Build | |
| on: | |
| pull_request: | |
| push: | |
| branches: | |
| - main | |
| # cancels concurrent in-progress runs of a pull request. If it is a push, do not cancel concurrent runs. | |
| # Pushes will generate a uniq run_id, github.event.pull_request.number will be empty | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.run_id}} | |
| cancel-in-progress: true | |
| env: | |
| working_directory: . | |
| jobs: | |
| clang-build-23: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| with: | |
| submodules: true | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.12' | |
| - name: Restore ccache | |
| uses: actions/cache/restore@v4 | |
| with: | |
| path: ~/.cache/ccache | |
| key: ${{ runner.os }}-clang23-ccache-v1-${{ github.run_id }} | |
| restore-keys: | | |
| ${{ runner.os }}-clang23-ccache-v1- | |
| - name: Setup ccache | |
| run: | | |
| sudo apt-get update | |
| sudo apt-get install -y ccache | |
| ccache --set-config=max_size=2G | |
| ccache --set-config=compression=true | |
| ccache --zero-stats | |
| - name: Build with clang++ | |
| working-directory: ${{ env.working_directory }} | |
| run: | | |
| tools/apt-install-things.sh & | |
| tools/pip-install-things.sh & | |
| wait | |
| source tools/setup-env.sh | |
| export NVFUSER_BUILD_NO_CUTLASS=true | |
| export NVFUSER_BUILD_CPP_STANDARD=23 | |
| export NVFUSER_BUILD_ENABLE_PCH=true | |
| pip install -v -e ./python --no-build-isolation | |
| - name: Show ccache statistics | |
| if: always() | |
| run: ccache --show-stats | |
| - name: Save ccache | |
| if: always() | |
| uses: actions/cache/save@v4 | |
| with: | |
| path: ~/.cache/ccache | |
| key: ${{ runner.os }}-clang23-ccache-v1-${{ github.run_id }} | |
| dynamic-type-meson: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| with: | |
| submodules: true | |
| - uses: actions/setup-python@v4 | |
| with: | |
| python-version: '3.10' | |
| - name: Run meson build of dynamic type library | |
| working-directory: ${{ env.working_directory }} | |
| run: | | |
| sudo apt update && sudo apt install -y libgtest-dev libgmock-dev libbenchmark-dev & | |
| pip install git+https://github.com/mesonbuild/meson ninja & | |
| wait | |
| cd lib/dynamic_type | |
| meson setup build | |
| meson compile -C build | |
| meson test --verbose -C build |