(git:6a2e663)
|
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 |
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 | ||
) |
...
opA | ... |
opB | ... |
m | ... |
n | ... |
k | ... |
alpha | ... |
A | ... |
lda | ... |
B | ... |
ldb | ... |
beta | ... |
C | ... |
ldc | ... |
ctx | ... |
Definition at line 70 of file local_gemm_api.F.
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
ctx | newly created context |
pu | processing unit to run the (s,d,c,z}dgemm |
Definition at line 151 of file local_gemm_api.F.
subroutine, public local_gemm_api::local_gemm_destroy | ( | type(c_ptr), intent(inout) | ctx | ) |
release resources associated to a gemm context
ctx | handle |
Definition at line 177 of file local_gemm_api.F.
subroutine, public local_gemm_api::local_gemm_set_op_threshold_gpu | ( | type(c_ptr) | ctx, |
integer, intent(in) | opThresholdGPU | ||
) |
...
ctx | ... |
opThresholdGPU | ... |
Definition at line 200 of file local_gemm_api.F.
subroutine, public local_gemm_api::local_gemm_set_library | ( | integer, intent(in) | dgemm_library | ) |
...
dgemm_library | ... |
Definition at line 219 of file local_gemm_api.F.
integer, parameter, public local_gemm_api::local_gemm_pu_host = 0 |
Definition at line 45 of file local_gemm_api.F.
integer, parameter, public local_gemm_api::local_gemm_pu_gpu = 1 |
Definition at line 45 of file local_gemm_api.F.