DmitryLyakh / TAL_SH

Tensor Algebra Library Routines for Shared Memory Systems

TAL-SH: Tensor Algebra Library for Shared Memory Computers: Nodes equipped with multicore CPU, NVIDIA GPU, AMD GPU, and Intel Xeon Phi. The library implements basic tensor algebra operations with interfaces to C, C++11, and Fortran 90+. Author: Dmitry I. Lyakh (Liakh): [e-mail]

Copyright (C) 2014-2022 Dmitry I. Lyakh (Liakh) Copyright (C) 2014-2022 Oak Ridge National Laboratory (UT-Battelle)

LICENSE: BSD 3-Clause

API reference manual: DOC/TALSH_manual.txt

BUILD: Modify the header of the Makefile accordingly and run make. If your system has MODULES, look up relevant paths via "module show", otherwise find the paths to BLAS and CUDA yourself. If you use ESSL and/or IBM XL compiler, you will need IBM XL paths as well. On Cray systems, you may also need to activate dynamic linking explicitly: export XTPE_LINK_TYPE = dynamic export CRAYPE_LINK_TYPE = dynamic

Typical build configuration examples:

GNU compiler, OpenBLAS:

$ export BLASLIB = OPENBLAS
$ export PATH_BLAS_OPENBLAS = /usr/local/blas/openblas/lib
$ export PATH_LAPACK_LIB = $PATH_BLAS_OPENBLAS
$ make

GNU compiler, MKL (Intel CPU or Intel Xeon Phi):

$ export BLASLIB = MKL
$ export PATH_INTEL = /opt/intel
$ export PATH_BLAS_MKL = $PATH_INTEL/mkl/lib/intel64
$ export PATH_LAPACK_LIB = $PATH_BLAS_MKL
$ make

GNU compiler, OpenBLAS, CUDA (NVIDIA GPU):

$ export BLASLIB = OPENBLAS
$ export PATH_BLAS_OPENBLAS = /usr/local/blas/openblas/lib
$ export PATH_LAPACK_LIB = $PATH_BLAS_OPENBLAS
$ export GPU_CUDA = CUDA
$ export GPU_SM_ARCH = 70
$ export PATH_CUDA = /usr/local/cuda
$ make

GNU compiler, OpenBLAS, ROCM (AMD GPU):

$ export BLASLIB = OPENBLAS
$ export PATH_BLAS_OPENBLAS = /usr/local/blas/openblas/lib
$ export PATH_LAPACK_LIB = $PATH_BLAS_OPENBLAS
$ export GPU_CUDA = CUDA
$ export USE_HIP = YES
$ export PATH_ROCM = /opt/rocm
$ make

EXAMPLES: A number of examples is available in test.cpp and main.F90.