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
Definitionclass, which defines the kernel specification.The
Solutionclass, which defines the kernel solution.The
Traceclass, which defines the kernel execution trace.The
TraceSetclass, which defines a set of kernel execution traces.