summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/miopen-tensile/package.py
diff options
context:
space:
mode:
authorTodd Gamblin <tgamblin@llnl.gov>2022-07-30 15:19:18 -0700
committerTodd Gamblin <tgamblin@llnl.gov>2022-07-31 13:29:20 -0700
commitf52f6e99dbf1131886a80112b8c79dfc414afb7c (patch)
tree05cb7d64b2395922f2f24683da49f472075be12c /var/spack/repos/builtin/packages/miopen-tensile/package.py
parent549ba1ed32372c67fc57271cde3797d58b7dec6e (diff)
downloadspack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.gz
spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.bz2
spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.tar.xz
spack-f52f6e99dbf1131886a80112b8c79dfc414afb7c.zip
black: reformat entire repository with black
Diffstat (limited to 'var/spack/repos/builtin/packages/miopen-tensile/package.py')
-rw-r--r--var/spack/repos/builtin/packages/miopen-tensile/package.py118
1 files changed, 62 insertions, 56 deletions
diff --git a/var/spack/repos/builtin/packages/miopen-tensile/package.py b/var/spack/repos/builtin/packages/miopen-tensile/package.py
index cf3bd94f10..656c994c83 100644
--- a/var/spack/repos/builtin/packages/miopen-tensile/package.py
+++ b/var/spack/repos/builtin/packages/miopen-tensile/package.py
@@ -10,77 +10,83 @@ from spack.package import *
class MiopenTensile(CMakePackage):
"""MIOpenTensile provides host-callable interfaces to Tensile library.
- MIOpenTensile supports one programming model: HIP"""
+ MIOpenTensile supports one programming model: HIP"""
homepage = "https://github.com/ROCmSoftwarePlatform/MIOpenTensile"
git = "https://github.com/ROCmSoftwarePlatform/MIOpenTensile.git"
- url = "https://github.com/ROCmSoftwarePlatform/MIOpentensile/archive/rocm-5.0.0.tar.gz"
- tags = ['rocm']
-
- maintainers = ['srekolam']
- libraries = ['libMIOpenTensile']
-
- version('5.1.0', sha256='f1ae57bd4df8c154357b3f17caf0cfd5f80ba16ffff67bf6219a56f1eb5f897d')
- version('5.0.2', sha256='7b85a6a37d0905b0a3baa8361fd71a5a32ad90f3a562fd5e1af7e2ba68099fa6')
- version('5.0.0', sha256='276ada52e2e8431851296a60df538e0171f8a1c4e9894de8954ffa9306cda2d8')
- version('4.5.2', sha256='eae14b20aec5ad57815c85d0571b7aecc3704696147f3cdbe34287e88da0c9e9')
- version('4.5.0', sha256='5f181f536040c0612bf889600f75951e7ec031ae5c4cb9c2c44f6ac3b15b004b')
-
- tensile_architecture = ('all', 'gfx906', 'gfx908', 'gfx803', 'gfx900')
-
- variant('tensile_architecture', default='all', values=tensile_architecture, multi=True)
- variant('build_type', default='Release', values=("Release", "Debug", "RelWithDebInfo"), description='CMake build type')
-
- patch('0002-Improve-compilation-by-using-local-tensile-path.patch', when='@4.5.0:')
-
- depends_on('cmake@3.5:', type='build')
- depends_on('msgpack-c@3:')
- depends_on('python@3.6:', type='build')
- depends_on('py-virtualenv', type='build')
- depends_on('perl-file-which', type='build')
- depends_on('py-pyyaml', type='build')
- depends_on('py-wheel', type='build')
- depends_on('py-msgpack', type='build')
- depends_on('py-pip', type='build')
-
- resource(name='Tensile',
- git='https://github.com/ROCmSoftwarePlatform/Tensile.git',
- commit='9cbabb07f81e932b9c98bf5ae48fbd7fcef615cf',
- when='@4.5.0:')
-
- for ver in ['4.5.0', '4.5.2', '5.0.0', '5.0.2', '5.1.0']:
- depends_on('rocm-cmake@%s:' % ver, type='build', when='@' + ver)
- depends_on('hip@' + ver, when='@' + ver)
- depends_on('llvm-amdgpu@' + ver, when='@' + ver)
- depends_on('rocminfo@' + ver, when='@' + ver)
+ url = "https://github.com/ROCmSoftwarePlatform/MIOpentensile/archive/rocm-5.0.0.tar.gz"
+ tags = ["rocm"]
+
+ maintainers = ["srekolam"]
+ libraries = ["libMIOpenTensile"]
+
+ version("5.1.0", sha256="f1ae57bd4df8c154357b3f17caf0cfd5f80ba16ffff67bf6219a56f1eb5f897d")
+ version("5.0.2", sha256="7b85a6a37d0905b0a3baa8361fd71a5a32ad90f3a562fd5e1af7e2ba68099fa6")
+ version("5.0.0", sha256="276ada52e2e8431851296a60df538e0171f8a1c4e9894de8954ffa9306cda2d8")
+ version("4.5.2", sha256="eae14b20aec5ad57815c85d0571b7aecc3704696147f3cdbe34287e88da0c9e9")
+ version("4.5.0", sha256="5f181f536040c0612bf889600f75951e7ec031ae5c4cb9c2c44f6ac3b15b004b")
+
+ tensile_architecture = ("all", "gfx906", "gfx908", "gfx803", "gfx900")
+
+ variant("tensile_architecture", default="all", values=tensile_architecture, multi=True)
+ variant(
+ "build_type",
+ default="Release",
+ values=("Release", "Debug", "RelWithDebInfo"),
+ description="CMake build type",
+ )
+
+ patch("0002-Improve-compilation-by-using-local-tensile-path.patch", when="@4.5.0:")
+
+ depends_on("cmake@3.5:", type="build")
+ depends_on("msgpack-c@3:")
+ depends_on("python@3.6:", type="build")
+ depends_on("py-virtualenv", type="build")
+ depends_on("perl-file-which", type="build")
+ depends_on("py-pyyaml", type="build")
+ depends_on("py-wheel", type="build")
+ depends_on("py-msgpack", type="build")
+ depends_on("py-pip", type="build")
+
+ resource(
+ name="Tensile",
+ git="https://github.com/ROCmSoftwarePlatform/Tensile.git",
+ commit="9cbabb07f81e932b9c98bf5ae48fbd7fcef615cf",
+ when="@4.5.0:",
+ )
+
+ for ver in ["4.5.0", "4.5.2", "5.0.0", "5.0.2", "5.1.0"]:
+ depends_on("rocm-cmake@%s:" % ver, type="build", when="@" + ver)
+ depends_on("hip@" + ver, when="@" + ver)
+ depends_on("llvm-amdgpu@" + ver, when="@" + ver)
+ depends_on("rocminfo@" + ver, when="@" + ver)
def setup_build_environment(self, env):
- env.set('CXX', self.spec['hip'].hipcc)
+ env.set("CXX", self.spec["hip"].hipcc)
@classmethod
def determine_version(cls, lib):
- match = re.search(r'lib\S*\.so\.\d+\.\d+\.(\d)(\d\d)(\d\d)',
- lib)
+ match = re.search(r"lib\S*\.so\.\d+\.\d+\.(\d)(\d\d)(\d\d)", lib)
if match:
- ver = '{0}.{1}.{2}'.format(int(match.group(1)),
- int(match.group(2)),
- int(match.group(3)))
+ ver = "{0}.{1}.{2}".format(
+ int(match.group(1)), int(match.group(2)), int(match.group(3))
+ )
else:
ver = None
return ver
def cmake_args(self):
- arch = self.spec.variants['tensile_architecture'].value
- tensile_path = join_path(self.stage.source_path, 'Tensile')
+ arch = self.spec.variants["tensile_architecture"].value
+ tensile_path = join_path(self.stage.source_path, "Tensile")
args = [
- self.define('TENSILE_USE_MSGPACK', 'ON'),
- self.define('COMPILER', 'hipcc'),
- self.define('TENSILE_USE_LLVM', 'OFF'),
- self.define('CODE_OBJECT_VERSION', 'V3'),
- self.define('TENSILE_LIBRARY_FORMAT', 'msgpack'),
- self.define('MIOPEN_TENSILE_SRC', 'asm_full'),
- self.define('Tensile_TEST_LOCAL_PATH', tensile_path)
+ self.define("TENSILE_USE_MSGPACK", "ON"),
+ self.define("COMPILER", "hipcc"),
+ self.define("TENSILE_USE_LLVM", "OFF"),
+ self.define("CODE_OBJECT_VERSION", "V3"),
+ self.define("TENSILE_LIBRARY_FORMAT", "msgpack"),
+ self.define("MIOPEN_TENSILE_SRC", "asm_full"),
+ self.define("Tensile_TEST_LOCAL_PATH", tensile_path),
]
- args.append(self.define('Tensile_ARCHITECTURE', arch))
+ args.append(self.define("Tensile_ARCHITECTURE", arch))
return args