We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 9d95e0b + 3d44525 commit ab4c889Copy full SHA for ab4c889
Formula/p/pytorch.rb
@@ -104,11 +104,15 @@ def install
104
# Avoid building AVX512 code
105
inreplace "cmake/Modules/FindAVX.cmake", /^CHECK_SSE\(CXX "AVX512"/, "#\\0"
106
107
+ # Avoid bundling libomp
108
+ inreplace "setup.py", /^(\s*)self\._embed_libomp\(\)$/, "\\1pass"
109
+
110
ENV["ATEN_NO_TEST"] = "ON"
111
ENV["BLAS"] = "OpenBLAS"
112
ENV["BUILD_CUSTOM_PROTOBUF"] = "OFF"
113
ENV["BUILD_PYTHON"] = "ON"
114
ENV["BUILD_TEST"] = "OFF"
115
+ ENV["OpenBLAS_HOME"] = Formula["openblas"].opt_prefix
116
ENV["PYTHON_EXECUTABLE"] = which(python3)
117
ENV["PYTORCH_BUILD_VERSION"] = version.to_s
118
ENV["PYTORCH_BUILD_NUMBER"] = "1"
0 commit comments