Function extract_function_name_from_compile

export_hf.py:139–152

Extract the function name passed to compile_cuda(). Looks for patterns like: compile_cuda(CUDA_SRC, "matmul_cuda") compile_cuda(CUDA_SRC, "softmax_cuda")

(source: str)

def extract_function_name_from_compile(source: str) -> Optional[str]:
    """
    Extract the function name passed to compile_cuda().

    Looks for patterns like:
      compile_cuda(CUDA_SRC, "matmul_cuda")
      compile_cuda(CUDA_SRC, "softmax_cuda")
    """
    match = re.search(
        r'compile_cuda\s*\(\s*CUDA_SRC\s*,\s*["\'](\w+)["\']', source
    )
    if match:
        return match.group(1)
    return None


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