From ef39a09940810b09777833b6731ee7eb40d0cead Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Wed, 6 Nov 2024 08:47:53 -0800 Subject: [PATCH] Tryign out H100 for float8 --- .github/workflows/float8_test.yml | 5 +++++ 1 file changed, 5 insertions(+) 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: