import { type Buffer } from '../device.ts';
import { DefaultMap, type Metadata } from '../helpers/helpers.ts';
import { type UPatInput } from '../ops.ts';
import { PatternMatcher, UOp, UPat, type Variable } from '../ops.ts';
import { ShapeTracker } from '../shape/shapetracker.ts';
export declare const tensor_uop_spec: PatternMatcher<unknown, boolean>;
export declare class ScheduleItem {
    ast: UOp;
    bufs: Buffer[];
    metadata: Metadata[];
    constructor(ast: UOp, bufs: Buffer[], metadata: Metadata[]);
    /**
     * Read/write || write only buffers in the schedule.
     */
    get outputs(): Buffer[];
    /**
     * Read only buffers in the schedule.
     */
    get inputs(): Buffer[];
    output_idxs: () => number[];
    toString: () => string;
}
export declare class ScheduleContext {
    tensor_uops: Map<UOp, UOp[]>;
    var_vals: Map<UOp, number>;
    assigns: Set<UOp>;
    realizes: Map<UOp, UOp>;
    allbufs: Map<UOp, UOp>;
    ops_metadata: Map<UOp, Metadata>;
    contiguous: Map<UOp, UOp>;
    children: DefaultMap<UOp, Map<UOp, undefined>>;
    preloads: DefaultMap<Buffer<object>, Map<UOp, undefined>>;
    becomes_map: Map<UOp, UOp>;
    constructor(tensor_uops?: Map<UOp, UOp[]>, // this maps BUFFER uops of this schedule to the underlying lazybuffer
    var_vals?: Map<UOp, number>, // this maps a BIND's DEFINE_VAR to its value
    assigns?: Set<UOp>, // this holds all the BUFFER uops we ASSIGN to in this schedule
    realizes?: Map<UOp, UOp>, // this holds all the BUFFER uops we mutate in this schedule
    allbufs?: Map<UOp, UOp>, // this maps BUFFER uops the actual op
    ops_metadata?: Map<UOp, Metadata>, // this maps fused ops to Metadata
    contiguous?: Map<UOp, UOp>, // this maps roots to places they are made contiguous
    children?: DefaultMap<UOp, Map<UOp, undefined>>, preloads?: DefaultMap<Buffer<object>, Map<UOp, undefined>>, becomes_map?: Map<UOp, UOp>);
}
export declare const apply_swizzle: (u: UOp) => UOp;
export declare const view_right: PatternMatcher<unknown, UOp | undefined>;
export declare class ScheduleItemContext {
    var_vals: Map<UOp, number>;
    sts: Set<ShapeTracker>;
    bufs: UOp[];
    constructor(var_vals?: Map<UOp, number>, sts?: Set<ShapeTracker>, bufs?: UOp[]);
}
export declare const _append_st_vars: (ctx: ScheduleItemContext, x: UOp) => UOp | undefined;
export declare const _append_buf: (ctx: ScheduleItemContext, x: UOp) => UOp;
export declare const to_si: PatternMatcher<ScheduleItemContext, UOp | undefined>;
export declare const multioutput: PatternMatcher<Map<UOp, UOp>, UOp | undefined>;
export declare const PROCESS_REPLAY_CAPTURE: Map<string, Uint8Array<ArrayBufferLike>>;
export declare const is_scheduled: (u: UOp) => boolean;
export declare const uval: (u: UOp) => UOp;
/**
 * recursively search the uop for groupable children, realize the UOp if a child can't group
 */
export declare const recursive_group: (tr: UOp, st: ShapeTracker, r: UOp, children: DefaultMap<UOp, Map<UOp, undefined>>, allbufs: Map<UOp, UOp>, realizes: Map<UOp, UOp>, reduce_for_op: Map<UOp, UOp>, group: Map<UOp, undefined>, cache: Map<[UOp, ShapeTracker], undefined>) => undefined;
export declare const get_isolated_children: (r: UOp, reduce_for_op: Map<UOp, UOp>, children: DefaultMap<UOp, Map<UOp, undefined>>, allbufs: Map<UOp, UOp>, realizes: Map<UOp, UOp>, group: Map<UOp, undefined>) => Map<UOp, undefined>;
/**
 * search the big graph for all the reduceops that need to realize, sometimes group/fuse the reduceop
 */
export declare const group_realizes: (ctx: ScheduleContext) => UOp[][];
export declare const sym: PatternMatcher<unknown, UOp | undefined>;
export declare class UPatScheduled extends UPat {
    constructor(args?: Partial<UPatInput>);
}
export declare const realize_before_view: (ctx: ScheduleContext, view: UOp, src: UOp, b: UOp) => undefined;
export declare const create_subbuffer: (base: UOp, b: UOp, root: UOp, x: UOp) => UOp | undefined;
export declare const do_realize: PatternMatcher<ScheduleContext, UOp | undefined>;
export declare const break_sched: PatternMatcher<ScheduleContext, UOp | undefined>;
export declare const create_ctx: PatternMatcher<ScheduleContext, UOp | undefined>;
export declare const remove_movement_ops: PatternMatcher<unknown, UOp | undefined>;
export declare const create_schedule_with_vars: (big_sink: UOp, skip_check?: boolean) => [ScheduleItem[], Map<Variable, number>, Map<UOp, UOp>];