Python API

The pages below are generated from Python source files at docs build time.

tools/

• tools.analyze_quant_ir
• tools.benchmark
• tools.build
• tools.chipyard
• tools.ci
• tools.compile
• tools.install_prebuilt
• tools.merlin
• tools.patches
• tools.ray_cmd
• tools.raycp.model
• tools.raycp.service
• tools.setup
• tools.strip_mlir_weights
• tools.targetgen.executor
• tools.targetgen.generator
• tools.targetgen.loader
• tools.targetgen.model
• tools.targetgen.orchestrator
• tools.targetgen.planner
• tools.targetgen.prompts
• tools.targetgen_cmd
• tools.utils

models/

• models.depth_anything_v2.depth_anything_v2_onnx
• models.diffusion.diffusion_policy
• models.dronet.dronet
• models.midas.midas_onnx
• models.mlp.mlp
• models.mlp_wide.mlp_wide
• models.opu_bench_suite.opu_bench_models
• models.quantize_models
• models.smolVLA.export_smolvla
• models.smolVLA.export_smolvla_int8
• models.tinydepth.tinydepth
• models.tinyllama.tinyllama_export
• models.yolov8_nano.yolov8nano_export

benchmarks/

• benchmarks.KernelBench.iree_compilation
• benchmarks.KernelBench.iree_run_subprocess
• benchmarks.SaturnNPU.kernel_library.attention
• benchmarks.SaturnNPU.kernel_library.attention_acc
• benchmarks.SaturnNPU.kernel_library.elementwise_add
• benchmarks.SaturnNPU.kernel_library.elementwise_div
• benchmarks.SaturnNPU.kernel_library.elementwise_mul
• benchmarks.SaturnNPU.kernel_library.elementwise_sub
• benchmarks.SaturnNPU.kernel_library.gelu_tanh
• benchmarks.SaturnNPU.kernel_library.matmul
• benchmarks.SaturnNPU.kernel_library.matmul_acc
• benchmarks.SaturnNPU.kernel_library.reduction_sum
• benchmarks.SaturnNPU.kernel_library.requant
• benchmarks.SaturnNPU.kernel_library.rms_norm
• benchmarks.SaturnNPU.kernel_library.rope_frequency
• benchmarks.SaturnNPU.kernel_library.silu
• benchmarks.SaturnNPU.kernel_library.smolvla_fused_attention
• benchmarks.SaturnNPU.kernel_library.softmax
• benchmarks.SaturnNPU.scripts.analyze_npu_graph
• benchmarks.SaturnNPU.scripts.export_golden_data
• benchmarks.SaturnNPU.scripts.extract_all_kernel_variants
• benchmarks.SaturnNPU.scripts.generate_kernel_catalog
• benchmarks.SaturnNPU.scripts.generate_mlir_golden_data
• benchmarks.SaturnNPU.scripts.generate_npu_golden_tests
• benchmarks.SaturnNPU.scripts.generate_pr_report
• benchmarks.SaturnNPU.scripts.plot_npu_coverage
• benchmarks.SaturnNPU.scripts.plot_sankey
• benchmarks.SaturnNPU.scripts.trace_layer_decomposition
• benchmarks.SaturnOPU.analyze_opu_coverage
• benchmarks.SaturnOPU.plot_firesim_results
• benchmarks.SaturnOPU.plot_opu_coverage
• benchmarks.SpacemiTX60.baseline_dual_model_async.plot_results
• benchmarks.SpacemiTX60.baseline_dual_model_async.reconstruct_timeline_plot
• benchmarks.SpacemiTX60.baseline_dual_model_async.run_benchmark_remote
• benchmarks.SpacemiTX60.parse_dual_model_log

samples/

• samples.SpacemiTX60.dispatch_scheduler.analysis.plot_dispatch_trace
• samples.research.model_computation_graph_generation.dot_dispatch_parser
• samples.research.model_computation_graph_generation.mlir_flow_parser
• samples.research.promise_schedule_multi_model.hw_annotation