Module gemm

scirs2_core::gpu::kernels::blas

Module gemm 

Source
Expand description

General matrix-matrix multiplication (GEMM) kernels for GPU

Implements C = alpha * A * B + beta * C where:

  • A is an M x K matrix
  • B is a K x N matrix
  • C is an M x N matrix
  • alpha and beta are scalar values

Structs§

GemmKernel
General matrix-matrix multiplication kernel

Enums§

GemmImpl
GEMM specialized implementation type