diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/fp16/package.py | 46 |
1 files changed, 42 insertions, 4 deletions
diff --git a/var/spack/repos/builtin/packages/fp16/package.py b/var/spack/repos/builtin/packages/fp16/package.py index 168ce71331..da92925e39 100644 --- a/var/spack/repos/builtin/packages/fp16/package.py +++ b/var/spack/repos/builtin/packages/fp16/package.py @@ -6,14 +6,52 @@ from spack import * -class Fp16(Package): +class Fp16(CMakePackage): """FP16 is a header-only library for conversion to/from half-precision floating point formats""" homepage = "https://github.com/Maratyszcza/FP16/" git = "https://github.com/Maratyszcza/FP16.git" - version('master') + version('master', branch='master') + version('2020-05-14', commit='4dfe081cf6bcd15db339cf2680b9281b8451eeb3') # py-torch@1.5:1.9 + version('2018-11-28', commit='febbb1c163726b5db24bed55cc9dc42529068997') # py-torch@1.1:1.4 + version('2018-10-10', commit='34d4bf01bbf7376f2baa71b8fa148b18524d45cf') # py-torch@1.0 + version('2018-02-25', commit='43d6d17df48ebf622587e7ed9472ea76573799b9') # py-torch@:0.4 - def install(self, spec, prefix): - install_tree('include', prefix.include) + depends_on('cmake@2.8.12:', type='build') + depends_on('ninja', type='build') + + generator = 'Ninja' + + resource( + name='psimd', + git='https://github.com/Maratyszcza/psimd.git', + branch='master', + destination='deps', + placement='psimd', + ) + resource( + name='googletest', + url='https://github.com/google/googletest/archive/release-1.8.0.zip', + sha256='f3ed3b58511efd272eb074a3a6d6fb79d7c2e6a0e374323d1e6bcbcc1ef141bf', + destination='deps', + placement='googletest', + ) + resource( + name='googlebenchmark', + url='https://github.com/google/benchmark/archive/v1.2.0.zip', + sha256='cc463b28cb3701a35c0855fbcefb75b29068443f1952b64dd5f4f669272e95ea', + destination='deps', + placement='googlebenchmark', + ) + + def cmake_args(self): + return [ + self.define('PSIMD_SOURCE_DIR', + join_path(self.stage.source_path, 'deps', 'psimd')), + self.define('GOOGLETEST_SOURCE_DIR', + join_path(self.stage.source_path, 'deps', 'googletest')), + self.define('GOOGLEBENCHMARK_SOURCE_DIR', + join_path(self.stage.source_path, 'deps', 'googlebenchmark')), + ] |