zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

QNNPACK: add py-torch submodule commits (#24329)

Browse Source
This commit is contained in:
Adam J. Stewart 2021-06-16 07:57:37 -05:00 committed by GitHub
parent 058ae3f0fd
commit ce0eb4862f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
var/spack/repos/builtin/packages/qnnpack/package.py
View File

@ -16,8 +16,13 @@ class Qnnpack(CMakePackage):
git = "https://github.com/pytorch/QNNPACK.git" git = "https://github.com/pytorch/QNNPACK.git"
version('master', branch='master') version('master', branch='master')
version('2019-08-28', commit='7d2a4e9931a82adc3814275b6219a03e24e36b4c') # py-torch@1.3:1.9
version('2018-12-27', commit='6c62fddc6d15602be27e9e4cbb9e985151d2fa82') # py-torch@1.2
version('2018-12-04', commit='ef05e87cef6b8e719989ce875b5e1c9fdb304c05') # py-torch@1.0:1.1
depends_on('cmake@3.5:', type='build') depends_on('cmake@3.5:', type='build')
depends_on('ninja', type='build')
depends_on('python', type='build')
resource( resource(
name='cpuinfo', name='cpuinfo',
@ -64,20 +69,22 @@ class Qnnpack(CMakePackage):
placement='pthreadpool' placement='pthreadpool'
) )
generator = 'Ninja'
def cmake_args(self): def cmake_args(self):
return [ return [
'-DCPUINFO_SOURCE_DIR={0}'.format( self.define('CPUINFO_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/cpuinfo')), join_path(self.stage.source_path, 'deps', 'cpuinfo')),
'-DFP16_SOURCE_DIR={0}'.format( self.define('FP16_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/fp16')), join_path(self.stage.source_path, 'deps', 'fp16')),
'-DFXDIV_SOURCE_DIR={0}'.format( self.define('FXDIV_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/fxdiv')), join_path(self.stage.source_path, 'deps', 'fxdiv')),
'-DPSIMD_SOURCE_DIR={0}'.format( self.define('PSIMD_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/psimd')), join_path(self.stage.source_path, 'deps', 'psimd')),
'-DPTHREADPOOL_SOURCE_DIR={0}'.format( self.define('PTHREADPOOL_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/pthreadpool')), join_path(self.stage.source_path, 'deps', 'pthreadpool')),
'-DGOOGLEBENCHMARK_SOURCE_DIR={0}'.format( self.define('GOOGLEBENCHMARK_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/googlebenchmark')), join_path(self.stage.source_path, 'deps', 'googlebenchmark')),
'-DGOOGLETEST_SOURCE_DIR={0}'.format( self.define('GOOGLETEST_SOURCE_DIR',
join_path(self.stage.source_path, 'deps/googletest')), join_path(self.stage.source_path, 'deps', 'googletest')),
] ]