Skip to content

Commit c1d92bc

Browse files
authored
Enable torch.compile validation on Amazon 2023 (#6434)
Which is currently failing: https://github.com/pytorch/test-infra/actions/runs/13973253199/job/39120358041 Test: ``` python3 smoke_test.py --package torchonly torch: 2.7.0+cu126 ATen/Parallel: at::get_num_threads() : 8 at::get_num_interop_threads() : 16 OpenMP 201511 (a.k.a. OpenMP 4.5) omp_get_max_threads() : 8 Intel(R) oneAPI Math Kernel Library Version 2024.2-Product Build 20240605 for Intel(R) 64 architecture applications mkl_get_max_threads() : 8 Intel(R) MKL-DNN v3.7.1 (Git Hash 8d263e693366ef8db40acc569cc7d8edf644556d) std::thread::hardware_concurrency() : 16 Environment variables: OMP_NUM_THREADS : [not set] MKL_NUM_THREADS : [not set] ATen parallel backend: OpenMP Skip version check for channel None as stable version is None Testing smoke_test_conv2d Testing smoke_test_linalg on cpu Testing smoke_test_compile for cpu and torch.float16 Testing smoke_test_compile for cpu and torch.float32 Testing smoke_test_compile for cpu and torch.float64 Picked CPU ISA VecAVX2 bit width 256 Testing smoke_test_compile with mode 'max-autotune' for torch.float32 ```
1 parent bdc4380 commit c1d92bc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/validate-linux-binaries.yml

+4
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,10 @@ jobs:
185185
script: |
186186
set -ex
187187
python3 -m ensurepip --upgrade
188+
# install Dev Tools and python-devel to test torch.compile
189+
yum groupinstall -y "Development Tools"
190+
yum install -y python-devel
191+
188192
CUDA_VERSION=$(python3 ../../test-infra/tools/scripts/get_stable_cuda_version.py --channel ${{ inputs.channel }})
189193
CUDA_VERSION_NODOT=$(echo $CUDA_VERSION | tr -d '.')
190194

0 commit comments

Comments
 (0)