gemv_f32_tiled_generic.go 291 B

1234567891011121314151617
  1. //go:build !amd64
  2. package matmul
  3. import "makarna/pkg/backend/cpu"
  4. func gemvFloat32Range(out, aRow, b []float32, K, startN, endN int) {
  5. if startN >= endN {
  6. return
  7. }
  8. bOff := startN * K
  9. for n := startN; n < endN; n++ {
  10. out[n] = cpu.DotFloat32(aRow, b[bOff:bOff+K])
  11. bOff += K
  12. }
  13. }