Skip to content

Krish120003/gemm-rust

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GEMM implemented in Rust

Ths is an implementation of GEMM in Rust. The goal is to optimize it to be somewhat competitive with numpy.

Usage

$ python3 baseline.py # requires numpy, to generate a baseline for correctness
$ cargo run --release

Current performance

These performance numbers are for a 1024x1024 matrix multiplication on a 10 Core M1 Pro MacBook Pro.

Both Numpy and Rust are limited to a single core to compare performance.

Numpy: 72.3 GFLOP/s
Rust: 0.95 GFLOP/s

(Long way to go!)

About

An implementation of general matrix multiplication in Rust.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors