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

New package: tiny tensor compiler (#44401)

Browse Source 
Signed-off-by: Carsten Uphoff <carsten.uphoff@intel.com>
This commit is contained in:
Carsten Uphoff 2024-05-28 19:27:44 +02:00 committed by GitHub
parent 00bcf935e8
commit 2bb20caa5f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 46 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
var/spack/repos/builtin/packages/tiny-tensor-compiler/package.py Normal file
View File

@ -0,0 +1,46 @@
# Copyright 2013-2024 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)
import os
from spack.package import *
class TinyTensorCompiler(CMakePackage):
"""A compiler for tensor computations on GPUs and other devices,
supporting the OpenCL, Level Zero, and SYCL runtime."""
homepage = "https://github.com/intel/tiny-tensor-compiler"
url = "https://github.com/intel/tiny-tensor-compiler/archive/refs/tags/v0.3.1.tar.gz"
maintainers("uphoffc")
license("BSD-3-Clause", checked_by="uphoffc")
version("0.3.1", sha256="e512b92f9ef8f21362ea4a8f2655338769bc7fcf9de543e3dc7db86b696695b3")
variant("shared", default=True, description="Shared library")
variant("level-zero", default=False, description="Build tinytc_ze (Level Zero runtime)")
variant("opencl", default=True, description="Build tintc_cl (OpenCL runtime)")
variant("sycl", default=False, description="Build tinytc_sycl (SYCL runtime)")
requires("+opencl +level-zero", when="+sycl")
depends_on("cmake@3.23.0:", type="build")
depends_on("double-batched-fft-library ~sycl ~level-zero ~opencl@0.5.1:", type="link")
depends_on("oneapi-level-zero@1.13:", when="+level-zero")
depends_on("opencl-icd-loader@2022.01.04:", when="+opencl", type="link")
def cmake_args(self):
cxx_compiler = os.path.basename(self.compiler.cxx)
if self.spec.satisfies("+sycl") and cxx_compiler not in ["icpx"]:
raise InstallError("The tinytc_sycl library requires the oneapi C++ compiler")
return [
self.define_from_variant("BUILD_SHARED_LIBS", "shared"),
self.define_from_variant("BUILD_SYCL", "sycl"),
self.define_from_variant("BUILD_LEVEL_ZERO", "level-zero"),
self.define_from_variant("BUILD_OPENCL", "opencl"),
]