File tree Expand file tree Collapse file tree 2 files changed +4
-11
lines changed Expand file tree Collapse file tree 2 files changed +4
-11
lines changed Original file line number Diff line number Diff line change @@ -17,10 +17,6 @@ concurrency:
1717env :
1818 HF_TOKEN : ${{ secrets.HF_TOKEN }}
1919
20- permissions :
21- id-token : write
22- contents : read
23-
2420jobs :
2521 test-nightly :
2622 strategy :
@@ -37,16 +33,10 @@ jobs:
3733 torch-spec : ' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
3834 gpu-arch-type : " cpu"
3935 gpu-arch-version : " "
40- - name : ROCM Nightly
41- runs-on : linux.rocm.gpu.2
42- torch-spec : ' --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3'
43- gpu-arch-type : " rocm"
44- gpu-arch-version : " 6.3"
4536
4637 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
4738 with :
4839 timeout : 120
49- no-sudo : ${{ matrix.gpu-arch-type == 'rocm' }}
5040 runner : ${{ matrix.runs-on }}
5141 gpu-arch-type : ${{ matrix.gpu-arch-type }}
5242 gpu-arch-version : ${{ matrix.gpu-arch-version }}
8171 torch-spec : ' torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121'
8272 gpu-arch-type : " cuda"
8373 gpu-arch-version : " 12.1"
74+
8475 - name : CPU 2.3
8576 runs-on : linux.4xlarge
8677 torch-spec : ' torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu'
10899 conda create -n venv python=3.9 -y
109100 conda activate venv
110101 echo "::group::Install newer objcopy that supports --set-section-alignment"
102+ yum install -y devtoolset-10-binutils
103+ export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
111104 python -m pip install --upgrade pip
112105 pip install ${{ matrix.torch-spec }}
113106 pip install -r dev-requirements.txt
Original file line number Diff line number Diff line change @@ -607,7 +607,7 @@ def _torch_version_at_least(min_version):
607607def is_MI300 ():
608608 if torch .cuda .is_available () and torch .version .hip :
609609 mxArchName = ["gfx940" , "gfx941" , "gfx942" ]
610- archName = torch .cuda .get_device_properties (0 ).gcnArchName
610+ archName = torch .cuda .get_device_properties ().gcnArchName
611611 for arch in mxArchName :
612612 if arch in archName :
613613 return True
You can’t perform that action at this time.
0 commit comments