Skip to content

Commit 84ce855

Browse files
[float8nocompile] Temporarily disable float8nocompile CI tests due to flakiness on sm89 (#1902)
1 parent 3fb1665 commit 84ce855

File tree

1 file changed

+29
-29
lines changed

1 file changed

+29
-29
lines changed

.github/workflows/float8nocompile_test.yaml

Lines changed: 29 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -21,33 +21,33 @@ concurrency:
2121
env:
2222
HF_TOKEN: ${{ secrets.HF_TOKEN }}
2323

24-
jobs:
25-
test:
26-
strategy:
27-
fail-fast: false
28-
matrix:
29-
include:
30-
- name: SM-89
31-
runs-on: linux.g6.4xlarge.experimental.nvidia.gpu
32-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
33-
gpu-arch-type: "cuda"
34-
gpu-arch-version: "12.1"
24+
# jobs:
25+
# test:
26+
# strategy:
27+
# fail-fast: false
28+
# matrix:
29+
# include:
30+
# - name: H100
31+
# runs-on: linux.aws.h100
32+
# torch-spec: '--pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124'
33+
# gpu-arch-type: "cuda"
34+
# gpu-arch-version: "12.4"
3535

36-
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
37-
with:
38-
timeout: 300
39-
runner: ${{ matrix.runs-on }}
40-
gpu-arch-type: ${{ matrix.gpu-arch-type }}
41-
gpu-arch-version: ${{ matrix.gpu-arch-version }}
42-
submodules: recursive
43-
script: |
44-
conda create -n venv python=3.9 -y
45-
conda activate venv
46-
export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
47-
python -m pip install --upgrade pip
48-
pip install ${{ matrix.torch-spec }}
49-
pip install -r dev-requirements.txt
50-
pip install .
51-
cd torchao/prototype/float8nocompile
52-
pytest kernels/ --verbose -s
53-
pytest test/train_test.py --verbose -s
36+
# uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
37+
# with:
38+
# timeout: 300
39+
# runner: ${{ matrix.runs-on }}
40+
# gpu-arch-type: ${{ matrix.gpu-arch-type }}
41+
# gpu-arch-version: ${{ matrix.gpu-arch-version }}
42+
# submodules: recursive
43+
# script: |
44+
# conda create -n venv python=3.9 -y
45+
# conda activate venv
46+
# export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
47+
# python -m pip install --upgrade pip
48+
# pip install ${{ matrix.torch-spec }}
49+
# pip install -r dev-requirements.txt
50+
# pip install .
51+
# cd torchao/prototype/float8nocompile
52+
# pytest kernels/ --verbose -s
53+
# pytest test/train_test.py --verbose -s

0 commit comments

Comments
 (0)