Module gemv

Module gemv 

Source
Expand description

General matrix-vector multiplication (GEMV) kernels for GPU

Implements y = alpha * A * x + beta * y where:

  • A is an M x N matrix
  • x is an N-dimensional vector
  • y is an M-dimensional vector
  • alpha and beta are scalar values

Structsยง

BatchGemvKernel
Batched GEMV kernel for processing multiple matrix-vector multiplications
GemvKernel
General matrix-vector multiplication kernel