g1y5x3 / gemm_optimization

optimize matrix multiplication using tinygrad's action space

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GEMM optimization

This repository performs similar GEMM optimization that was written by SIBOEHM. However, instead do it manually by writting all the kernels, it uses the abstraction layer and action space in tinygrad.

About

optimize matrix multiplication using tinygrad's action space


Languages

Language:Python 100.0%