(git:6a2e663)
local_gemm_api Module Reference

Functions/Subroutines

subroutine, public local_gemm (opA, opB, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc, ctx)
 ... More...
 
subroutine, public local_gemm_create (ctx, pu)
 create a context for handling gemm offloading More...
 
subroutine, public local_gemm_destroy (ctx)
 release resources associated to a gemm context More...
 
subroutine, public local_gemm_set_op_threshold_gpu (ctx, opThresholdGPU)
 ... More...
 
subroutine, public local_gemm_set_library (dgemm_library)
 ... More...
 

Variables

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

Function/Subroutine Documentation

◆ local_gemm()

subroutine, public local_gemm_api::local_gemm ( character, intent(in)  opA,
character, intent(in)  opB,
integer, intent(in)  m,
integer, intent(in)  n,
integer, intent(in)  k,
real(8), intent(in)  alpha,
real(8), dimension(:, :), intent(in), target  A,
integer, intent(in)  lda,
real(8), dimension(:, :), intent(in), target  B,
integer, intent(in)  ldb,
real(8), intent(in)  beta,
real(8), dimension(:, :), intent(inout), target  C,
integer, intent(in)  ldc,
type(c_ptr), intent(inout), optional  ctx 
)

...

Parameters
opA...
opB...
m...
n...
k...
alpha...
A...
lda...
B...
ldb...
beta...
C...
ldc...
ctx...

Definition at line 70 of file local_gemm_api.F.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ local_gemm_create()

subroutine, public local_gemm_api::local_gemm_create ( type(c_ptr), intent(out)  ctx,
integer, intent(in)  pu 
)

create a context for handling gemm offloading

Parameters
ctxnewly created context
puprocessing unit to run the (s,d,c,z}dgemm

Definition at line 151 of file local_gemm_api.F.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ local_gemm_destroy()

subroutine, public local_gemm_api::local_gemm_destroy ( type(c_ptr), intent(inout)  ctx)

release resources associated to a gemm context

Parameters
ctxhandle

Definition at line 177 of file local_gemm_api.F.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ local_gemm_set_op_threshold_gpu()

subroutine, public local_gemm_api::local_gemm_set_op_threshold_gpu ( type(c_ptr)  ctx,
integer, intent(in)  opThresholdGPU 
)

...

Parameters
ctx...
opThresholdGPU...

Definition at line 200 of file local_gemm_api.F.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ local_gemm_set_library()

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

...

Parameters
dgemm_library...

Definition at line 219 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 45 of file local_gemm_api.F.

◆ local_gemm_pu_gpu

integer, parameter, public local_gemm_api::local_gemm_pu_gpu = 1

Definition at line 45 of file local_gemm_api.F.