tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[BYOC] Refactor and update `RunCodegen` API #365

Closed masahi closed 1 year ago

masahi commented 1 year ago

A part of https://github.com/tlc-pack/relax/issues/364

Currently, RunCodegen pass sends one function at a time to an external backend. This prevents sharing some internal states in the backend compiler across compilation of different functions. For example, for CUTLASS BYOC, we want to cache the kernel profiling result to avoid profiling the same workload repeatedly.

Also makes the following changes:

cc @sunggg