(git:c5411e0)
Loading...
Searching...
No Matches
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)
 ...
 
subroutine, public local_gemm_create (ctx, pu)
 create a context for handling gemm offloading
 
subroutine, public local_gemm_destroy (ctx)
 release resources associated to a gemm context
 
subroutine, public local_gemm_set_op_threshold_gpu (ctx, opthresholdgpu)
 ...
 
subroutine, public local_gemm_set_library (dgemm_library)
 ...
 

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.