linear.go 301 B

12345678910111213
  1. //go:build !cuda
  2. // Package matmul provides matrix multiplication operations
  3. package matmul
  4. import (
  5. "makarna/pkg/backend/cpu"
  6. )
  7. // Linear dispatches to the CPU implementation when CUDA is disabled.
  8. func Linear(input, weight, output *cpu.Tensor) error {
  9. return linearCPU(input, weight, output)
  10. }