WGSL textureNumLevels execution tests (#3855)

* WGSL textureNumLayers execution test

src/webgpu/shader/execution/expression/call/builtin/textureNumLevels.spec.ts

@@ -5,9 +5,38 @@ Returns the number of mip levels of a texture.
 `;
 
 import { makeTestGroup } from '../../../../../../common/framework/test_group.js';
-import { GPUTest } from '../../../../../gpu_test.js';
+import { getTextureDimensionFromView } from '../../../../../util/texture/base.js';
 
-export const g = makeTestGroup(GPUTest);
+import { kSampleTypeInfo, WGSLTextureQueryTest } from './texture_utils.js';
+
+function getLevelSettingsAndExpected(viewType: 'full' | 'partial', mipLevelCount: number) {
+  return viewType === 'partial'
+    ? {
+        baseMipLevel: 1,
+        mipLevelCount: 2,
+        expected: [2],
+      }
+    : {
+        baseMipLevel: 0,
+        mipLevelCount,
+        expected: [mipLevelCount],
+      };
+}
+
+const kTextureTypeToViewDimension = {
+  texture_1d: '1d',
+  texture_2d: '2d',
+  texture_2d_array: '2d-array',
+  texture_3d: '3d',
+  texture_cube: 'cube',
+  texture_cube_array: 'cube-array',
+  texture_depth_2d: '2d',
+  texture_depth_2d_array: '2d-array',
+  texture_depth_cube: 'cube',
+  texture_depth_cube_array: 'cube-array',
+} as const;
+
+export const g = makeTestGroup(WGSLTextureQueryTest);
 
 g.test('sampled')
   .specURL('https://www.w3.org/TR/WGSL/#texturenumlevels')
@@ -34,12 +63,60 @@ Parameters
         'texture_2d_array',
         'texture_3d',
         'texture_cube',
-        'texture_cube_array`',
+        'texture_cube_array',
       ] as const)
       .beginSubcases()
-      .combine('sampled_type', ['f32-only', 'i32', 'u32'] as const)
+      .combine('sampled_type', ['f32', 'i32', 'u32'] as const)
+      .combine('view_type', ['full', 'partial'] as const)
+      // 1d textures can't have mipLevelCount > 0
+      .filter(t => t.texture_type !== 'texture_1d' || t.view_type !== 'partial')
   )
-  .unimplemented();
+  .beforeAllSubcases(t => {
+    t.skipIfTextureViewDimensionNotSupported(kTextureTypeToViewDimension[t.params.texture_type]);
+  })
+  .fn(t => {
+    const { texture_type, sampled_type, view_type } = t.params;
+    const { format } = kSampleTypeInfo[sampled_type];
+
+    const viewDimension = kTextureTypeToViewDimension[texture_type];
+    const dimension = getTextureDimensionFromView(viewDimension);
+    const isCube = texture_type.includes('cube');
+    const width = 64;
+    const height = dimension === '1d' ? 1 : width;
+    const depthOrArrayLayers = isCube ? 6 : 1;
+    const mipCount = dimension === '1d' ? 1 : 4;
+    const texture = t.createTextureTracked({
+      format,
+      dimension,
+      usage: GPUTextureUsage.TEXTURE_BINDING,
+      size: {
+        width,
+        height,
+        depthOrArrayLayers,
+      },
+      mipLevelCount: mipCount,
+    });
+
+    const code = `
+@group(0) @binding(0) var t: ${texture_type}<${sampled_type}>;
+@group(0) @binding(1) var<storage, read_write> result: u32;
+@compute @workgroup_size(1) fn cs() {
+  result = textureNumLevels(t);
+}
+    `;
+
+    const { baseMipLevel, mipLevelCount, expected } = getLevelSettingsAndExpected(
+      view_type,
+      mipCount
+    );
+    const view = texture.createView({
+      dimension: viewDimension,
+      baseMipLevel,
+      mipLevelCount,
+    });
+
+    t.executeAndExpectResult(code, view, expected);
+  });
 
 g.test('depth')
   .specURL('https://www.w3.org/TR/WGSL/#texturenumlevels')
@@ -55,11 +132,57 @@ Parameters
 `
   )
   .params(u =>
-    u.combine('texture_type', [
-      'texture_depth_2d',
-      'texture_depth_2d_array',
-      'texture_depth_cube',
-      'texture_depth_cube_array',
-    ] as const)
+    u
+      .combine('texture_type', [
+        'texture_depth_2d',
+        'texture_depth_2d_array',
+        'texture_depth_cube',
+        'texture_depth_cube_array',
+      ] as const)
+      .combine('view_type', ['full', 'partial'] as const)
   )
-  .unimplemented();
+  .beforeAllSubcases(t => {
+    t.skipIfTextureViewDimensionNotSupported(kTextureTypeToViewDimension[t.params.texture_type]);
+  })
+  .fn(t => {
+    const { texture_type, view_type } = t.params;
+
+    const viewDimension = kTextureTypeToViewDimension[texture_type];
+    const dimension = getTextureDimensionFromView(viewDimension);
+    const isCube = texture_type.includes('cube');
+    const width = 64;
+    const height = dimension === '1d' ? 1 : width;
+    const depthOrArrayLayers = isCube ? 6 : 1;
+    const mipCount = dimension === '1d' ? 1 : 4;
+    const texture = t.createTextureTracked({
+      format: 'depth32float',
+      dimension,
+      usage: GPUTextureUsage.TEXTURE_BINDING,
+      size: {
+        width,
+        height,
+        depthOrArrayLayers,
+      },
+      mipLevelCount: mipCount,
+    });
+
+    const code = `
+@group(0) @binding(0) var t: ${texture_type};
+@group(0) @binding(1) var<storage, read_write> result: u32;
+@compute @workgroup_size(1) fn cs() {
+  result = textureNumLevels(t);
+}
+    `;
+
+    const { baseMipLevel, mipLevelCount, expected } = getLevelSettingsAndExpected(
+      view_type,
+      mipCount
+    );
+    const view = texture.createView({
+      dimension: viewDimension,
+      baseMipLevel,
+      mipLevelCount,
+    });
+
+    t.executeAndExpectResult(code, view, expected);
+  });