import type {
  ExecutionV1GraphSchema,
  ExecutionV1SessionSchema,
  ExecutionV1PoliciesSchema,
  ExecutionV1ExpandedStepSchema,
} from '../../../config/schema/execution-v1.schema.js';
import type { KernelPathSchema } from '../../../config/schema/index.js';

export interface ExecutionV1LaneIntegrity {
  declared: {
    activationDtype: string | null;
    mathDtype: string | null;
    accumDtype: string | null;
    kvDtype: string | null;
  };
  executed: {
    activationDtype: string | null;
    mathDtype: string | null;
    accumDtype: string | null;
    kvDtype: string | null;
  };
  status: 'matches' | 'transformed';
  transforms: string[];
  policy?: {
    kind: string | null;
    dtypeEffect: string | null;
    reason: string | null;
    evidence: string[];
  } | null;
}

export interface ExecutionV1CompiledState {
  session: ExecutionV1SessionSchema;
  policies: ExecutionV1PoliciesSchema;
  resolvedSteps: {
    prefill: ExecutionV1ExpandedStepSchema[];
    decode: ExecutionV1ExpandedStepSchema[];
    all: ExecutionV1ExpandedStepSchema[];
  };
  runtimeInferencePatch: Record<string, unknown>;
  appliedTransforms?: string[];
  laneIntegrity?: ExecutionV1LaneIntegrity;
  fallbackKernelPath?: KernelPathSchema | null;
}

export declare function hasExecutionV1(
  manifestInference: { schema?: string | null; execution?: unknown }
): boolean;

export declare function compileExecutionV1(options?: {
  manifestInference: {
    schema: string;
    execution: ExecutionV1GraphSchema;
    session: ExecutionV1SessionSchema;
  };
  modelId?: string;
  numLayers?: number;
  headDim?: number | null;
  runtimeSession?: ExecutionV1SessionSchema | null;
  runtimeCompute?: Record<string, unknown> | null;
  kernelPathPolicy?: Record<string, unknown> | null;
  executionPatch?: Record<string, unknown> | null;
  capabilities?: {
    hasSubgroups?: boolean;
    hasF16?: boolean;
    hasSubgroupsF16?: boolean;
    maxWorkgroupSize?: number;
    maxBufferSize?: number;
  } | null;
  platform?: {
    id?: string;
    vendor?: string;
    architecture?: string;
  } | null;
}): ExecutionV1CompiledState;

export declare function applyExecutionV1RuntimeConfig(options?: {
  runtimeConfig: Record<string, unknown>;
  runtimeOverrides?: Record<string, unknown> | null;
  manifest: {
    inference?: { schema?: string; execution?: ExecutionV1GraphSchema; session?: ExecutionV1SessionSchema };
    modelId?: string;
    architecture?: { numLayers?: number };
  };
  modelId?: string;
  numLayers?: number;
}): {
  runtimeConfig: Record<string, unknown>;
  executionV1State: ExecutionV1CompiledState | null;
};