WebGPURenderer: Handle OutOfMemory in Timestamp Tracking

src/renderers/webgpu/WebGPUBackend.js

@@ -332,7 +332,7 @@ class WebGPUBackend extends Backend {
 
 			//
 
-			occlusionQuerySet = device.createQuerySet( { type: 'occlusion', count: occlusionQueryCount } );
+			occlusionQuerySet = device.createQuerySet( { type: 'occlusion', count: occlusionQueryCount, label: `occlusionQuerySet_${ renderContext.id }` } );
 
 			renderContextData.occlusionQuerySet = occlusionQuerySet;
 			renderContextData.occlusionQueryIndex = 0;
@@ -1166,27 +1166,44 @@ class WebGPUBackend extends Backend {
 	}
 
 
-	initTimestampQuery( renderContext, descriptor ) {
+	async initTimestampQuery( renderContext, descriptor ) {
 
 		if ( ! this.trackTimestamp ) return;
 
 		const renderContextData = this.get( renderContext );
 
 		if ( ! renderContextData.timeStampQuerySet ) {
 
-			// Create a GPUQuerySet which holds 2 timestamp query results: one for the
-			// beginning and one for the end of compute pass execution.
-			const timeStampQuerySet = this.device.createQuerySet( { type: 'timestamp', count: 2 } );
+
+			// Push an error scope to catch any errors during query set creation
+			this.device.pushErrorScope( 'out-of-memory' );
+
+			const timeStampQuerySet = await this.device.createQuerySet( { type: 'timestamp', count: 2, label: `timestamp_renderContext_${renderContext.id}` } );
+
+			// Pop the error scope and check for errors
+			const error = await this.device.popErrorScope();
+
+			if ( error ) {
+
+				if ( ! renderContextData.attemptingTimeStampQuerySetFailed ) {
+
+					console.error( `[GPUOutOfMemoryError][renderContext_${renderContext.id}]:\nFailed to create timestamp query set. This may be because timestamp queries are already running in other tabs.` );
+					renderContextData.attemptingTimeStampQuerySetFailed = true;
+
+				}
+
+				renderContextData.timeStampQuerySet = null; // Mark as unavailable
+				return;
+
+			}
 
 			const timestampWrites = {
 				querySet: timeStampQuerySet,
 				beginningOfPassWriteIndex: 0, // Write timestamp in index 0 when pass begins.
 				endOfPassWriteIndex: 1, // Write timestamp in index 1 when pass ends.
 			};
 
-			Object.assign( descriptor, {
-				timestampWrites,
-			} );
+			Object.assign( descriptor, { timestampWrites } );
 
 			renderContextData.timeStampQuerySet = timeStampQuerySet;
 
@@ -1202,6 +1219,7 @@ class WebGPUBackend extends Backend {
 
 		const renderContextData = this.get( renderContext );
 
+		if ( ! renderContextData.timeStampQuerySet ) return;
 
 		const size = 2 * BigInt64Array.BYTES_PER_ELEMENT;
 
@@ -1238,6 +1256,8 @@ class WebGPUBackend extends Backend {
 
 		const renderContextData = this.get( renderContext );
 
+		if ( ! renderContextData.timeStampQuerySet ) return;
+
 		if ( renderContextData.currentTimestampQueryBuffers === undefined ) return;
 
 		const { resultBuffer, isMappingPending } = renderContextData.currentTimestampQueryBuffers;