ybai62868 / AutoBench4TensorComputation

automatic benchmark of tensor program generation and optimization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Auto-Bench-for-TensorComputation-Research

[PRs Welcome]

Benchmark scripts for Tensor Compilers and Popular Frameworks

Background

Currently, there are lots of deep learning frameworks and compilers for the performance improvement.

When I want to know the specific optimization difference on a backend, it is difficult to find a tool which can do the automatic generation of tensor programs and testing to make benchmarking easier.

Our Solution

About

automatic benchmark of tensor program generation and optimization

License:MIT License


Languages

Language:Python 85.4%Language:Cuda 13.1%Language:Shell 1.4%