zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pthreadpool: use same flags as PyTorch (#45521)

Browse Source
This commit is contained in:
Adam J. Stewart 2024-08-02 10:11:24 +02:00 committed by GitHub
parent 4e087349a4
commit 53bc782278
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
var/spack/repos/builtin/packages/pthreadpool/package.py
View File

@ -15,7 +15,7 @@ class Pthreadpool(CMakePackage):
license("BSD-2-Clause") license("BSD-2-Clause")
version("master", branch="master") version("master", branch="master")
version("2023-08-29", commit="4fe0e1e183925bf8cfa6aae24237e724a96479b8") # py-torch@2.2 version("2023-08-29", commit="4fe0e1e183925bf8cfa6aae24237e724a96479b8") # py-torch@2.2:
version("2021-04-13", commit="a134dd5d4cee80cce15db81a72e7f929d71dd413") # py-torch@1.9:2.1 version("2021-04-13", commit="a134dd5d4cee80cce15db81a72e7f929d71dd413") # py-torch@1.9:2.1
version("2020-10-05", commit="fa75e65a58a5c70c09c30d17a1fe1c1dff1093ae") # py-torch@1.8 version("2020-10-05", commit="fa75e65a58a5c70c09c30d17a1fe1c1dff1093ae") # py-torch@1.8
version("2020-06-15", commit="029c88620802e1361ccf41d1970bd5b07fd6b7bb") # py-torch@1.6:1.7 version("2020-06-15", commit="029c88620802e1361ccf41d1970bd5b07fd6b7bb") # py-torch@1.6:1.7
@ -23,8 +23,8 @@ class Pthreadpool(CMakePackage):
version("2018-10-08", commit="13da0b4c21d17f94150713366420baaf1b5a46f4") # py-torch@1.0:1.4 version("2018-10-08", commit="13da0b4c21d17f94150713366420baaf1b5a46f4") # py-torch@1.0:1.4
version("2018-02-25", commit="2b06b31f6a315162348e1f3c24325eedaf6cc559") # py-torch@:0.4 version("2018-02-25", commit="2b06b31f6a315162348e1f3c24325eedaf6cc559") # py-torch@:0.4
depends_on("c", type="build") # generated depends_on("c", type="build")
depends_on("cxx", type="build") # generated depends_on("cxx", type="build")
generator("ninja") generator("ninja")
depends_on("cmake@3.5:", type="build") depends_on("cmake@3.5:", type="build")
@ -54,7 +54,6 @@ class Pthreadpool(CMakePackage):
def cmake_args(self): def cmake_args(self):
return [ return [
self.define("BUILD_SHARED_LIBS", True),
self.define("FXDIV_SOURCE_DIR", join_path(self.stage.source_path, "deps", "fxdiv")), self.define("FXDIV_SOURCE_DIR", join_path(self.stage.source_path, "deps", "fxdiv")),
self.define( self.define(
"GOOGLETEST_SOURCE_DIR", join_path(self.stage.source_path, "deps", "googletest") "GOOGLETEST_SOURCE_DIR", join_path(self.stage.source_path, "deps", "googletest")
@ -63,6 +62,9 @@ def cmake_args(self):
"GOOGLEBENCHMARK_SOURCE_DIR", "GOOGLEBENCHMARK_SOURCE_DIR",
join_path(self.stage.source_path, "deps", "googlebenchmark"), join_path(self.stage.source_path, "deps", "googlebenchmark"),
), ),
self.define("PTHREADPOOL_BUILD_TESTS", self.run_tests), # https://github.com/pytorch/pytorch/blob/main/cmake/Dependencies.cmake
self.define("PTHREADPOOL_BUILD_BENCHMARKS", self.run_tests), self.define("PTHREADPOOL_BUILD_TESTS", False),
self.define("PTHREADPOOL_BUILD_BENCHMARKS", False),
self.define("PTHREADPOOL_LIBRARY_TYPE", "static"),
self.define("PTHREADPOOL_ALLOW_DEPRECATED_API", True),
] ]