diff --git a/.github/workflows/float8_test.yml b/.github/workflows/float8_test.yml index f90282011e..3d5fe063a5 100644 --- a/.github/workflows/float8_test.yml +++ b/.github/workflows/float8_test.yml @@ -28,6 +28,11 @@ jobs: torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121' gpu-arch-type: "cuda" gpu-arch-version: "12.1" + - name: H100 + runs-on: linux.aws.h100 + torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121' + gpu-arch-type: "cuda" + gpu-arch-version: "12.1" uses: pytorch/test-infra/.github/workflows/linux_job.yml@main with: