Skip to content

README.md

Latest commit

 

History

History
40 lines (30 loc) · 896 Bytes

README.md

File metadata and controls

40 lines (30 loc) · 896 Bytes

YATeTo

It is Yet Another Tensor Toolbox for discontinuous Galerkin methods and other applications. You can find much more information about the package in this paper.

Installation

pip install -e .

or include it into your project as a submodule.

Usage

from yateto import *

...
def add(g):
  N = 8
  A = Tensor('A', (N, N))
  B = Tensor('B', (N, N, N))
  w = Tensor('w', (N,))
  C = Tensor('C', (N, N))

  kernel = C['ij'] <= 2.0 * C['ij'] + A['lj'] * B['ikl'] * w['k']
  g.add(name='kernel', ast=kernel)

# 'd' - double precision; 'hsw' - haswell-like architecture
arch = useArchitectureIdentifiedBy("dhsw")
generator = Generator(arch)
add(generator)
generator.generate(output_dir, gemm_cfg=GeneratorCollection([LIBXSMM(arch), Eigen(arch)]))
...

Note that the interfaces Yateto generates require C++17.