Function read_starter_kernel

extract.py:246–259

Read the starter kernel file. Returns None if not found. For backend='triton': reads from kernels/{op_type}.py For backend='cuda': reads from kernels/cuda/{op_type}.py

(op_type: str, backend: str = "triton")

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

def read_starter_kernel(op_type: str, backend: str = "triton") -> Optional[str]:
    """Read the starter kernel file. Returns None if not found.

    For backend='triton': reads from kernels/{op_type}.py
    For backend='cuda':   reads from kernels/cuda/{op_type}.py
    """
    if backend == "cuda":
        path = os.path.join(KERNELS_DIR, "cuda", f"{op_type}.py")
    else:
        path = os.path.join(KERNELS_DIR, f"{op_type}.py")
    if not os.path.exists(path):
        return None
    with open(path, "r", encoding="utf-8") as f:
        return f.read()


def extract_kernel_body(starter_code: str) -> str: