138 lines
5.5 KiB
Ruby
138 lines
5.5 KiB
Ruby
class Tbb < Formula
|
|
desc "Rich and complete approach to parallelism in C++"
|
|
homepage "https://github.com/oneapi-src/oneTBB"
|
|
url "https://github.com/oneapi-src/oneTBB/archive/refs/tags/v2021.7.0.tar.gz"
|
|
sha256 "2cae2a80cda7d45dc7c072e4295c675fff5ad8316691f26f40539f7e7e54c0cc"
|
|
license "Apache-2.0"
|
|
|
|
bottle do
|
|
sha256 cellar: :any, arm64_ventura: "e872d553d5c06d401f725fc6d9d6489cf888c66f7cc1cb1efffaf3640f79c100"
|
|
sha256 cellar: :any, arm64_monterey: "97c4b2a2c11ba82f58dd035f7a48e4a2ba15a619c84965345ec30848a9e0878d"
|
|
sha256 cellar: :any, arm64_big_sur: "d2a5e661d1a86f8e3279399efd50d6c8696fb83ee9359856e0f0a6e8c72141d5"
|
|
sha256 cellar: :any, ventura: "a094729f72f4d89bc7c5c1511fc92d9aa32282125dc08fad25c3921e79d02584"
|
|
sha256 cellar: :any, monterey: "0f2c2a55a0ef29487183373986ee366db3dca5dc6ddac1622bf7c5f555cb9deb"
|
|
sha256 cellar: :any, big_sur: "0a714ba09eb9717b540be7ca5b262cc9fd1300c1793817b265e66f1de37fa4f3"
|
|
sha256 cellar: :any, catalina: "81857d93aaa85e0fd941274661d323c701e3201d218cccb7dcf9c2ff9d80c0bb"
|
|
sha256 cellar: :any_skip_relocation, x86_64_linux: "6f156f15d33d15b3213a01d4a803d3412df93d39dc3c28de03049e2ebf6b6b4b"
|
|
end
|
|
|
|
# If adding `hwloc` for TBBBind, you *must* add a test for its functionality.
|
|
# https://github.com/oneapi-src/oneTBB/blob/690aaf497a78a75ff72cddb084579427ab0a8ffc/CMakeLists.txt#L226-L228
|
|
depends_on "cmake" => :build
|
|
depends_on "python@3.10" => [:build, :test]
|
|
depends_on "swig" => :build
|
|
|
|
# Fix installation of Python components
|
|
# See https://github.com/oneapi-src/oneTBB/issues/343
|
|
patch :DATA
|
|
|
|
# Fix thread creation under heavy load.
|
|
# https://github.com/oneapi-src/oneTBB/pull/824
|
|
# Needed for mold: https://github.com/rui314/mold/releases/tag/v1.4.0
|
|
patch do
|
|
url "https://github.com/oneapi-src/oneTBB/commit/f12c93efd04991bc982a27e2fa6142538c33ca82.patch?full_index=1"
|
|
sha256 "637a65cca11c81fa696112aca714879a2202a20e426eff2be8d2318e344ae15c"
|
|
end
|
|
|
|
def install
|
|
# Prevent `setup.py` from installing tbb4py directly into HOMEBREW_PREFIX.
|
|
# We need this due to our `python@3.10` patch.
|
|
python = Formula["python@3.10"].opt_bin/"python3.10"
|
|
site_packages = Language::Python.site_packages(python)
|
|
inreplace "python/CMakeLists.txt", "@@SITE_PACKAGES@@", site_packages
|
|
|
|
tbb_site_packages = prefix/site_packages/"tbb"
|
|
ENV.append "LDFLAGS", "-Wl,-rpath,#{rpath},-rpath,#{rpath(source: tbb_site_packages)}"
|
|
|
|
args = %w[
|
|
-DTBB_TEST=OFF
|
|
-DTBB4PY_BUILD=ON
|
|
]
|
|
|
|
system "cmake", "-S", ".", "-B", "build/shared",
|
|
"-DBUILD_SHARED_LIBS=ON",
|
|
"-DCMAKE_INSTALL_RPATH=#{rpath}",
|
|
*args, *std_cmake_args
|
|
system "cmake", "--build", "build/shared"
|
|
system "cmake", "--install", "build/shared"
|
|
|
|
system "cmake", "-S", ".", "-B", "build/static",
|
|
"-DBUILD_SHARED_LIBS=OFF",
|
|
*args, *std_cmake_args
|
|
system "cmake", "--build", "build/static"
|
|
lib.install buildpath.glob("build/static/*/libtbb*.a")
|
|
|
|
cd "python" do
|
|
ENV.append_path "CMAKE_PREFIX_PATH", prefix.to_s
|
|
ENV["TBBROOT"] = prefix
|
|
|
|
system python, *Language::Python.setup_install_args(prefix, python)
|
|
end
|
|
|
|
return unless OS.linux?
|
|
|
|
inreplace_files = prefix.glob("rml/CMakeFiles/irml.dir/{flags.make,build.make,link.txt}")
|
|
inreplace inreplace_files, Superenv.shims_path/ENV.cxx, ENV.cxx
|
|
end
|
|
|
|
test do
|
|
# The glob that installs these might fail,
|
|
# so let's check their existence.
|
|
assert_path_exists lib/"libtbb.a"
|
|
assert_path_exists lib/"libtbbmalloc.a"
|
|
|
|
(testpath/"sum1-100.cpp").write <<~EOS
|
|
#include <iostream>
|
|
#include <tbb/blocked_range.h>
|
|
#include <tbb/parallel_reduce.h>
|
|
|
|
int main()
|
|
{
|
|
auto total = tbb::parallel_reduce(
|
|
tbb::blocked_range<int>(0, 100),
|
|
0.0,
|
|
[&](tbb::blocked_range<int> r, int running_total)
|
|
{
|
|
for (int i=r.begin(); i < r.end(); ++i) {
|
|
running_total += i + 1;
|
|
}
|
|
|
|
return running_total;
|
|
}, std::plus<int>()
|
|
);
|
|
|
|
std::cout << total << std::endl;
|
|
return 0;
|
|
}
|
|
EOS
|
|
|
|
system ENV.cxx, "sum1-100.cpp", "--std=c++14", "-L#{lib}", "-ltbb", "-o", "sum1-100"
|
|
assert_equal "5050", shell_output("./sum1-100").chomp
|
|
|
|
system Formula["python@3.10"].opt_bin/"python3.10", "-c", "import tbb"
|
|
end
|
|
end
|
|
|
|
__END__
|
|
diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
|
|
index 1d2b05f..81ba8de 100644
|
|
--- a/python/CMakeLists.txt
|
|
+++ b/python/CMakeLists.txt
|
|
@@ -40,7 +40,7 @@ add_custom_target(
|
|
${PYTHON_EXECUTABLE} ${PYTHON_BUILD_WORK_DIR}/setup.py
|
|
build -b${PYTHON_BUILD_WORK_DIR}
|
|
build_ext ${TBB4PY_INCLUDE_STRING} -L$<TARGET_FILE_DIR:TBB::tbb>
|
|
- install --prefix ${PYTHON_BUILD_WORK_DIR}/build -f
|
|
+ install --prefix ${PYTHON_BUILD_WORK_DIR}/build --install-lib ${PYTHON_BUILD_WORK_DIR}/build/@@SITE_PACKAGES@@ -f
|
|
COMMENT "Build and install to work directory the oneTBB Python module"
|
|
)
|
|
|
|
@@ -49,7 +49,7 @@ add_test(NAME python_test
|
|
-DPYTHON_MODULE_BUILD_PATH=${PYTHON_BUILD_WORK_DIR}/build
|
|
-P ${PROJECT_SOURCE_DIR}/cmake/python/test_launcher.cmake)
|
|
|
|
-install(DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/${PYTHON_BUILD_WORK_DIR}/build/
|
|
+install(DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/${PYTHON_BUILD_WORK_DIR}/
|
|
DESTINATION .
|
|
COMPONENT tbb4py)
|