Index _ | A | B | C | D | E | F | G | H | I | L | M | N | O | P | R | S | T | U | V | W _ __init__() (flashinfer_bench.compile.Builder method) (flashinfer_bench.compile.BuilderRegistry method) (flashinfer_bench.compile.BuildError method) (flashinfer_bench.compile.builders.PythonBuilder method) (flashinfer_bench.compile.builders.TorchBuilder method) (flashinfer_bench.compile.builders.TritonBuilder method) (flashinfer_bench.compile.builders.TVMFFIBuilder method) (flashinfer_bench.compile.Runnable method) __new__() (flashinfer_bench.compile.BuildError method) (flashinfer_bench.data.definition.DType method) (flashinfer_bench.data.EvaluationStatus method) (flashinfer_bench.data.SupportedLanguages method) A add_traces() (flashinfer_bench.data.TraceSet method) add_workload_blob_tensor() (flashinfer_bench.data.TraceSet method) add_workload_traces() (flashinfer_bench.data.TraceSet method) apply() (in module flashinfer_bench) author (flashinfer_bench.data.Solution attribute) axes (flashinfer_bench.data.Definition attribute) (flashinfer_bench.data.Workload attribute) B backup_traces() (flashinfer_bench.data.TraceSet method) BFLOAT16 (flashinfer_bench.data.definition.DType attribute) binding (flashinfer_bench.data.BuildSpec attribute) BOOL (flashinfer_bench.data.definition.DType attribute) build() (flashinfer_bench.compile.Builder method) (flashinfer_bench.compile.BuilderRegistry method) (flashinfer_bench.compile.builders.PythonBuilder method) (flashinfer_bench.compile.builders.TorchBuilder method) (flashinfer_bench.compile.builders.TritonBuilder method) (flashinfer_bench.compile.builders.TVMFFIBuilder method) build_reference() (flashinfer_bench.compile.BuilderRegistry method) build_type (flashinfer_bench.compile.RunnableMetadata attribute) Builder (class in flashinfer_bench.compile) BuilderRegistry (class in flashinfer_bench.compile) BuildError C call_destination_passing() (flashinfer_bench.compile.Runnable method) call_kwargs() (flashinfer_bench.compile.Runnable method) call_value_returning() (flashinfer_bench.compile.Runnable method) can_build() (flashinfer_bench.compile.Builder method) (flashinfer_bench.compile.builders.PythonBuilder method) (flashinfer_bench.compile.builders.TorchBuilder method) (flashinfer_bench.compile.builders.TritonBuilder method) (flashinfer_bench.compile.builders.TVMFFIBuilder method) cleanup() (flashinfer_bench.compile.BuilderRegistry method) (flashinfer_bench.compile.Runnable method) COMPILE_ERROR (flashinfer_bench.data.EvaluationStatus attribute) const_axes (flashinfer_bench.data.Definition property) constraints (flashinfer_bench.data.Definition attribute) content (flashinfer_bench.data.SourceFile attribute) correctness (flashinfer_bench.data.Evaluation attribute) CPP (flashinfer_bench.data.SupportedLanguages attribute) CUDA (flashinfer_bench.data.SupportedLanguages attribute) D definition (flashinfer_bench.compile.RunnableMetadata attribute) (flashinfer_bench.data.Solution attribute) (flashinfer_bench.data.Trace attribute) definition_name (flashinfer_bench.compile.RunnableMetadata attribute) definitions (flashinfer_bench.data.TraceSet attribute) dependencies (flashinfer_bench.data.BuildSpec attribute) description (flashinfer_bench.data.AxisConst attribute) (flashinfer_bench.data.AxisVar attribute) (flashinfer_bench.data.Definition attribute) (flashinfer_bench.data.Solution attribute) (flashinfer_bench.data.TensorSpec attribute) destination_passing_style (flashinfer_bench.compile.RunnableMetadata attribute) (flashinfer_bench.data.BuildSpec attribute) disable_apply() (in module flashinfer_bench) disable_tracing() (in module flashinfer_bench) DType (class in flashinfer_bench.data.definition) dtype (flashinfer_bench.data.TensorSpec attribute) E enable_apply() (in module flashinfer_bench) enable_tracing() (in module flashinfer_bench) entry_point (flashinfer_bench.data.BuildSpec attribute) environment (flashinfer_bench.data.Evaluation attribute) evaluation (flashinfer_bench.data.Trace attribute) EvaluationStatus (class in flashinfer_bench.data) extra (flashinfer_bench.data.Correctness attribute) extract_solution_to_files() (in module flashinfer_bench.agents) F FFI_PROMPT (in module flashinfer_bench.agents.ffi_prompt) FFI_PROMPT_SIMPLE (in module flashinfer_bench.agents.ffi_prompt) filter_traces() (flashinfer_bench.data.TraceSet method) flashinfer_bench.agents.ffi_prompt module flashinfer_bench_list_ncu_options() (in module flashinfer_bench.agents) flashinfer_bench_run_ncu() (in module flashinfer_bench.agents) flashinfer_bench_run_sanitizer() (in module flashinfer_bench.agents) FLOAT16 (flashinfer_bench.data.definition.DType attribute) FLOAT32 (flashinfer_bench.data.definition.DType attribute) FLOAT4_E2M1 (flashinfer_bench.data.definition.DType attribute) FLOAT8_E4M3FN (flashinfer_bench.data.definition.DType attribute) FLOAT8_E5M2 (flashinfer_bench.data.definition.DType attribute) from_path() (flashinfer_bench.data.TraceSet class method) function_to_schema() (in module flashinfer_bench.agents) G get_all_tool_schemas() (in module flashinfer_bench.agents) get_axes_values() (flashinfer_bench.data.Definition method) get_axes_values_from_inputs() (flashinfer_bench.data.Definition method) get_best_trace() (flashinfer_bench.data.TraceSet method) get_entry_path() (flashinfer_bench.data.Solution method) get_entry_source() (flashinfer_bench.data.Solution method) get_entry_symbol() (flashinfer_bench.data.Solution method) get_input_shapes() (flashinfer_bench.data.Definition method) get_instance() (flashinfer_bench.compile.BuilderRegistry class method) get_output_shapes() (flashinfer_bench.data.Definition method) get_solution() (flashinfer_bench.data.TraceSet method) get_workload_blob_tensor() (flashinfer_bench.data.TraceSet method) H hardware (flashinfer_bench.data.Environment attribute) hash() (flashinfer_bench.data.Solution method) I INCORRECT_DTYPE (flashinfer_bench.data.EvaluationStatus attribute) INCORRECT_NUMERICAL (flashinfer_bench.data.EvaluationStatus attribute) INCORRECT_SHAPE (flashinfer_bench.data.EvaluationStatus attribute) inputs (flashinfer_bench.data.Definition attribute) (flashinfer_bench.data.Workload attribute) InputSpec (in module flashinfer_bench.data) INT16 (flashinfer_bench.data.definition.DType attribute) INT32 (flashinfer_bench.data.definition.DType attribute) INT64 (flashinfer_bench.data.definition.DType attribute) INT8 (flashinfer_bench.data.definition.DType attribute) is_available() (flashinfer_bench.compile.Builder static method) (flashinfer_bench.compile.builders.PythonBuilder static method) (flashinfer_bench.compile.builders.TorchBuilder static method) (flashinfer_bench.compile.builders.TritonBuilder static method) (flashinfer_bench.compile.builders.TVMFFIBuilder static method) is_successful() (flashinfer_bench.data.Trace method) is_workload_trace() (flashinfer_bench.data.Trace method) L language (flashinfer_bench.data.BuildSpec attribute) latency_ms (flashinfer_bench.data.Performance attribute) libs (flashinfer_bench.data.Environment attribute) log (flashinfer_bench.data.Evaluation attribute) M max_absolute_error (flashinfer_bench.data.Correctness attribute) max_relative_error (flashinfer_bench.data.Correctness attribute) merge_kwargs_to_args() (flashinfer_bench.data.Definition method) metadata (flashinfer_bench.compile.Runnable attribute) misc (flashinfer_bench.compile.RunnableMetadata attribute) model_post_init() (flashinfer_bench.data.Solution method) module flashinfer_bench.agents.ffi_prompt N name (flashinfer_bench.data.Definition attribute) (flashinfer_bench.data.Solution attribute) O op_type (flashinfer_bench.data.Definition attribute) outputs (flashinfer_bench.data.Definition attribute) P pack_solution_from_files() (in module flashinfer_bench.agents) PASSED (flashinfer_bench.data.EvaluationStatus attribute) path (flashinfer_bench.data.SafetensorsInput attribute) (flashinfer_bench.data.SourceFile attribute) performance (flashinfer_bench.data.Evaluation attribute) PYTHON (flashinfer_bench.data.SupportedLanguages attribute) PythonBuilder (class in flashinfer_bench.compile.builders) R reference (flashinfer_bench.data.Definition attribute) reference_latency_ms (flashinfer_bench.data.Performance attribute) root (flashinfer_bench.data.TraceSet attribute) Runnable (class in flashinfer_bench.compile) RunnableMetadata (class in flashinfer_bench.compile) RUNTIME_ERROR (flashinfer_bench.data.EvaluationStatus attribute) S shape (flashinfer_bench.data.TensorSpec attribute) solution (flashinfer_bench.data.Trace attribute) solution_name (flashinfer_bench.compile.RunnableMetadata attribute) solutions (flashinfer_bench.data.TraceSet attribute) sources (flashinfer_bench.data.Solution attribute) spec (flashinfer_bench.data.Solution attribute) speedup_factor (flashinfer_bench.data.Performance attribute) status (flashinfer_bench.data.Evaluation attribute) summary() (flashinfer_bench.data.TraceSet method) SupportedLanguages (class in flashinfer_bench.data) T tags (flashinfer_bench.data.Definition attribute) target_hardware (flashinfer_bench.data.BuildSpec attribute) tensor_key (flashinfer_bench.data.SafetensorsInput attribute) TILELANG (flashinfer_bench.data.SupportedLanguages attribute) TIMEOUT (flashinfer_bench.data.EvaluationStatus attribute) timestamp (flashinfer_bench.data.Evaluation attribute) to_dict() (flashinfer_bench.data.TraceSet method) torch_input_dtypes (flashinfer_bench.data.Definition property) torch_output_dtypes (flashinfer_bench.data.Definition property) TorchBuilder (class in flashinfer_bench.compile.builders) traces (flashinfer_bench.data.TraceSet attribute) TraceSet (class in flashinfer_bench.data) TRITON (flashinfer_bench.data.SupportedLanguages attribute) TritonBuilder (class in flashinfer_bench.compile.builders) TVMFFIBuilder (class in flashinfer_bench.compile.builders) type (flashinfer_bench.data.AxisConst attribute) (flashinfer_bench.data.AxisVar attribute) (flashinfer_bench.data.RandomInput attribute) (flashinfer_bench.data.SafetensorsInput attribute) (flashinfer_bench.data.ScalarInput attribute) U uuid (flashinfer_bench.data.Workload attribute) V value (flashinfer_bench.data.AxisConst attribute) (flashinfer_bench.data.ScalarInput attribute) var_axes (flashinfer_bench.data.Definition property) var_axes_bindings (flashinfer_bench.data.Definition property) W workload (flashinfer_bench.data.Trace attribute) workloads (flashinfer_bench.data.TraceSet attribute)
FlashInfer-Bench