patrto: lenet - change createContext to use GPUDevice

patrto · Nov 12, 2024 · 01ee569 · 01ee569
1 parent eddcc1c
commit 01ee569
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lenet/lenet.js b/lenet/lenet.js
@@ -25,7 +25,10 @@ export class LeNet {
       throw new Error('Incorrect weights file');
     }
 
-    this.context_ = await navigator.ml.createContext(contextOptions);
+    let adapter = await navigator.gpu.requestAdapter();
+    let device = await adapter.requestDevice();
+
+    this.context_ = await navigator.ml.createContext(device);
     this.builder_ = new MLGraphBuilder(this.context_);
     const inputShape = /* nchw */ [1, 1, 28, 28];
     const inputDesc = {