pulp-platform / snitch_cluster

An energy-efficient RISC-V floating-point compute cluster.
https://pulp-platform.github.io/snitch_cluster/
Apache License 2.0
51 stars 51 forks source link

GEMM improvements #123

Closed viv-eth closed 4 months ago

viv-eth commented 7 months ago

The configuration and function handling have been improved.

  1. The configuration now accepts the function signature as input, which will automatically call the correct kernel implementation.
  2. Furthermore, a new gemm_args_t struct is defined holding all necessary layer parameters. These parameters are transferred from DRAM to TCDM to avoid significant function setup overheads.
  3. The function signatures have been unified across all implementations.