diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/gpu-burn/Makefile.patch | 18 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/gpu-burn/package.py | 55 |
2 files changed, 73 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/gpu-burn/Makefile.patch b/var/spack/repos/builtin/packages/gpu-burn/Makefile.patch new file mode 100644 index 0000000000..57378a6876 --- /dev/null +++ b/var/spack/repos/builtin/packages/gpu-burn/Makefile.patch @@ -0,0 +1,18 @@ +--- a/Makefile 2018-12-08 05:36:48.000000000 -0600 ++++ b/Makefile 2019-10-04 13:08:37.095337704 -0500 +@@ -1,12 +1,6 @@ +-CUDAPATH=/usr/local/cuda +- +-# Have this point to an old enough gcc (for nvcc) +-GCCPATH=/usr +- + NVCC=nvcc +-CCPATH=${GCCPATH}/bin + + drv: +- PATH=${PATH}:.:${CCPATH}:${PATH} ${NVCC} -I${CUDAPATH}/include -arch=compute_30 -ptx compare.cu -o compare.ptx +- g++ -O3 -Wno-unused-result -I${CUDAPATH}/include -c gpu_burn-drv.cpp +- g++ -o gpu_burn gpu_burn-drv.o -O3 -lcuda -L${CUDAPATH}/lib64 -L${CUDAPATH}/lib -Wl,-rpath=${CUDAPATH}/lib64 -Wl,-rpath=${CUDAPATH}/lib -lcublas -lcudart -o gpu_burn ++ ${NVCC} -arch=compute_30 -ptx compare.cu -o compare.ptx ++ g++ -O3 -Wno-unused-result -c gpu_burn-drv.cpp ++ g++ -o gpu_burn gpu_burn-drv.o -O3 -lcuda -lcublas -lcudart -o gpu_burn diff --git a/var/spack/repos/builtin/packages/gpu-burn/package.py b/var/spack/repos/builtin/packages/gpu-burn/package.py new file mode 100644 index 0000000000..fd11176baa --- /dev/null +++ b/var/spack/repos/builtin/packages/gpu-burn/package.py @@ -0,0 +1,55 @@ +# Copyright 2013-2019 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 GpuBurn(MakefilePackage, CudaPackage): + """Multi-GPU CUDA stress test. Note that the file pointed to by COMPARE_PTX + needs to be copied or linked to the current working directory before + running gpu_burn.""" + + homepage = "http://wili.cc/blog/gpu-burn.html" + url = "http://wili.cc/blog/entries/gpu-burn/gpu_burn-1.0.tar.gz" + + version('1.0', sha256='d55994f0bee8dabf021966dbe574ef52be1e43386faeee91318dd4ebb36aa74a') + + patch('Makefile.patch') + + # This package uses CudaPackage to pick up the cuda_arch variant. A side + # effect is that it also picks up the cuda variant, but cuda is required + # for gpu-burn so is not really a variant. + variant('cuda', 'True', description='Use CUDA; must be true') + + conflicts('~cuda', msg='gpu-burn requires cuda') + + def edit(self, spec, prefix): + # update cuda architecture if necessary + if '+cuda' in self.spec: + cuda_arch = self.spec.variants['cuda_arch'].value + archflag = '' + + if cuda_arch != 'none': + if len(cuda_arch) > 1: + raise InstallError( + 'gpu-burn only supports compilation for a single GPU' + 'type.' + ) + archflag = '-arch=compute_{0}'.format(cuda_arch[0]) + + filter_file('-arch=compute_30', archflag, + 'Makefile', string=True) + + def install(self, spec, prefix): + mkdir(prefix.bin) + mkdir(prefix.share) + install('gpu_burn', prefix.bin) + install('compare.ptx', prefix.share) + + # The gpu_burn program looks for the compare.ptx file in the current + # working directory. Create an environment variable that can be pointed to + # so that it can be copied or linked. + def setup_environment(self, spack_env, run_env): + run_env.set('COMPARE_PTX', join_path(self.prefix.share, 'compare.ptx')) |