Function export_kernel

export_hf.py:674–799

Main export pipeline. Parameters ---------- kernel_path : str Path to the AutoKernel kernel file (kernel.py or similar). name : str Name for the exported kernel project (used in build.toml, module name). Must be a valid Python identifier. output_dir

(
    kernel_path: str,
    name: str,
    output_dir: str,
    repo_id: Optional[str] = None,
)

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

def export_kernel(
    kernel_path: str,
    name: str,
    output_dir: str,
    repo_id: Optional[str] = None,
) -> str:
    """
    Main export pipeline.

    Parameters
    ----------
    kernel_path : str
        Path to the AutoKernel kernel file (kernel.py or similar).
    name : str
        Name for the exported kernel project (used in build.toml, module name).
        Must be a valid Python identifier.
    output_dir : str
        Directory where the HF Kernels project will be created.
    repo_id : str, optional
        HuggingFace repo ID (e.g., "rightnow-ai/matmul-kernel").
        Used in documentation and __init__.py usage examples.

    Returns
    -------
    str
        Path to the exported project directory.
    """
    # Validate name is a valid Python identifier
    if not name.isidentifier():
        print(f"ERROR: '{name}' is not a valid Python identifier.")
        print("       Use a name like 'my_matmul' or 'fused_attention'.")
        sys.exit(1)

    # Default repo_id
    if repo_id is None:
        repo_id = f"your-username/{name}"

    # Read the kernel file
    if not os.path.exists(kernel_path):
        print(f"ERROR: Kernel file not found: {kernel_path}")
        sys.exit(1)

    with open(kernel_path, "r", encoding="utf-8") as f:
        source = f.read()

    if not source.strip():
        print(f"ERROR: Kernel file is empty: {kernel_path}")
        sys.exit(1)

    # Detect backend
    backend = detect_backend(source)
    kernel_type = detect_kernel_type(source)

    print(f"=== AutoKernel HuggingFace Kernels Export ===")
    print()
    print(f"  Kernel file:  {kernel_path}")
    print(f"  Backend:      {backend}")
    if kernel_type: