patrto: lenet - change createContext to use GPUDevice

patrto · Aug 28, 2024 · 74d1f10 · 74d1f10
1 parent 100ae69
commit 74d1f10
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lenet/lenet.js b/lenet/lenet.js
@@ -22,7 +22,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];
     let input = this.builder_.input('input', {