flashinfer_bench.data

This is the Python API for the FlashInfer Trace Schema. FlashInfer-Bench uses Pydantic to describe all schemas in the dataset. The following classes are defined:

  • The Definition class, which defines the kernel specification.

  • The Solution class, which defines the kernel implementation.

  • The Workload class, which defines the kernel’s input tensors.

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

  • The TraceSet class, which describes the whole dataset, including multiple definitions, solutions, workloads, and traces.