@@ -21,33 +21,33 @@ concurrency:
2121env :
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