FlashInfer Trace Schema Python APIΒΆ

FlashInfer-Bench provides a schema for the FlashInfer Trace database. This document includes the Python API for the schema, including

  • The Definition class, which defines the kernel specification.

  • The Solution class, which defines the kernel solution.

  • The Trace class, which defines the kernel execution trace.

  • The TraceSet class, which defines a set of kernel execution traces.