summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/gpu-burn/Makefile.patch18
-rw-r--r--var/spack/repos/builtin/packages/gpu-burn/package.py55
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'))