@kaminos/webgpu-inference-kit 0.1.2 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kaminos/webgpu-inference-kit",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "private": false,
   "type": "module",
   "description": "Composable browser WebGPU inference route contracts, runtime profiles, and scheduler envelopes.",

package/src/kernel-profile.js

@@ -1,4 +1,4 @@
-export const WEBGPU_INFERENCE_KIT_VERSION = '0.1.2';
+export const WEBGPU_INFERENCE_KIT_VERSION = '0.1.3';
 const DEFAULT_KIT_VERSION = WEBGPU_INFERENCE_KIT_VERSION;
 const DEFAULT_TIMING_SOURCE = 'queue-submit-wait';
 

package/src/route-boundary.js

@@ -95,6 +95,49 @@ function optionalRoleNames(roles) {
   return roles.filter(role => role.required === false).map(role => role.role);
 }
 
+function routeTimingStageNames(timings = {}) {
+  const names = new Set();
+  const addStageName = stage => {
+    if (isNonEmptyString(stage)) names.add(stage);
+    if (isNonEmptyString(stage?.name)) names.add(stage.name);
+  };
+
+  if (Array.isArray(timings.stages)) {
+    timings.stages.forEach(addStageName);
+  }
+
+  const profile = timings.profile;
+  if (profile && typeof profile === 'object') {
+    if (Array.isArray(profile.stageNames)) {
+      profile.stageNames.forEach(addStageName);
+    }
+    if (Array.isArray(profile.stages)) {
+      profile.stages.forEach(addStageName);
+    }
+  }
+
+  return names;
+}
+
+function validateRouteTiming(errors, receipt, route) {
+  const timings = receipt?.timings;
+  if (!timings || !route) return;
+
+  if (isNonEmptyString(route.timingSource) && timings.source !== route.timingSource) {
+    errors.push(`receipt.timings.source must be ${route.timingSource}`);
+  }
+
+  const requiredStages = Array.isArray(route.requiredStages) ? route.requiredStages : [];
+  if (requiredStages.length === 0) return;
+
+  const stageNames = routeTimingStageNames(timings);
+  for (const stageName of requiredStages) {
+    if (!stageNames.has(stageName)) {
+      errors.push(`receipt.timings missing required stage ${stageName}`);
+    }
+  }
+}
+
 function validateArtifacts(errors, artifacts, roles, path, { requireHash }) {
   const knownRoles = roleSet(roles);
   const artifactsByRole = new Map();
@@ -322,6 +365,7 @@ export function validateRouteWorkerResult(result, route) {
     if (result.receipt.effectiveRouteId !== route.routeId) {
       errors.push('receipt.effectiveRouteId must match route definition');
     }
+    if (routeResult.ok) validateRouteTiming(errors, result.receipt, route);
   }
 
   const backendResult = validateWebGpuBackendIdentity(result.backend);