Merge pull request #296 from Honry/remove-MLTensorUsage

Fix MLTensorUsage is undefined issue
webmachinelearning · Nov 14, 2024 · 5be72c7 · 5be72c7
2 parents 77a98b9 + f45c753
commit 5be72c7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/common/utils.js b/common/utils.js
@@ -306,8 +306,11 @@ export function getUrlParams() {
 
 export async function isWebNN() {
   if (typeof MLGraphBuilder !== 'undefined') {
-    const context = await navigator.ml.createContext();
-    return !context.tf;
+    // Polyfill MLTensorUsage to make it compatible with old version of Chrome.
+    if (typeof MLTensorUsage == 'undefined') {
+      window.MLTensorUsage = {WEBGPU_INTEROP: 1, READ: 2, WRITE: 4};
+    }
+    return true;
   } else {
     return false;
   }

diff --git a/nnotepad/js/nnotepad.js b/nnotepad/js/nnotepad.js
@@ -52,7 +52,8 @@ class WebNNUtil {
       dataType: isShapeMethod ? operand.dataType() : operand.dataType,
       dimensions: isShapeMethod ? operand.shape() : operand.shape,
       shape: isShapeMethod ? operand.shape() : operand.shape,
-      usage: MLTensorUsage.READ,
+      usage: typeof MLTensorUsage == 'undefined' ?
+          undefined : MLTensorUsage.READ,
       readable: true,
     };
     const tensor = await context.createTensor(desc);