Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.
/ TileGraph Public archive

TileGraph is an experimental DNN compiler that utilizes static code generation and kernel fusion techniques.

License

Notifications You must be signed in to change notification settings

KuangjuX/TileGraph

Repository files navigation

TileGraph

Note: TiledKernel is no longer an active project, and the successor is ThrillerFlow.

TileGraph is an experimental DNN compiler that utilizes static code generation and kernel fusion techniques.

设计文档 | Design Docs

Overview

Get Started

Clone Project

git clone [email protected]:KuangjuX/TileGraph.git
git submodule update --init --recursive

Install Dependencies

cd 3rd-party/fmt && mkdir build && cd build && cmake -DCMAKE_POSITION_INDEPENDENT_CODE=TRUE .. && make

Test

make test

Run Examples

make examples

Reference Projects

  • InfiniTensor/RefactorGraph: 分层解耦的深度学习推理引擎
  • microsoft/nnfusion: A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.

Reference Papers

  • BOLT: BRIDGING THE GAP BETWEEN AUTO-TUNERS AND HARDWARE-NATIVE PERFORMANCE(MLSys'22)
  • Welder: Scheduling Deep Learning Memory Access via Tile-graph(OSDI'23)
  • Graphene: An IR for Optimized Tensor Computations on GPUs(ASPLOS'23)

About

TileGraph is an experimental DNN compiler that utilizes static code generation and kernel fusion techniques.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages