fix: wait for context lost promise

antvis · Jun 13, 2023 · f64e45d · f64e45d
1 parent bc371cf
commit f64e45d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/packages/g-plugin-webgpu-device/src/WebGPUDeviceContribution.ts b/packages/g-plugin-webgpu-device/src/WebGPUDeviceContribution.ts
@@ -32,13 +32,15 @@ export class WebGPUDeviceContribution implements DeviceContribution {
     );
     const device = await adapter.requestDevice({ requiredFeatures });
 
-    // @see https://github.com/gpuweb/gpuweb/blob/main/design/ErrorHandling.md#fatal-errors-requestadapter-requestdevice-and-devicelost
-    const { onContextLost } = this.pluginOptions;
-    device.lost.then(() => {
-      if (onContextLost) {
-        onContextLost();
-      }
-    });
+    if (device) {
+      // @see https://github.com/gpuweb/gpuweb/blob/main/design/ErrorHandling.md#fatal-errors-requestadapter-requestdevice-and-devicelost
+      const { onContextLost } = this.pluginOptions;
+      device.lost.then(() => {
+        if (onContextLost) {
+          onContextLost();
+        }
+      });
+    }
 
     if (device === null) return null;