80 lines
3.7 KiB
Python
80 lines
3.7 KiB
Python
# Copyright 2013-2021 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 Mivisionx(CMakePackage):
|
|
"""MIVisionX toolkit is a set of comprehensive computer
|
|
vision and machine intelligence libraries, utilities, and
|
|
applications bundled into a single toolkit."""
|
|
|
|
homepage = "https://github.com/GPUOpen-ProfessionalCompute-Libraries/MIVisionX"
|
|
git = "https://github.com/GPUOpen-ProfessionalCompute-Libraries/MIVisionX.git"
|
|
url = "https://github.com/GPUOpen-ProfessionalCompute-Libraries/MIVisionX/archive/rocm-4.3.0.tar.gz"
|
|
|
|
maintainers = ['srekolam', 'arjun-raj-kuppala']
|
|
|
|
def url_for_version(self, version):
|
|
if version == Version('1.7'):
|
|
return "https://github.com/GPUOpen-ProfessionalCompute-Libraries/MIVisionX/archive/1.7.tar.gz"
|
|
|
|
url = "https://github.com/GPUOpen-ProfessionalCompute-Libraries/MIVisionX/archive/rocm-{0}.tar.gz"
|
|
return url.format(version)
|
|
|
|
version('4.3.0', sha256='31f6ab9fb7f40b23d4b24c9a70f809623720943e0492c3d357dd22dcfa50efb2')
|
|
version('4.2.0', sha256='172857b1b340373ae81ed6aa241559aa781e32250e75c82d7ba3c002930a8a3a')
|
|
version('4.1.0', sha256='0b431a49807682b9a81adac6a64160a0712ddaa3963e0f05595c93b92be777ea')
|
|
version('4.0.0', sha256='e09d4890b729740ded056b3974daea84c8eb1fc93714c52bf89f853c2eef1fb5')
|
|
version('3.10.0', sha256='8a67fae77a05ef60a501e64a572a7bd2ccb9243518b1414112ccd1d1f78d08c8')
|
|
version('3.9.0', sha256='892812cc6e6977ed8cd4b69c63f4c17be43b83c78eeafd9549236c17f6eaa2af')
|
|
version('3.8.0', sha256='4e177a9b5dcae671d6ea9f0686cf5f73fcd1e3feb3c50425c8c6d43ac5d77feb')
|
|
version('3.7.0', sha256='3ce13c6449739c653139fc121411d94eaa9d764d3d339c4c78fab4b8aa199965')
|
|
version('1.7', sha256='ff77142fd4d4a93136fd0ac17348861f10e8f5d5f656fa9dacee08d8fcd2b1d8')
|
|
|
|
variant('build_type', default='Release', values=("Release", "Debug"), description='CMake build type')
|
|
|
|
def patch(self):
|
|
if '@4.2.0' in self.spec:
|
|
filter_file('${ROCM_PATH}/opencl',
|
|
self.spec['rocm-opencl'].prefix,
|
|
'amd_openvx/cmake/FindOpenCL.cmake',
|
|
string=True)
|
|
filter_file('/opt/rocm/mivisionx/include',
|
|
self.spec['mivisionx'].prefix.include,
|
|
'utilities/mv_deploy/CMakeLists.txt',
|
|
string=True)
|
|
|
|
def flag_handler(self, name, flags):
|
|
spec = self.spec
|
|
protobuf = spec['protobuf'].prefix.include
|
|
if name == 'cxxflags':
|
|
flags.append('-I{0}'.format(protobuf))
|
|
return (flags, None, None)
|
|
|
|
depends_on('cmake@3.5:', type='build')
|
|
depends_on('ffmpeg@:4', type='build')
|
|
depends_on('protobuf@:3', type='build')
|
|
depends_on('opencv@:3.4 +calib3d+core+features2d+highgui+imgcodecs+imgproc+video+videoio+flann+photo+objdetect', type='build')
|
|
depends_on('rocm-opencl@3.5.0', when='@1.7')
|
|
depends_on('rocm-cmake@3.5.0', type='build', when='@1.7')
|
|
depends_on('miopen-opencl@3.5.0', when='@1.7')
|
|
depends_on('miopengemm@1.1.6', when='@1.7')
|
|
depends_on('openssl', when='@4.0.0:4.2.0')
|
|
|
|
for ver in ['3.7.0', '3.8.0', '3.9.0', '3.10.0', '4.0.0', '4.1.0', '4.2.0',
|
|
'4.3.0']:
|
|
depends_on('rocm-opencl@' + ver, when='@' + ver)
|
|
depends_on('miopengemm@' + ver, when='@' + ver)
|
|
depends_on('miopen-opencl@' + ver, when='@' + ver)
|
|
|
|
def cmake_args(self):
|
|
spec = self.spec
|
|
protobuf = spec['protobuf'].prefix.include
|
|
args = [
|
|
self.define('CMAKE_CXX_FLAGS', '-I{0}'.format(protobuf))
|
|
]
|
|
return args
|