Module kernel 

Consolidated kernel splitting and pipeline orchestration.

This module contains:

• KernelContext for tracking state during kernel splitting
• split_store for splitting computation at STORE boundaries
• run_kernel_split_pipeline for full pipeline orchestration
• PcontigConfig for partial contiguous buffer removal
• Two-stage reduction splitting (split_reduceop)

Consolidated from: kernel_context.rs, split_kernel.rs, pipeline.rs, buffer_cost.rs, split_reduceop.rs

Structs

KernelAstMarker

Marker metadata for kernel AST SINKs.

KernelContext

Context for tracking state during kernel splitting.

LocalAddBufferContext

Per-kernel context for tracking state during kernel splitting.

PcontigConfig

Configuration for partial contiguous optimization.

SplitReduceOpConfig

Configuration for split_reduceop optimization.

Functions

collect_range_ids

Extract all RANGE axis IDs from a UOp tree.

run_kernel_split_pipeline

Run the kernel splitting pipeline.

split_reduceop

Split large REDUCE_AXIS into two stages.

split_store

Split STORE and END operations into individual kernels.