Function main

export_hf.py:806–864

()

# ---------------------------------------------------------------------------

def main() -> None:
    parser = argparse.ArgumentParser(
        description=(
            "Export an optimized AutoKernel kernel to HuggingFace Kernels format. "
            "Supports both CUDA C++ and Triton backends."
        ),
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog=textwrap.dedent("""\
            examples:
              # Export the default kernel.py
              uv run export_hf.py --name my_matmul

              # Export a specific kernel file with repo ID
              uv run export_hf.py --name my_matmul --kernel workspace/kernel_matmul_1.py \\
                                  --repo-id rightnow-ai/matmul-kernel

              # Custom output directory
              uv run export_hf.py --name my_matmul --output /tmp/hf_export/
        """),
    )
    parser.add_argument(
        "--name",
        type=str,
        required=True,
        help=(
            "Name for the exported kernel project. Must be a valid Python identifier "
            "(e.g., 'my_matmul', 'fused_attention')."
        ),
    )
    parser.add_argument(
        "--kernel",
        type=str,
        default=DEFAULT_KERNEL_PATH,
        help=f"Path to the kernel file to export (default: kernel.py)",
    )
    parser.add_argument(
        "--output",
        type=str,
        default=DEFAULT_OUTPUT_DIR,
        help=f"Output directory for the HF Kernels project (default: workspace/hf_export/)",
    )
    parser.add_argument(
        "--repo-id",
        type=str,
        default=None,
        help=(
            "HuggingFace repo ID (e.g., 'rightnow-ai/matmul-kernel'). "
            "Used in documentation and usage examples."
        ),
    )

    args = parser.parse_args()

    export_kernel(
        kernel_path=args.kernel,
        name=args.name,
        output_dir=args.output,
        repo_id=args.repo_id,