diff options
author | Andrey Parfenov <andrey.parfenov@intel.com> | 2023-05-12 18:17:59 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-12 12:17:59 -0400 |
commit | 99d511d3b052f1c286fb2687bd84ccc816306f62 (patch) | |
tree | ec22d073569cc3f24a81fccab1463dda13d4ec09 | |
parent | ab8661533b37320f60350276034e1a03d5185027 (diff) | |
download | spack-99d511d3b052f1c286fb2687bd84ccc816306f62.tar.gz spack-99d511d3b052f1c286fb2687bd84ccc816306f62.tar.bz2 spack-99d511d3b052f1c286fb2687bd84ccc816306f62.tar.xz spack-99d511d3b052f1c286fb2687bd84ccc816306f62.zip |
Add more variants for STREAM to customize build (#37283)
* Added STREAM builds customization
* Changed stream_type to enum
* fix code style issues
Signed-off-by: Andrey Parfenov <andrey.parfenov@intel.com>
* rm not necessary optimization
Signed-off-by: Andrey Parfenov <andrey.parfenov@intel.com>
---------
Signed-off-by: Andrey Parfenov <andrey.parfenov@intel.com>
Co-authored-by: iermolae <igor.ermolaev@intel.com>
-rw-r--r-- | var/spack/repos/builtin/packages/stream/package.py | 29 |
1 files changed, 27 insertions, 2 deletions
diff --git a/var/spack/repos/builtin/packages/stream/package.py b/var/spack/repos/builtin/packages/stream/package.py index 7c580f74ec..931d40369f 100644 --- a/var/spack/repos/builtin/packages/stream/package.py +++ b/var/spack/repos/builtin/packages/stream/package.py @@ -18,12 +18,26 @@ class Stream(MakefilePackage): variant("openmp", default=False, description="Build with OpenMP support") + variant("stream_array_size", default="none", description="Size of work arrays in elements") + variant( + "ntimes", + default="none", + description='STREAM runs each kernel "NTIMES" times and reports the *best* result', + ) + variant("offset", default="none", description="Relative alignment between arrays") + variant( + "stream_type", + default="none", + values=("none", "float", "double", "int", "long"), + description="Datatype of arrays elements", + ) + def edit(self, spec, prefix): makefile = FileFilter("Makefile") # Use the Spack compiler wrappers - makefile.filter("CC = .*", "CC = cc") - makefile.filter("FC = .*", "FC = f77") + makefile.filter("CC = .*", "CC = {0}".format(spack_cc)) + makefile.filter("FC = .*", "FC = {0}".format(spack_f77)) cflags = "-O2" fflags = "-O2" @@ -33,6 +47,17 @@ class Stream(MakefilePackage): if "%aocc" in self.spec: cflags += " -mcmodel=large -ffp-contract=fast -fnt-store" + if self.spec.variants["stream_array_size"].value != "none": + cflags += " -DSTREAM_ARRAY_SIZE={0}".format( + self.spec.variants["stream_array_size"].value + ) + if self.spec.variants["ntimes"].value != "none": + cflags += " -DNTIMES={0}".format(self.spec.variants["ntimes"].value) + if self.spec.variants["offset"].value != "none": + cflags += " -DOFFSET={0}".format(self.spec.variants["offset"].value) + if self.spec.variants["stream_type"].value != "none": + cflags += " -DSTREAM_TYPE={0}".format(self.spec.variants["stream_type"].value) + # Set the appropriate flags for this compiler makefile.filter("CFLAGS = .*", "CFLAGS = {0}".format(cflags)) makefile.filter("FFLAGS = .*", "FFLAGS = {0}".format(fflags)) |