@dexcost/sdk 0.2.0

package/dist/core/gpu-accountant.d.ts

@@ -0,0 +1,104 @@
+/**
+ * Per-task GPU accountant — Phase 2 v1 capture.
+ *
+ * One per dexcost task. Lives on Task as `_gpu` (mirrors `_compute` /
+ * `_network`). Holds the start cgroup snapshot, the per-device NVML
+ * start-snapshot timestamps (Decision #8 persistent state), and the
+ * device indexes.
+ *
+ * At task finalize:
+ *
+ * 1. Snapshots NVML utilization across all devices.
+ * 2. Walks the cgroup PIDs (Decision #1) and accumulates SM-time across
+ *    them per device.
+ * 3. Computes the window-averaged `sm_util_pct` per Decision #3 (NOT a
+ *    point sample at finalize).
+ * 4. Resolves a coarse GPU SKU hint from the productName.
+ * 5. Emits ONE gpu_cost event (cost_pending=true; back-filled at task
+ *    finalize) AND N gpu_utilization_signal events (one per device that
+ *    the task's cgroup touched).
+ *
+ * Idempotent — second call to snapshotEndAndBuild returns
+ * `{ costDetails: null, signalEvents: null }` per capture spec §5.3.
+ *
+ * **TS deviation**: accepts a `GpuAccountantHooks` object so tests can
+ * inject deterministic stubs without monkeypatching ESM module imports
+ * (matches the gpu-runtime pattern). Default hooks delegate to the
+ * production nvml-reader + cgroup-walker modules.
+ *
+ * Browser safety: snapshotStart short-circuits off-Node; snapshotEndAndBuild
+ * returns nulls.
+ *
+ * Mirrors python/src/dexcost/gpu_accountant.py.
+ */
+import { GpuRuntimeKind } from "./gpu-runtime.js";
+import type { CloudEnv } from "../cloud-detect.js";
+import { type CgroupScope } from "./cgroup-walker.js";
+import { type MemInfo, type UtilSample } from "./nvml-reader.js";
+export declare function _resetWarningStateForTests(): void;
+export interface GpuAccountantHooks {
+    initNvml: () => boolean;
+    getDeviceCount: () => number | null;
+    getProductName: (index: number) => string | null;
+    getMigMode: (index: number) => boolean;
+    getMemoryInfo: (index: number) => MemInfo | null;
+    classifyScope: () => CgroupScope;
+    enumeratePids: (scope: CgroupScope) => number[] | null;
+    getProcessUtilization: (index: number, lastSeen: Record<number, number>) => Record<number, UtilSample[]> | null;
+}
+export interface GpuCostDetails {
+    billing_model: string;
+    gpu_vendor: string;
+    gpu_sku: string | null;
+    gpu_count: number;
+    region: string | null;
+    duration_ms: number;
+    gpu_seconds_used: number;
+    instance_type: string | null;
+    vgpu_profile: string | null;
+    mig_profile: string | null;
+    cost_pending: true;
+    _nvml_product_name_lower?: string;
+    _cgroup_scope_fallback?: string;
+}
+export interface GpuUtilizationSignal {
+    gpu_index: number;
+    gpu_sku: string | null;
+    sm_util_pct: number | null;
+    mem_util_pct: number | null;
+    vram_used_peak_bytes: number;
+    vram_total_bytes: number;
+    process_count: number;
+    sample_count: number;
+    task_duration_ms: number;
+}
+export declare class GpuAccountant {
+    readonly runtime: GpuRuntimeKind;
+    readonly cloudEnv: CloudEnv;
+    private readonly hooks;
+    private _frozen;
+    private _scope;
+    private _initialPids;
+    private _initialTimestamps;
+    private _deviceIndexes;
+    private _deviceProductNames;
+    private _deviceMigModes;
+    private _vramTotal;
+    private _vramUsedPeak;
+    private _pidsTouchedPerDevice;
+    constructor(runtime: GpuRuntimeKind, cloudEnv: CloudEnv, hooks?: Partial<GpuAccountantHooks>);
+    /** Initialize NVML, snapshot cgroup PIDs, capture baseline NVML timestamps. */
+    snapshotStart(): void;
+    /**
+     * Build (gpu_cost_event, [gpu_utilization_signal_events]) at task finalize.
+     *
+     * Returns nulls when frozen, NVML unavailable, or 0 devices touched.
+     */
+    snapshotEndAndBuild(durationMs: number): {
+        costDetails: GpuCostDetails | null;
+        signalEvents: GpuUtilizationSignal[] | null;
+    };
+    private buildCostEvent;
+    private resolveVgpuProfile;
+}
+//# sourceMappingURL=gpu-accountant.d.ts.map

package/dist/core/gpu-accountant.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-accountant.d.ts","sourceRoot":"","sources":["../../src/core/gpu-accountant.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAIL,KAAK,WAAW,EACjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAOL,KAAK,OAAO,EACZ,KAAK,UAAU,EAChB,MAAM,kBAAkB,CAAC;AAM1B,wBAAgB,0BAA0B,IAAI,IAAI,CAEjD;AAWD,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,OAAO,CAAC;IACxB,cAAc,EAAE,MAAM,MAAM,GAAG,IAAI,CAAC;IACpC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,GAAG,IAAI,CAAC;IACjD,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC;IACvC,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,IAAI,CAAC;IACjD,aAAa,EAAE,MAAM,WAAW,CAAC;IACjC,aAAa,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,MAAM,EAAE,GAAG,IAAI,CAAC;IACvD,qBAAqB,EAAE,CACrB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAC7B,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC;CAC1C;AA+BD,MAAM,WAAW,cAAc;IAC7B,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE,IAAI,CAAC;IACnB,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,oBAAoB,EAAE,MAAM,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAwCD,qBAAa,aAAa;IACxB,QAAQ,CAAC,OAAO,EAAE,cAAc,CAAC;IACjC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAqB;IAE3C,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,YAAY,CAA0B;IAE9C,OAAO,CAAC,kBAAkB,CAA8C;IACxE,OAAO,CAAC,cAAc,CAAgB;IACtC,OAAO,CAAC,mBAAmB,CAAyB;IACpD,OAAO,CAAC,eAAe,CAAiB;IAExC,OAAO,CAAC,UAAU,CAA8B;IAChD,OAAO,CAAC,aAAa,CAA8B;IAEnD,OAAO,CAAC,qBAAqB,CAAmC;gBAG9D,OAAO,EAAE,cAAc,EACvB,QAAQ,EAAE,QAAQ,EAClB,KAAK,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC;IAOrC,+EAA+E;IAC/E,aAAa,IAAI,IAAI;IA4CrB;;;;OAIG;IACH,mBAAmB,CACjB,UAAU,EAAE,MAAM,GACjB;QAAE,WAAW,EAAE,cAAc,GAAG,IAAI,CAAC;QAAC,YAAY,EAAE,oBAAoB,EAAE,GAAG,IAAI,CAAA;KAAE;IAyLtF,OAAO,CAAC,cAAc;IAkCtB,OAAO,CAAC,kBAAkB;CAM3B"}

package/dist/core/gpu-accountant.js

@@ -0,0 +1,383 @@
+/**
+ * Per-task GPU accountant — Phase 2 v1 capture.
+ *
+ * One per dexcost task. Lives on Task as `_gpu` (mirrors `_compute` /
+ * `_network`). Holds the start cgroup snapshot, the per-device NVML
+ * start-snapshot timestamps (Decision #8 persistent state), and the
+ * device indexes.
+ *
+ * At task finalize:
+ *
+ * 1. Snapshots NVML utilization across all devices.
+ * 2. Walks the cgroup PIDs (Decision #1) and accumulates SM-time across
+ *    them per device.
+ * 3. Computes the window-averaged `sm_util_pct` per Decision #3 (NOT a
+ *    point sample at finalize).
+ * 4. Resolves a coarse GPU SKU hint from the productName.
+ * 5. Emits ONE gpu_cost event (cost_pending=true; back-filled at task
+ *    finalize) AND N gpu_utilization_signal events (one per device that
+ *    the task's cgroup touched).
+ *
+ * Idempotent — second call to snapshotEndAndBuild returns
+ * `{ costDetails: null, signalEvents: null }` per capture spec §5.3.
+ *
+ * **TS deviation**: accepts a `GpuAccountantHooks` object so tests can
+ * inject deterministic stubs without monkeypatching ESM module imports
+ * (matches the gpu-runtime pattern). Default hooks delegate to the
+ * production nvml-reader + cgroup-walker modules.
+ *
+ * Browser safety: snapshotStart short-circuits off-Node; snapshotEndAndBuild
+ * returns nulls.
+ *
+ * Mirrors python/src/dexcost/gpu_accountant.py.
+ */
+import { GpuRuntimeKind } from "./gpu-runtime.js";
+import { classifyScope as defaultClassifyScope, enumeratePids as defaultEnumeratePids, fallbackLabelFor, } from "./cgroup-walker.js";
+import { getDeviceCount as defaultGetDeviceCount, getMemoryInfo as defaultGetMemoryInfo, getMigMode as defaultGetMigMode, getProcessUtilization as defaultGetProcessUtilization, getProductName as defaultGetProductName, initNvml as defaultInitNvml, } from "./nvml-reader.js";
+// ─── Warn-once state ────────────────────────────────────────────────────────
+const _warnedModes = new Set();
+export function _resetWarningStateForTests() {
+    _warnedModes.clear();
+}
+function _warnOnce(mode, message) {
+    if (_warnedModes.has(mode))
+        return;
+    _warnedModes.add(mode);
+    // eslint-disable-next-line no-console
+    console.warn(message);
+}
+function defaultHooks() {
+    return {
+        initNvml: defaultInitNvml,
+        getDeviceCount: defaultGetDeviceCount,
+        getProductName: defaultGetProductName,
+        getMigMode: defaultGetMigMode,
+        getMemoryInfo: defaultGetMemoryInfo,
+        classifyScope: defaultClassifyScope,
+        enumeratePids: defaultEnumeratePids,
+        getProcessUtilization: defaultGetProcessUtilization,
+    };
+}
+// Map GpuRuntimeKind → details.billing_model for pricing dispatch.
+const BILLING_MODEL_FOR_RUNTIME = {
+    [GpuRuntimeKind.Modal]: "per_gpu_second_active",
+    [GpuRuntimeKind.RunPod]: "per_gpu_second_active",
+    [GpuRuntimeKind.Replicate]: "per_gpu_second_active",
+    [GpuRuntimeKind.LambdaLabs]: "per_gpu_hour_reserved",
+    [GpuRuntimeKind.CoreWeave]: "per_gpu_hour_reserved",
+    [GpuRuntimeKind.GcpGceN1Attached]: "per_gpu_hour_reserved",
+    [GpuRuntimeKind.AwsEc2Gpu]: "per_instance_hour",
+    [GpuRuntimeKind.GcpGceBundled]: "per_instance_hour",
+    [GpuRuntimeKind.AzureVmGpu]: "per_instance_hour",
+    [GpuRuntimeKind.AzureVmVgpu]: "per_vgpu_hour",
+};
+// ─── Helpers ────────────────────────────────────────────────────────────────
+function _isNode() {
+    return typeof process !== "undefined" && !!process.versions?.node;
+}
+function resolveSkuFromProductName(name) {
+    if (!name)
+        return null;
+    if (name.includes("h100"))
+        return "h100-80gb-sxm5";
+    if (name.includes("h200"))
+        return "h200-141gb-sxm5";
+    if (name.includes("a100")) {
+        if (name.includes("40gb"))
+            return "a100-40gb-sxm4";
+        return "a100-80gb-sxm4";
+    }
+    if (name.includes("a10g"))
+        return "a10g-24gb";
+    if (name.includes("a10-4q"))
+        return "a10-vgpu-1of6";
+    if (name.includes("a10-8q"))
+        return "a10-vgpu-1of3";
+    if (name.includes("a10-12q"))
+        return "a10-vgpu-1of2";
+    if (name.includes("a10-24q") || name.includes("a10"))
+        return "a10";
+    if (name.includes("l40s"))
+        return "l40s-48gb";
+    if (name.includes("l4"))
+        return "l4-24gb";
+    if (name.includes("tesla t4") || name.includes("nvidia t4"))
+        return "t4-16gb";
+    if (name.includes("rtx 6000"))
+        return "rtx-6000-24gb";
+    return null;
+}
+const AZURE_VGPU_PROFILE_BY_INSTANCE = {
+    Standard_NV6ads_A10_v5: "1/6 A10",
+    Standard_NV12ads_A10_v5: "1/3 A10",
+    Standard_NV18ads_A10_v5: "1/2 A10",
+    Standard_NV36ads_A10_v5: "full A10",
+    Standard_NV72ads_A10_v5: "2x A10",
+};
+// ═════════════════════════════════════════════════════════════════════════════
+// GpuAccountant
+// ═════════════════════════════════════════════════════════════════════════════
+export class GpuAccountant {
+    runtime;
+    cloudEnv;
+    hooks;
+    _frozen = false;
+    _scope = null;
+    _initialPids = new Set();
+    // Decision #8: per-device-per-PID lastSeenTimeStamp persisted across calls.
+    _initialTimestamps = {};
+    _deviceIndexes = [];
+    _deviceProductNames = [];
+    _deviceMigModes = [];
+    // Per-device peak VRAM tracker.
+    _vramTotal = {};
+    _vramUsedPeak = {};
+    // Per-device PID set observed across the task.
+    _pidsTouchedPerDevice = {};
+    constructor(runtime, cloudEnv, hooks) {
+        this.runtime = runtime;
+        this.cloudEnv = cloudEnv;
+        this.hooks = { ...defaultHooks(), ...(hooks ?? {}) };
+    }
+    /** Initialize NVML, snapshot cgroup PIDs, capture baseline NVML timestamps. */
+    snapshotStart() {
+        if (!_isNode())
+            return;
+        if (!this.hooks.initNvml())
+            return;
+        const count = this.hooks.getDeviceCount() ?? 0;
+        if (count <= 0)
+            return;
+        this._deviceIndexes = Array.from({ length: count }, (_, i) => i);
+        for (const i of this._deviceIndexes) {
+            const name = this.hooks.getProductName(i);
+            this._deviceProductNames.push(name);
+            const mig = this.hooks.getMigMode(i);
+            this._deviceMigModes.push(mig);
+            if (mig) {
+                _warnOnce(`gpu_mig_detected_full_billing_applied:device${i}`, `NVML reports MIG enabled on device ${i} (productName=${String(name)}); ` +
+                    `Decision #2 — full-GPU rate applied. details.mig_profile populated ` +
+                    `for v1.1 forward-compat.`);
+            }
+            const mem = this.hooks.getMemoryInfo(i);
+            if (mem) {
+                this._vramTotal[i] = mem.totalBytes;
+                this._vramUsedPeak[i] = mem.usedBytes;
+            }
+            this._initialTimestamps[i] = {};
+            this._pidsTouchedPerDevice[i] = new Set();
+            // Baseline NVML sample — captures the per-PID lastSeenTimeStamp.
+            const baseline = this.hooks.getProcessUtilization(i, this._initialTimestamps[i]);
+            if (baseline) {
+                for (const pidKey of Object.keys(baseline)) {
+                    this._pidsTouchedPerDevice[i].add(Number(pidKey));
+                }
+            }
+        }
+        // Snapshot cgroup PIDs (Decision #1 scope classification).
+        this._scope = this.hooks.classifyScope();
+        const pids = this.hooks.enumeratePids(this._scope);
+        this._initialPids = new Set(pids ?? [process.pid]);
+    }
+    /**
+     * Build (gpu_cost_event, [gpu_utilization_signal_events]) at task finalize.
+     *
+     * Returns nulls when frozen, NVML unavailable, or 0 devices touched.
+     */
+    snapshotEndAndBuild(durationMs) {
+        if (this._frozen)
+            return { costDetails: null, signalEvents: null };
+        this._frozen = true;
+        if (this._deviceIndexes.length === 0) {
+            return { costDetails: null, signalEvents: null };
+        }
+        // End-snapshot cgroup walk + Decision #1 fallback label.
+        const scope = this._scope ?? this.hooks.classifyScope();
+        const endPidsList = this.hooks.enumeratePids(scope);
+        let fallbackLabel;
+        let currentPids;
+        if (endPidsList === null) {
+            fallbackLabel = "self_pid_only";
+            currentPids = new Set([process.pid]);
+        }
+        else {
+            fallbackLabel = fallbackLabelFor(scope);
+            currentPids = new Set(endPidsList);
+        }
+        // Union of start + end PIDs.
+        const cgroupPidUnion = new Set([
+            ...this._initialPids,
+            ...currentPids,
+        ]);
+        // Canonical SKU from first non-null productName (homogeneous-device assumption).
+        const canonicalProductName = this._deviceProductNames.find((n) => n !== null && n !== undefined) ?? null;
+        const gpuSku = resolveSkuFromProductName(canonicalProductName);
+        // Decision #2 transparency: MIG presence surfaced regardless of cgroup-touch.
+        let migProfile = null;
+        if (this._deviceMigModes.some((m) => m)) {
+            migProfile = "mig_detected";
+        }
+        const degenerateWindow = durationMs <= 0;
+        const perDeviceGpuSeconds = {};
+        const signalEvents = [];
+        let anyPidTouched = false;
+        for (const i of this._deviceIndexes) {
+            // Sprint 2 Theme C / §3.1.1 (B2 TS port) — snapshot per-PID
+            // baseline timestamps BEFORE the end NVML call mutates
+            // `_initialTimestamps[i]` in place. Reading the mutated map
+            // afterwards would zero every PID's first-sample dt.
+            const baselineTsPerPid = {
+                ...this._initialTimestamps[i],
+            };
+            const endSamples = this.hooks.getProcessUtilization(i, this._initialTimestamps[i]) ?? {};
+            for (const pidKey of Object.keys(endSamples)) {
+                this._pidsTouchedPerDevice[i].add(Number(pidKey));
+            }
+            const mem = this.hooks.getMemoryInfo(i);
+            if (mem) {
+                this._vramUsedPeak[i] = Math.max(this._vramUsedPeak[i] ?? 0, mem.usedBytes);
+            }
+            // Filter to cgroup-PID union. Each value is a list of samples.
+            const relevantByPid = {};
+            for (const [pidKey, samples] of Object.entries(endSamples)) {
+                const pid = Number(pidKey);
+                if (cgroupPidUnion.has(pid) && samples.length > 0) {
+                    relevantByPid[pid] = samples;
+                }
+            }
+            const relevantPidCount = Object.keys(relevantByPid).length;
+            if (relevantPidCount > 0) {
+                anyPidTouched = true;
+                // B2: integrate sm_util × dt across the sample sequence.
+                // Two semantics for "first sample of a PID with no baseline":
+                //  - Device had ZERO PIDs at start → first sample's window
+                //    extends back to the derived task_start_ts.
+                //  - Other PIDs were active but this one wasn't → PID
+                //    joined mid-task; first-sample dt is 0.
+                const deviceHadBaselinePids = Object.keys(baselineTsPerPid).length > 0;
+                let maxSampleTs = 0;
+                for (const samples of Object.values(relevantByPid)) {
+                    for (const s of samples) {
+                        if (s.timeStamp > maxSampleTs)
+                            maxSampleTs = s.timeStamp;
+                    }
+                }
+                const taskStartTs = Math.max(0, maxSampleTs - durationMs * 1000);
+                let gpuSecondsForDevice = 0;
+                let memUtilSum = 0;
+                let memUtilN = 0;
+                for (const [pidKey, samples] of Object.entries(relevantByPid)) {
+                    const pid = Number(pidKey);
+                    let baselineForPid;
+                    if (pid in baselineTsPerPid) {
+                        baselineForPid = baselineTsPerPid[pid];
+                    }
+                    else if (deviceHadBaselinePids) {
+                        baselineForPid = samples[0].timeStamp;
+                    }
+                    else {
+                        baselineForPid = taskStartTs;
+                    }
+                    let prevTs = baselineForPid;
+                    for (const s of samples) {
+                        const dtUs = Math.max(0, s.timeStamp - prevTs);
+                        gpuSecondsForDevice +=
+                            (s.smUtil / 100.0) * (dtUs / 1_000_000.0);
+                        prevTs = s.timeStamp;
+                        memUtilSum += s.memUtil;
+                        memUtilN += 1;
+                    }
+                }
+                perDeviceGpuSeconds[i] = gpuSecondsForDevice;
+                let smUtilPct;
+                if (durationMs > 0) {
+                    const windowS = durationMs / 1000.0;
+                    smUtilPct = Math.min(100.0, (gpuSecondsForDevice / windowS) * 100.0);
+                }
+                else {
+                    smUtilPct = null;
+                }
+                const memUtilAvg = memUtilN > 0 ? memUtilSum / memUtilN : 0;
+                signalEvents.push({
+                    gpu_index: i,
+                    gpu_sku: gpuSku,
+                    sm_util_pct: smUtilPct,
+                    mem_util_pct: memUtilAvg,
+                    vram_used_peak_bytes: this._vramUsedPeak[i] ?? 0,
+                    vram_total_bytes: this._vramTotal[i] ?? 0,
+                    process_count: this._pidsTouchedPerDevice[i].size,
+                    sample_count: memUtilN,
+                    task_duration_ms: durationMs,
+                });
+            }
+            else if (degenerateWindow) {
+                signalEvents.push({
+                    gpu_index: i,
+                    gpu_sku: gpuSku,
+                    sm_util_pct: null,
+                    mem_util_pct: null,
+                    vram_used_peak_bytes: this._vramUsedPeak[i] ?? 0,
+                    vram_total_bytes: this._vramTotal[i] ?? 0,
+                    process_count: this._pidsTouchedPerDevice[i].size,
+                    sample_count: 0,
+                    task_duration_ms: durationMs,
+                });
+            }
+        }
+        const shouldEmitCost = anyPidTouched ||
+            fallbackLabel !== null ||
+            degenerateWindow ||
+            this._deviceMigModes.some((m) => m);
+        if (!shouldEmitCost) {
+            return { costDetails: null, signalEvents: null };
+        }
+        const totalGpuSeconds = Object.values(perDeviceGpuSeconds).reduce((a, b) => a + b, 0);
+        const costDetails = this.buildCostEvent({
+            durationMs,
+            gpuSku,
+            gpuCount: this._deviceIndexes.length,
+            gpuSecondsUsed: totalGpuSeconds,
+            migProfile,
+            fallbackLabel,
+        });
+        return {
+            costDetails,
+            signalEvents: signalEvents.length > 0 ? signalEvents : null,
+        };
+    }
+    // ------------------------------------------------------------------
+    // Event builders
+    // ------------------------------------------------------------------
+    buildCostEvent(args) {
+        const details = {
+            billing_model: BILLING_MODEL_FOR_RUNTIME[this.runtime] ?? "per_gpu_second_active",
+            gpu_vendor: "nvidia", // Decision #5
+            gpu_sku: args.gpuSku,
+            gpu_count: args.gpuCount,
+            region: this.cloudEnv.region,
+            duration_ms: args.durationMs,
+            gpu_seconds_used: args.gpuSecondsUsed,
+            instance_type: this.cloudEnv.instanceType ?? null,
+            vgpu_profile: this.resolveVgpuProfile(),
+            mig_profile: args.migProfile,
+            cost_pending: true,
+        };
+        const productName = this._deviceProductNames.find((n) => n !== null && n !== undefined);
+        if (productName) {
+            details._nvml_product_name_lower = productName;
+        }
+        if (args.fallbackLabel) {
+            details._cgroup_scope_fallback = args.fallbackLabel;
+        }
+        return details;
+    }
+    resolveVgpuProfile() {
+        if (this.runtime !== GpuRuntimeKind.AzureVmVgpu)
+            return null;
+        const it = this.cloudEnv.instanceType ?? null;
+        if (!it)
+            return null;
+        return AZURE_VGPU_PROFILE_BY_INSTANCE[it] ?? null;
+    }
+}
+//# sourceMappingURL=gpu-accountant.js.map

package/dist/core/gpu-accountant.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-accountant.js","sourceRoot":"","sources":["../../src/core/gpu-accountant.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,EACL,aAAa,IAAI,oBAAoB,EACrC,aAAa,IAAI,oBAAoB,EACrC,gBAAgB,GAEjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,cAAc,IAAI,qBAAqB,EACvC,aAAa,IAAI,oBAAoB,EACrC,UAAU,IAAI,iBAAiB,EAC/B,qBAAqB,IAAI,4BAA4B,EACrD,cAAc,IAAI,qBAAqB,EACvC,QAAQ,IAAI,eAAe,GAG5B,MAAM,kBAAkB,CAAC;AAE1B,+EAA+E;AAE/E,MAAM,YAAY,GAAgB,IAAI,GAAG,EAAE,CAAC;AAE5C,MAAM,UAAU,0BAA0B;IACxC,YAAY,CAAC,KAAK,EAAE,CAAC;AACvB,CAAC;AAED,SAAS,SAAS,CAAC,IAAY,EAAE,OAAe;IAC9C,IAAI,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;QAAE,OAAO;IACnC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACvB,sCAAsC;IACtC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;AACxB,CAAC;AAkBD,SAAS,YAAY;IACnB,OAAO;QACL,QAAQ,EAAE,eAAe;QACzB,cAAc,EAAE,qBAAqB;QACrC,cAAc,EAAE,qBAAqB;QACrC,UAAU,EAAE,iBAAiB;QAC7B,aAAa,EAAE,oBAAoB;QACnC,aAAa,EAAE,oBAAoB;QACnC,aAAa,EAAE,oBAAoB;QACnC,qBAAqB,EAAE,4BAA4B;KACpD,CAAC;AACJ,CAAC;AAED,mEAAmE;AACnE,MAAM,yBAAyB,GAA2B;IACxD,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,uBAAuB;IAC/C,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,uBAAuB;IAChD,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,uBAAuB;IACnD,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,uBAAuB;IACpD,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,uBAAuB;IACnD,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,uBAAuB;IAC1D,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,mBAAmB;IAC/C,CAAC,cAAc,CAAC,aAAa,CAAC,EAAE,mBAAmB;IACnD,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,mBAAmB;IAChD,CAAC,cAAc,CAAC,WAAW,CAAC,EAAE,eAAe;CAC9C,CAAC;AAgCF,+EAA+E;AAE/E,SAAS,OAAO;IACd,OAAO,OAAO,OAAO,KAAK,WAAW,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC;AACpE,CAAC;AAED,SAAS,yBAAyB,CAAC,IAA+B;IAChE,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACvB,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,gBAAgB,CAAC;IACnD,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,iBAAiB,CAAC;IACpD,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,gBAAgB,CAAC;QACnD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,WAAW,CAAC;IAC9C,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,eAAe,CAAC;IACpD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,eAAe,CAAC;IACpD,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,eAAe,CAAC;IACrD,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACnE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO,WAAW,CAAC;IAC9C,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAC1C,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC;QAAE,OAAO,SAAS,CAAC;IAC9E,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;QAAE,OAAO,eAAe,CAAC;IACtD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,8BAA8B,GAA2B;IAC7D,sBAAsB,EAAE,SAAS;IACjC,uBAAuB,EAAE,SAAS;IAClC,uBAAuB,EAAE,SAAS;IAClC,uBAAuB,EAAE,UAAU;IACnC,uBAAuB,EAAE,QAAQ;CAClC,CAAC;AAEF,gFAAgF;AAChF,gBAAgB;AAChB,gFAAgF;AAEhF,MAAM,OAAO,aAAa;IACf,OAAO,CAAiB;IACxB,QAAQ,CAAW;IACX,KAAK,CAAqB;IAEnC,OAAO,GAAG,KAAK,CAAC;IAChB,MAAM,GAAuB,IAAI,CAAC;IAClC,YAAY,GAAgB,IAAI,GAAG,EAAE,CAAC;IAC9C,4EAA4E;IACpE,kBAAkB,GAA2C,EAAE,CAAC;IAChE,cAAc,GAAa,EAAE,CAAC;IAC9B,mBAAmB,GAAsB,EAAE,CAAC;IAC5C,eAAe,GAAc,EAAE,CAAC;IACxC,gCAAgC;IACxB,UAAU,GAA2B,EAAE,CAAC;IACxC,aAAa,GAA2B,EAAE,CAAC;IACnD,+CAA+C;IACvC,qBAAqB,GAAgC,EAAE,CAAC;IAEhE,YACE,OAAuB,EACvB,QAAkB,EAClB,KAAmC;QAEnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,YAAY,EAAE,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;IACvD,CAAC;IAED,+EAA+E;IAC/E,aAAa;QACX,IAAI,CAAC,OAAO,EAAE;YAAE,OAAO;QACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE;YAAE,OAAO;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAC/C,IAAI,KAAK,IAAI,CAAC;YAAE,OAAO;QACvB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QACjE,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;YAC1C,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACrC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC/B,IAAI,GAAG,EAAE,CAAC;gBACR,SAAS,CACP,+CAA+C,CAAC,EAAE,EAClD,sCAAsC,CAAC,iBAAiB,MAAM,CAAC,IAAI,CAAC,KAAK;oBACvE,qEAAqE;oBACrE,0BAA0B,CAC7B,CAAC;YACJ,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACxC,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,UAAU,CAAC;gBACpC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,SAAS,CAAC;YACxC,CAAC;YACD,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;YAChC,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,EAAU,CAAC;YAClD,iEAAiE;YACjE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAC/C,CAAC,EACD,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAC3B,CAAC;YACF,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC3C,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;QACH,CAAC;QAED,2DAA2D;QAC3D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;QACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnD,IAAI,CAAC,YAAY,GAAG,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IACrD,CAAC;IAED;;;;OAIG;IACH,mBAAmB,CACjB,UAAkB;QAElB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QACnE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrC,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QACnD,CAAC;QAED,yDAAyD;QACzD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;QACxD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,aAA4B,CAAC;QACjC,IAAI,WAAwB,CAAC;QAC7B,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;YACzB,aAAa,GAAG,eAAe,CAAC;YAChC,WAAW,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACxC,WAAW,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC;QAED,6BAA6B;QAC7B,MAAM,cAAc,GAAG,IAAI,GAAG,CAAS;YACrC,GAAG,IAAI,CAAC,YAAY;YACpB,GAAG,WAAW;SACf,CAAC,CAAC;QAEH,iFAAiF;QACjF,MAAM,oBAAoB,GACxB,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS,CAAC,IAAI,IAAI,CAAC;QAC9E,MAAM,MAAM,GAAG,yBAAyB,CAAC,oBAAoB,CAAC,CAAC;QAE/D,8EAA8E;QAC9E,IAAI,UAAU,GAAkB,IAAI,CAAC;QACrC,IAAI,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxC,UAAU,GAAG,cAAc,CAAC;QAC9B,CAAC;QAED,MAAM,gBAAgB,GAAG,UAAU,IAAI,CAAC,CAAC;QAEzC,MAAM,mBAAmB,GAA2B,EAAE,CAAC;QACvD,MAAM,YAAY,GAA2B,EAAE,CAAC;QAChD,IAAI,aAAa,GAAG,KAAK,CAAC;QAE1B,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACpC,4DAA4D;YAC5D,uDAAuD;YACvD,4DAA4D;YAC5D,qDAAqD;YACrD,MAAM,gBAAgB,GAA2B;gBAC/C,GAAG,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC;aAC9B,CAAC;YAEF,MAAM,UAAU,GACd,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACxE,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC7C,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YACpD,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACxC,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAC9B,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,EAC1B,GAAG,CAAC,SAAS,CACd,CAAC;YACJ,CAAC;YAED,+DAA+D;YAC/D,MAAM,aAAa,GAAiC,EAAE,CAAC;YACvD,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC3D,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC3B,IAAI,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAClD,aAAa,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;gBAC/B,CAAC;YACH,CAAC;YACD,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC;YAE3D,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;gBACzB,aAAa,GAAG,IAAI,CAAC;gBAErB,yDAAyD;gBACzD,8DAA8D;gBAC9D,2DAA2D;gBAC3D,gDAAgD;gBAChD,sDAAsD;gBACtD,4CAA4C;gBAC5C,MAAM,qBAAqB,GAAG,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;gBACvE,IAAI,WAAW,GAAG,CAAC,CAAC;gBACpB,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;oBACnD,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;wBACxB,IAAI,CAAC,CAAC,SAAS,GAAG,WAAW;4BAAE,WAAW,GAAG,CAAC,CAAC,SAAS,CAAC;oBAC3D,CAAC;gBACH,CAAC;gBACD,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,UAAU,GAAG,IAAI,CAAC,CAAC;gBAEjE,IAAI,mBAAmB,GAAG,CAAC,CAAC;gBAC5B,IAAI,UAAU,GAAG,CAAC,CAAC;gBACnB,IAAI,QAAQ,GAAG,CAAC,CAAC;gBACjB,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;oBAC9D,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;oBAC3B,IAAI,cAAsB,CAAC;oBAC3B,IAAI,GAAG,IAAI,gBAAgB,EAAE,CAAC;wBAC5B,cAAc,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;oBACzC,CAAC;yBAAM,IAAI,qBAAqB,EAAE,CAAC;wBACjC,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;oBACxC,CAAC;yBAAM,CAAC;wBACN,cAAc,GAAG,WAAW,CAAC;oBAC/B,CAAC;oBACD,IAAI,MAAM,GAAG,cAAc,CAAC;oBAC5B,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;wBACxB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,GAAG,MAAM,CAAC,CAAC;wBAC/C,mBAAmB;4BACjB,CAAC,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,WAAW,CAAC,CAAC;wBAC5C,MAAM,GAAG,CAAC,CAAC,SAAS,CAAC;wBACrB,UAAU,IAAI,CAAC,CAAC,OAAO,CAAC;wBACxB,QAAQ,IAAI,CAAC,CAAC;oBAChB,CAAC;gBACH,CAAC;gBACD,mBAAmB,CAAC,CAAC,CAAC,GAAG,mBAAmB,CAAC;gBAE7C,IAAI,SAAwB,CAAC;gBAC7B,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;oBACnB,MAAM,OAAO,GAAG,UAAU,GAAG,MAAM,CAAC;oBACpC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,mBAAmB,GAAG,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;gBACvE,CAAC;qBAAM,CAAC;oBACN,SAAS,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,MAAM,UAAU,GAAG,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE5D,YAAY,CAAC,IAAI,CAAC;oBAChB,SAAS,EAAE,CAAC;oBACZ,OAAO,EAAE,MAAM;oBACf,WAAW,EAAE,SAAS;oBACtB,YAAY,EAAE,UAAU;oBACxB,oBAAoB,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;oBAChD,gBAAgB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC;oBACzC,aAAa,EAAE,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,IAAI;oBACjD,YAAY,EAAE,QAAQ;oBACtB,gBAAgB,EAAE,UAAU;iBAC7B,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,gBAAgB,EAAE,CAAC;gBAC5B,YAAY,CAAC,IAAI,CAAC;oBAChB,SAAS,EAAE,CAAC;oBACZ,OAAO,EAAE,MAAM;oBACf,WAAW,EAAE,IAAI;oBACjB,YAAY,EAAE,IAAI;oBAClB,oBAAoB,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;oBAChD,gBAAgB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC;oBACzC,aAAa,EAAE,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,IAAI;oBACjD,YAAY,EAAE,CAAC;oBACf,gBAAgB,EAAE,UAAU;iBAC7B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,cAAc,GAClB,aAAa;YACb,aAAa,KAAK,IAAI;YACtB,gBAAgB;YAChB,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QACnD,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,MAAM,CAC/D,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EACf,CAAC,CACF,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC;YACtC,UAAU;YACV,MAAM;YACN,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM;YACpC,cAAc,EAAE,eAAe;YAC/B,UAAU;YACV,aAAa;SACd,CAAC,CAAC;QACH,OAAO;YACL,WAAW;YACX,YAAY,EAAE,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI;SAC5D,CAAC;IACJ,CAAC;IAED,qEAAqE;IACrE,iBAAiB;IACjB,qEAAqE;IAE7D,cAAc,CAAC,IAOtB;QACC,MAAM,OAAO,GAAmB;YAC9B,aAAa,EACX,yBAAyB,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,uBAAuB;YACpE,UAAU,EAAE,QAAQ,EAAE,cAAc;YACpC,OAAO,EAAE,IAAI,CAAC,MAAM;YACpB,SAAS,EAAE,IAAI,CAAC,QAAQ;YACxB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM;YAC5B,WAAW,EAAE,IAAI,CAAC,UAAU;YAC5B,gBAAgB,EAAE,IAAI,CAAC,cAAc;YACrC,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,IAAI;YACjD,YAAY,EAAE,IAAI,CAAC,kBAAkB,EAAE;YACvC,WAAW,EAAE,IAAI,CAAC,UAAU;YAC5B,YAAY,EAAE,IAAI;SACnB,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS,CACrC,CAAC;QACF,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,wBAAwB,GAAG,WAAW,CAAC;QACjD,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,OAAO,CAAC,sBAAsB,GAAG,IAAI,CAAC,aAAa,CAAC;QACtD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,kBAAkB;QACxB,IAAI,IAAI,CAAC,OAAO,KAAK,cAAc,CAAC,WAAW;YAAE,OAAO,IAAI,CAAC;QAC7D,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,IAAI,CAAC;QAC9C,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC;QACrB,OAAO,8BAA8B,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACpD,CAAC;CACF"}

package/dist/core/gpu-runtime.d.ts

@@ -0,0 +1,58 @@
+/**
+ * Active GPU runtime resolver — Phase 2 Task 3.
+ *
+ * Sibling of `compute-runtime.ts`. Coexists without modification —
+ * compute-runtime answers "which compute billing model" and gpu-runtime
+ * answers "which GPU billing model (if any)".
+ *
+ * Cascade priority (capture spec §5.5):
+ *
+ * 1. Serverless GPU env vars (MODAL_TASK_ID / RUNPOD_POD_ID /
+ *    REPLICATE_MODEL) — win immediately when NVML is available.
+ * 2. IaaS GPU via cloud_detect — provider AWS/GCP/Azure AND instanceType
+ *    matches a GPU-family regex AND NVML reports ≥1 device. Classifies as
+ *    AWS_EC2_GPU / GCP_GCE_BUNDLED / GCP_GCE_N1_ATTACHED (Decision #9) /
+ *    AZURE_VM_GPU / AZURE_VM_VGPU (Decision #10).
+ * 3. Reserved-GPU providers — Lambda Labs / CoreWeave from cloud_detect
+ *    AND NVML reports ≥1 device.
+ * 4. NONE when NVML unavailable, 0 devices, or unmatched runtime.
+ *
+ * **String values match Python EXACTLY** — cross-SDK event portability.
+ *
+ * **TS deviation**: takes an `Options` object with hook functions. This
+ * lets tests inject deterministic stubs without monkeypatching module-
+ * level imports (ESM bindings are immutable). The default options use the
+ * production nvml-reader + cloud-detect modules.
+ *
+ * Mirrors python/src/dexcost/gpu_runtime.py.
+ */
+import { type CloudEnv } from "../cloud-detect.js";
+/** GPU runtime kind discriminator. String values pinned to Python. */
+export declare const GpuRuntimeKind: {
+    readonly Modal: "modal";
+    readonly RunPod: "runpod";
+    readonly Replicate: "replicate";
+    readonly LambdaLabs: "lambda_labs";
+    readonly CoreWeave: "coreweave";
+    readonly AwsEc2Gpu: "aws_ec2_gpu";
+    readonly GcpGceBundled: "gcp_gce_bundled";
+    readonly GcpGceN1Attached: "gcp_gce_n1_attached";
+    readonly AzureVmGpu: "azure_vm_gpu";
+    readonly AzureVmVgpu: "azure_vm_vgpu";
+    readonly None: "none";
+};
+export type GpuRuntimeKind = (typeof GpuRuntimeKind)[keyof typeof GpuRuntimeKind];
+export interface ResolveGpuRuntimeOptions {
+    nvmlAvailable?: () => boolean;
+    getDeviceCount?: () => number | null;
+    getCloudEnv?: () => CloudEnv;
+}
+/**
+ * Return the active GPU runtime, or NONE when there's no GPU.
+ *
+ * The cascade short-circuits on the FIRST positive match. If NVML can't
+ * initialize or reports 0 devices, returns NONE regardless of env-var
+ * signals — a Modal task on a CPU-only Modal function emits no GPU events.
+ */
+export declare function resolveGpuRuntime(options?: ResolveGpuRuntimeOptions): GpuRuntimeKind;
+//# sourceMappingURL=gpu-runtime.d.ts.map

package/dist/core/gpu-runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-runtime.d.ts","sourceRoot":"","sources":["../../src/core/gpu-runtime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAEH,OAAO,EAAqC,KAAK,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAMtF,sEAAsE;AACtE,eAAO,MAAM,cAAc;;;;;;;;;;;;CAYjB,CAAC;AAEX,MAAM,MAAM,cAAc,GACxB,CAAC,OAAO,cAAc,CAAC,CAAC,MAAM,OAAO,cAAc,CAAC,CAAC;AAEvD,MAAM,WAAW,wBAAwB;IACvC,aAAa,CAAC,EAAE,MAAM,OAAO,CAAC;IAC9B,cAAc,CAAC,EAAE,MAAM,MAAM,GAAG,IAAI,CAAC;IACrC,WAAW,CAAC,EAAE,MAAM,QAAQ,CAAC;CAC9B;AA0CD;;;;;;GAMG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,CAAC,EAAE,wBAAwB,GACjC,cAAc,CA0ChB"}