Skip to content

Run NCCL tests on the JAX-specific base container #837

Run NCCL tests on the JAX-specific base container

Run NCCL tests on the JAX-specific base container #837

Workflow file for this run

name: NCCL on Kubernetes
on:
schedule:
- cron: '30 8 * * *'
pull_request:
types:
- opened
- reopened
- ready_for_review
- synchronize
paths-ignore:
- '**.md'
- '.github/triage/**'
workflow_dispatch:
inputs:
CONTAINER:
type: string
default: ghcr.io/nvidia/jax:base
description: Container to test, this is assumed to already contain the NCCL tests
required: true
concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
permissions:
actions: write # to cancel previous workflows
contents: read # to fetch code
packages: write # to upload container
jobs:
nccl-tests:
uses: ./.github/workflows/_test_nccl.yaml
with:
CONTAINER: ${{ inputs.CONTAINER }}
secrets: inherit