(git:f2099e5)
Loading...
Searching...
No Matches
local_gemm_api Module Reference

Data Types

type  local_gemm_ctxt_type

Functions/Subroutines

subroutine local_dgemm (opa, opb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc, ctx)
 Local GEMM on contiguous arrays, using BLAS or the configured SPLA backend. Each concurrent caller must own its context. No distributed matrix metadata is used.
subroutine, public local_gemm_set_library (dgemm_library)
 Select the backend for subsequent local GEMM calls and context creation.

Variables

integer, parameter, public local_gemm_pu_host = 0
integer, parameter, public local_gemm_pu_gpu = 1

Function/Subroutine Documentation

◆ local_dgemm()

subroutine local_gemm_api::local_dgemm ( character, intent(in) opa,
character, intent(in) opb,
integer, intent(in) m,
integer, intent(in) n,
integer, intent(in) k,
real (kind=dp), intent(in) alpha,
real (kind=dp), dimension(lda, *), intent(in), target a,
integer, intent(in) lda,
real (kind=dp), dimension(ldb, *), intent(in), target b,
integer, intent(in) ldb,
real (kind=dp), intent(in) beta,
real (kind=dp), dimension(ldc, *), intent(inout), target c,
integer, intent(in) ldc,
class(local_gemm_ctxt_type), intent(inout) ctx )

Local GEMM on contiguous arrays, using BLAS or the configured SPLA backend. Each concurrent caller must own its context. No distributed matrix metadata is used.

Parameters
opAoperation on A (N/T/C, case insensitive)
opBoperation on B (N/T/C, case insensitive)
moutput rows
noutput columns
kcontraction dimension
alphaproduct scale
Aleft operand
ldaleading dimension of A
Bright operand
ldbleading dimension of B
betaoutput scale
Coutput, must not overlap A or B
ldcleading dimension of C
ctxcaller-owned context

Definition at line 79 of file local_gemm_api.F.

Here is the call graph for this function:

◆ local_gemm_set_library()

subroutine, public local_gemm_api::local_gemm_set_library ( integer, intent(in) dgemm_library)

Select the backend for subsequent local GEMM calls and context creation.

Parameters
dgemm_librarybackend selector from input_constants (SPLA or BLAS)

Definition at line 271 of file local_gemm_api.F.

Here is the caller graph for this function:

Variable Documentation

◆ local_gemm_pu_host

integer, parameter, public local_gemm_api::local_gemm_pu_host = 0

Definition at line 41 of file local_gemm_api.F.

◆ local_gemm_pu_gpu

integer, parameter, public local_gemm_api::local_gemm_pu_gpu = 1

Definition at line 41 of file local_gemm_api.F.