[CI] Test PyTorch on M1 using OpenMP (#114738)

malfet · pytorchmergebot · commit b20330ef810f · 2023-11-29T17:41:35.000Z
Baby step towards #114721 Pull Request resolved: #114738 Approved by: https://github.com/DanilBaibak, https://github.com/atalman
diff --git a/.ci/pytorch/macos-build.sh b/.ci/pytorch/macos-build.sh
@@ -43,7 +43,7 @@ cross_compile_arm64() {
 compile_arm64() {
   # Compilation for arm64
   # TODO: Compile with OpenMP support (but this causes CI regressions as cross-compilation were done with OpenMP disabled)
-  USE_DISTRIBUTED=0 USE_OPENMP=0 MACOSX_DEPLOYMENT_TARGET=11.0 WERROR=1 BUILD_TEST=OFF USE_PYTORCH_METAL=1 python setup.py bdist_wheel
+  USE_DISTRIBUTED=0 USE_OPENMP=1 MACOSX_DEPLOYMENT_TARGET=11.0 WERROR=1 BUILD_TEST=OFF USE_PYTORCH_METAL=1 python setup.py bdist_wheel
 }
 
 compile_x86_64() {

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ cross_compile_arm64() {`
`43`	`43`	`compile_arm64() {`
`44`	`44`	`# Compilation for arm64`
`45`	`45`	`# TODO: Compile with OpenMP support (but this causes CI regressions as cross-compilation were done with OpenMP disabled)`
`46`		`- USE_DISTRIBUTED=0 USE_OPENMP=0 MACOSX_DEPLOYMENT_TARGET=11.0 WERROR=1 BUILD_TEST=OFF USE_PYTORCH_METAL=1 python setup.py bdist_wheel`
	`46`	`+ USE_DISTRIBUTED=0 USE_OPENMP=1 MACOSX_DEPLOYMENT_TARGET=11.0 WERROR=1 BUILD_TEST=OFF USE_PYTORCH_METAL=1 python setup.py bdist_wheel`
`47`	`47`	`}`
`48`	`48`
`49`	`49`	`compile_x86_64() {`