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
This commit is contained in:
		
							
								
								
									
										45
									
								
								var/spack/repos/builtin/packages/libcuml/package.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										45
									
								
								var/spack/repos/builtin/packages/libcuml/package.py
									
									
									
									
									
										Normal file
									
								
							| @@ -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 | ||||||
		Reference in New Issue
	
	Block a user
	 Jen Herting
					Jen Herting