From b660a2112fa400de618ae90c16b3ac7b1429c496 Mon Sep 17 00:00:00 2001 From: Jen Herting Date: Wed, 11 Nov 2020 11:55:57 -0500 Subject: New package: libcuml (#19471) * [libcuml] created template * [libcuml] set cmake list dir * [libcuml] added some dependencies * [libcuml] adding some default cmake args * [libcuml] depends on nccl * [libcuml] temporarily disabling cuml prims building * [libcuml] building internal faiss? * [libcuml] added dependency treelite * [libcuml] depends on google_test * [libcuml] forcing single gpu per: https://github.com/rapidsai/cuml/issues/2528#issuecomment-657675957 * [libcuml] added prims? * [libcuml] adding libcumlcomms ? * Revert "[libcuml] adding libcumlcomms ?" This reverts commit 0e0765b7d4d27bdc01d54d24e8c2f9eac1cb4eb2. * Revert "[libcuml] added prims?" This reverts commit d0b1f868ea5e425cc7d1a5ce7adc642258dfc38e. * [libcuml] added varient single gpu * [libcuml] removed variant singlegpu * [libcuml] added homepage and description. removed fixmes * [libcuml] flake8 * [libcuml] cleaning up leftover commented out code * [libcuml] updated checksum added ucx dependency --- .../repos/builtin/packages/libcuml/package.py | 45 ++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 var/spack/repos/builtin/packages/libcuml/package.py diff --git a/var/spack/repos/builtin/packages/libcuml/package.py b/var/spack/repos/builtin/packages/libcuml/package.py new file mode 100644 index 0000000000..1645896546 --- /dev/null +++ b/var/spack/repos/builtin/packages/libcuml/package.py @@ -0,0 +1,45 @@ +# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other +# Spack Project Developers. See the top-level COPYRIGHT file for details. +# +# SPDX-License-Identifier: (Apache-2.0 OR MIT) + +from spack import * + + +class Libcuml(CMakePackage): + """cuML is a suite of libraries that implement machine + learning algorithms and mathematical primitives functions + that share compatible APIs with other RAPIDS projects.""" + + homepage = "https://rapids.ai" + url = "https://github.com/rapidsai/cuml/archive/v0.15.0.tar.gz" + + version('0.15.0', sha256='5c9c656ae4eaa94a426e07d7385fd5ea0e5dc7abff806af2941aee10d4ca99c7') + + depends_on('cmake@3.14:', type='build') + depends_on('zlib') + depends_on('libcudf@0.8:') + depends_on('cuda@9.2:') + depends_on('blas') + depends_on('nccl@2.4:') + depends_on('treelite') + depends_on('googletest') + depends_on('libcumlprims') + depends_on('mpi') + depends_on('ucx') + + root_cmakelists_dir = 'cpp' + + def cmake_args(self): + args = [] + + args.append("-DNCCL_PATH={0}".format(self.spec['nccl'].prefix)) + args.append("-DBUILD_CUML_C_LIBRARY=ON") + args.append("-DWITH_UCX=ON") + args.append("-DNVTX=OFF") + args.append("-DBUILD_STATIC_FAISS=ON") + args.append("-DSINGLEGPU=OFF") + args.append("-DENABLE_CUMLPRIMS_MG=ON") + args.append("-DBUILD_CUML_MPI_COMMS=ON") + + return args -- cgit v1.2.3-60-g2f50