MCPcopy Create free account
hub / github.com/evilsocket/cake / matmul

Method matmul

cake-core/src/backends/mod.rs:185–187  ·  view source on GitHub ↗

Matrix multiplication. GPU backends override to use accelerated matmul. Default: delegates to candle's CPU/CUDA/Metal matmul.

(&self, a: &Tensor, b: &Tensor)

Source from the content-addressed store, hash-verified

183 /// Matrix multiplication. GPU backends override to use accelerated matmul.
184 /// Default: delegates to candle's CPU/CUDA/Metal matmul.
185 fn matmul(&self, a: &Tensor, b: &Tensor) -> Result<Tensor> {
186 a.matmul(b)
187 }
188
189 /// Pre-process a linear weight for optimal matmul performance.
190 /// Called once at model load time. Backends can pre-convert dtype,

Callers 1

linear_forwardMethod · 0.45

Implementers 5

mod.rscake-core/src/backends/rocm/mod.rs
mod.rscake-core/src/backends/cpu/mod.rs
mod.rscake-core/src/backends/metal/mod.rs
mod.rscake-core/src/backends/vulkan/mod.rs
mod.rscake-core/src/backends/cuda/mod.rs

Calls

no outgoing calls

Tested by

no test coverage detected