Allow copying one layer of depth/stencil textures (#8020)

This fixes a regression introduced by #7935.

Author: andyleiserson

cts_runner/test.lst

@@ -5,6 +5,12 @@ webgpu:api,operation,compute,basic:memcpy:*
 //FAIL: webgpu:api,operation,compute,basic:large_dispatch:*
 webgpu:api,operation,compute_pipeline,overrides:*
 webgpu:api,operation,device,lost:*
+// This is all the storeOp tests, but some of them fail if run in a single invocation.
+// https://github.com/gfx-rs/wgpu/issues/8021
+webgpu:api,operation,render_pass,storeOp:render_pass_store_op,color_attachment_with_depth_stencil_attachment:*
+webgpu:api,operation,render_pass,storeOp:render_pass_store_op,color_attachment_only:*
+webgpu:api,operation,render_pass,storeOp:render_pass_store_op,multiple_color_attachments:*
+webgpu:api,operation,render_pass,storeOp:render_pass_store_op,depth_stencil_attachment_only:*
 webgpu:api,validation,encoding,beginComputePass:*
 webgpu:api,validation,encoding,beginRenderPass:*
 webgpu:api,validation,encoding,cmds,clearBuffer:*

wgpu-core/src/command/transfer.rs

@@ -455,15 +455,17 @@ pub(crate) fn validate_texture_copy_range<T>(
     // physical size can be larger than the virtual
     let extent = extent_virtual.physical_size(desc.format);
 
-    // Multisampled and depth-stencil formats do not support partial copies.
+    // Multisampled and depth-stencil formats do not support partial copies
+    // on x and y dimensions, but do support copying a subset of layers.
     let requires_exact_size = desc.format.is_depth_stencil_format() || desc.sample_count > 1;
 
     // Return `Ok` if a run `size` texels long starting at `start_offset` is
     // valid for `texture_size`. Otherwise, return an appropriate a`Err`.
     let check_dimension = |dimension: TextureErrorDimension,
                            start_offset: u32,
                            size: u32,
-                           texture_size: u32|
+                           texture_size: u32,
+                           requires_exact_size: bool|
      -> Result<(), TransferError> {
         if requires_exact_size && (start_offset != 0 || size != texture_size) {
             Err(TransferError::UnsupportedPartialTransfer {
@@ -495,18 +497,21 @@ pub(crate) fn validate_texture_copy_range<T>(
         texture_copy_view.origin.x,
         copy_size.width,
         extent.width,
+        requires_exact_size,
     )?;
     check_dimension(
         TextureErrorDimension::Y,
         texture_copy_view.origin.y,
         copy_size.height,
         extent.height,
+        requires_exact_size,
     )?;
     check_dimension(
         TextureErrorDimension::Z,
         texture_copy_view.origin.z,
         copy_size.depth_or_array_layers,
         extent.depth_or_array_layers,
+        false, // partial copy always allowed on Z/layer dimension
     )?;
 
     if texture_copy_view.origin.x % block_width != 0 {