carlushuang / Tensile

Stretching GPU performance for GEMMs and tensor contractions.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A tool for creating a benchmark-driven backend library for GEMMs, GEMM-like problems (such as batched GEMM), N-dimensional tensor contractions, and anything else that multiplies two multi-dimensional objects together on a GPU.

See Tensile Wiki for documentation.

About

Stretching GPU performance for GEMMs and tensor contractions.

License:MIT License


Languages

Language:Python 83.5%Language:C++ 12.8%Language:CMake 3.1%Language:Makefile 0.2%Language:Shell 0.2%Language:C 0.2%Language:Awk 0.0%