summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/rccl/package.py
diff options
context:
space:
mode:
Diffstat (limited to 'var/spack/repos/builtin/packages/rccl/package.py')
-rw-r--r--var/spack/repos/builtin/packages/rccl/package.py221
1 files changed, 53 insertions, 168 deletions
diff --git a/var/spack/repos/builtin/packages/rccl/package.py b/var/spack/repos/builtin/packages/rccl/package.py
index 9b388d1a27..03e8089738 100644
--- a/var/spack/repos/builtin/packages/rccl/package.py
+++ b/var/spack/repos/builtin/packages/rccl/package.py
@@ -14,98 +14,35 @@ class Rccl(CMakePackage):
implementing all-reduce, all-gather, reduce, broadcast,
and reduce-scatter."""
- homepage = "https://github.com/ROCmSoftwarePlatform/rccl"
- git = "https://github.com/ROCmSoftwarePlatform/rccl.git"
- url = "https://github.com/ROCmSoftwarePlatform/rccl/archive/rocm-5.5.0.tar.gz"
+ homepage = "https://github.com/ROCm/rccl"
+ git = "https://github.com/ROCm/rccl.git"
+ url = "https://github.com/ROCm/rccl/archive/rocm-6.2.4.tar.gz"
tags = ["rocm"]
maintainers("srekolam", "renjithravindrankannath")
libraries = ["librccl"]
+ version("6.2.4", sha256="12a04743ed89a74b4a08aa046b6a549d385e15d6866042fd41eac8f085f50eea")
+ version("6.2.1", sha256="0f5e35c7afbb21c1d49ff201b7d1ddf163d853c27c75c3eaf7b449f4dc1e2188")
+ version("6.2.0", sha256="a29c94ea3b9c1a0121d7b1450cb01a697f9f9132169632312b9b0bf744d3c0e3")
+ version("6.1.2", sha256="98af99c12d800f5439c7740d797162c35810a25e08e3b11b397d3300d3c0148e")
+ version("6.1.1", sha256="6368275059ba190d554535d5aeaa5c2510d944b56efd85c90a1701d0292a14c5")
+ version("6.1.0", sha256="c6308f6883cbd63dceadbe4ee154cc6fa9e6bdccbd2f0fda295b564b0cf01e9a")
+ version("6.0.2", sha256="5c8495acba3d620b751e729d1157e7b4eea8f5e5692c50ce47c5204d3dfd443c")
+ version("6.0.0", sha256="0496d5a5f2e48c92cd390ab318df31a53cf7ec590988c2574c9f3d99c38b0fa7")
version("5.7.1", sha256="fb4c1f0084196d1226ce8a726d0f012d3890b54508a06ca87bbda619be8b90b1")
version("5.7.0", sha256="4c2825a3e4323ef3c2f8855ef445c1a81cf1992fb37e3e8a07a50db354aa3954")
version("5.6.1", sha256="27ec6b86a1a329684d808f728c1fce134517ac8e6e7047689f95dbf8386c077e")
version("5.6.0", sha256="cce13c8a9e233e7ddf91a67b1626b7aaeaf818fefe61af8de6b6b6ff47cb358c")
version("5.5.1", sha256="f6b9dc6dafeb49d95c085825876b09317d8252771c746ccf5aa19a9204a404b2")
version("5.5.0", sha256="be2964b408741d046bcd606d339a233d1d1deac7b841647ec53d6d62d71452ba")
- version("5.4.3", sha256="a2524f602bd7b3b6afeb8ba9aff660216ee807fa836e46442d068b5ed5f51a4d")
- version("5.4.0", sha256="213f4f3d75389be588673e43f563e5c0d6908798228b0b6a71f27138fd4ed0c7")
- version("5.3.3", sha256="8995a2d010ad0748fc85ac06e8da7e8d110ba996db04d42b77526c9c059c05bb")
- version("5.3.0", sha256="51da5099fa58c2be882319cebe9ceabe2062feebcc0c5849e8c109030882c10a")
- version("5.2.3", sha256="ecba09f4c95b4b2dae81b88231a972ac956d29909b5e712e21cf2a74bd251ff4")
- version("5.2.1", sha256="cfd17dc003f19900e44928d81111570d3720d4905321f2a18c909909c4bee822")
- version("5.2.0", sha256="6ee3a04da0d16eb53f768a088633a7d8ecc4416a2d0c07f7ba8426ab7892b060")
- version("5.1.3", sha256="56491257f27b48bf85f4b91434a2a6e49a448337c889db181b02c8a4a260a4bc")
- version("5.1.0", sha256="02b0180857e615326f9cab775573436b9162899ad8e526830f54392b8a51b1f5")
- version(
- "5.0.2",
- sha256="a2377ad2332b93d3443a8ee74f4dd9f965ae8cbbfad473f8f57ca17905389a39",
- deprecated=True,
- )
- version(
- "5.0.0",
- sha256="80eb70243f11b80e215458a67c278cd5a655f6e486289962b92ba3504e50af5c",
- deprecated=True,
- )
- version(
- "4.5.2",
- sha256="36de0d3f3ffad491758d89c208ef72c5be5e0db766053a9c766e9c5c6a33a487",
- deprecated=True,
- )
- version(
- "4.5.0",
- sha256="f806f9f65c490abddc562cb4812e12701582bbb449e41cc4797d00e0dedf084e",
- deprecated=True,
- )
- version(
- "4.3.1",
- sha256="c5db71423dc654e8d2c3111e142e65c89436bc636827d95d41a09a87f44fe246",
- deprecated=True,
- )
- version(
- "4.3.0",
- sha256="b5231d8c5ab034a583feceebcef68d0cc0b05ec5a683f802fc7747c89f27d5f6",
- deprecated=True,
- )
- version(
- "4.2.0",
- sha256="2829fae40ebc1d8be201856d2193a941c87e9cf38dca0a2f4414e675c1742f20",
- deprecated=True,
- )
- version(
- "4.1.0",
- sha256="88ec9b43c31cb054fe6aa28bcc0f4b510213635268f951939d6980eee5bb3680",
- deprecated=True,
- )
- version(
- "4.0.0",
- sha256="0632a15b3d6b5981c05377cf4aeb51546f4c4901fd7c37fb0c98071851ad531a",
- deprecated=True,
- )
- version(
- "3.10.0",
- sha256="d9dd0b0d8b9d056fc5e6c7b814520800190952acd30dac3a7c462c4cb6f42bb3",
- deprecated=True,
- )
- version(
- "3.9.0",
- sha256="ff9d03154d668093309ff814a33788f2cc093b3c627e78e42ae246e6017408b0",
- deprecated=True,
- )
- version(
- "3.8.0",
- sha256="0b6676d06bdb1f65d511a95db9f842a3443def83d75759dfdf812b5e62d8c910",
- deprecated=True,
- )
- version(
- "3.7.0",
- sha256="8273878ff71aac2e7adf5cc8562d2933034c6c6b3652f88fbe3cd4f2691036e3",
- deprecated=True,
- )
- version(
- "3.5.0",
- sha256="290b57a66758dce47d0bfff3f5f8317df24764e858af67f60ddcdcadb9337253",
- deprecated=True,
- )
+ with default_args(deprecated=True):
+ version("5.4.3", sha256="a2524f602bd7b3b6afeb8ba9aff660216ee807fa836e46442d068b5ed5f51a4d")
+ version("5.4.0", sha256="213f4f3d75389be588673e43f563e5c0d6908798228b0b6a71f27138fd4ed0c7")
+ version("5.3.3", sha256="8995a2d010ad0748fc85ac06e8da7e8d110ba996db04d42b77526c9c059c05bb")
+ version("5.3.0", sha256="51da5099fa58c2be882319cebe9ceabe2062feebcc0c5849e8c109030882c10a")
+
+ depends_on("c", type="build") # generated
+ depends_on("cxx", type="build") # generated
amdgpu_targets = ROCmPackage.amdgpu_targets
@@ -116,66 +53,14 @@ class Rccl(CMakePackage):
sticky=True,
)
- patch("0001-Fix-numactl-path-issue.patch", when="@3.7.0:4.3.2")
- patch("0002-Fix-numactl-rocm-smi-path-issue.patch", when="@4.5.0:5.2.1")
patch("0003-Fix-numactl-rocm-smi-path-issue.patch", when="@5.2.3:5.6")
+ patch("0004-Set-rocm-core-path-for-version-file.patch", when="@6.0:")
depends_on("cmake@3.5:", type="build")
- for ver in [
- "3.5.0",
- "3.7.0",
- "3.8.0",
- "3.9.0",
- "3.10.0",
- "4.0.0",
- "4.1.0",
- "4.2.0",
- "4.3.0",
- "4.3.1",
- "4.5.0",
- "4.5.2",
- "5.0.0",
- "5.0.2",
- "5.1.0",
- "5.1.3",
- "5.2.0",
- "5.2.1",
- "5.2.3",
- "5.3.0",
- "5.3.3",
- "5.4.0",
- "5.4.3",
- "5.5.0",
- "5.5.1",
- "5.6.0",
- "5.6.1",
- "5.7.0",
- "5.7.1",
- ]:
- depends_on("rocm-cmake@%s:" % ver, type="build", when="@" + ver)
- depends_on("hip@" + ver, when="@" + ver)
- depends_on("comgr@" + ver, when="@" + ver)
- depends_on("hsa-rocr-dev@" + ver, when="@" + ver)
+ depends_on("chrpath", when="@5.3.0:5", type="build")
+ depends_on("numactl@2:")
for ver in [
- "3.7.0",
- "3.8.0",
- "3.9.0",
- "3.10.0",
- "4.0.0",
- "4.1.0",
- "4.2.0",
- "4.3.0",
- "4.3.1",
- "4.5.0",
- "4.5.2",
- "5.0.0",
- "5.0.2",
- "5.1.0",
- "5.1.3",
- "5.2.0",
- "5.2.1",
- "5.2.3",
"5.3.0",
"5.3.3",
"5.4.0",
@@ -186,34 +71,38 @@ class Rccl(CMakePackage):
"5.6.1",
"5.7.0",
"5.7.1",
+ "6.0.0",
+ "6.0.2",
+ "6.1.0",
+ "6.1.1",
+ "6.1.2",
+ "6.2.0",
+ "6.2.1",
+ "6.2.4",
]:
- depends_on("numactl@2:", when="@" + ver)
+ depends_on(f"rocm-cmake@{ver}:", type="build", when=f"@{ver}")
+ depends_on(f"hip@{ver}", when=f"@{ver}")
+ depends_on(f"comgr@{ver}", when=f"@{ver}")
+ depends_on(f"hsa-rocr-dev@{ver}", when=f"@{ver}")
+ depends_on(f"rocm-smi-lib@{ver}", when=f"@{ver}")
+
for ver in [
- "4.5.0",
- "4.5.2",
- "5.0.0",
- "5.0.2",
- "5.1.0",
- "5.1.3",
- "5.2.0",
- "5.2.1",
- "5.2.3",
- "5.3.0",
- "5.3.3",
- "5.4.0",
- "5.4.3",
"5.5.0",
"5.5.1",
"5.6.0",
"5.6.1",
"5.7.0",
"5.7.1",
+ "6.0.0",
+ "6.0.2",
+ "6.1.0",
+ "6.1.2",
+ "6.2.0",
+ "6.2.1",
+ "6.2.4",
]:
- depends_on("rocm-smi-lib@" + ver, when="@" + ver)
- depends_on("chrpath", when="@5.3.0:")
+ depends_on(f"rocm-core@{ver}", when=f"@{ver}")
- for ver in ["5.5.0", "5.5.1", "5.6.0", "5.6.1", "5.7.0", "5.7.1"]:
- depends_on("rocm-core@" + ver, when="@" + ver)
depends_on("googletest@1.11.0:", when="@5.3:")
@classmethod
@@ -229,28 +118,24 @@ class Rccl(CMakePackage):
def setup_build_environment(self, env):
env.set("CXX", self.spec["hip"].hipcc)
+ env.set("ROCMCORE_PATH", self.spec["rocm-core"].prefix)
def cmake_args(self):
- args = []
- if "@3.7.0:" in self.spec:
- args.append(self.define("NUMACTL_DIR", self.spec["numactl"].prefix))
-
+ args = [
+ self.define("NUMACTL_DIR", self.spec["numactl"].prefix),
+ self.define("ROCM_SMI_DIR", self.spec["rocm-smi-lib"].prefix),
+ ]
if "auto" not in self.spec.variants["amdgpu_target"]:
args.append(self.define_from_variant("AMDGPU_TARGETS", "amdgpu_target"))
if self.spec.satisfies("^cmake@3.21.0:3.21.2"):
args.append(self.define("__skip_rocmclang", "ON"))
- if self.spec.satisfies("@4.5.0:"):
- args.append(self.define("ROCM_SMI_DIR", self.spec["rocm-smi-lib"].prefix))
if self.spec.satisfies("@5.3.0:"):
args.append(self.define("BUILD_TESTS", "ON"))
return args
- def test(self):
- if self.spec.satisfies("@:5.3.0"):
- print("Skipping: stand-alone tests")
- return
- test_dir = join_path(self.spec["rccl"].prefix, "bin")
- with working_dir(test_dir, create=True):
- self.run_test("UnitTests")
+ def test_unit(self):
+ """Run unit tests"""
+ unit_tests = which(join_path(self.prefix.bin, "rccl-UnitTests"))
+ unit_tests()
generated by cgit v1.2.3-70-g09d2 (git 2.46.0) at 2025-07-06 15:17:08 +0000