Fix type mismatch in wgpuRenderBundleEncoderSetPushConstants

Updated the shader stage type from WGPUShaderStageFlags to WGPUShaderStage to align with API requirements. Adjusted conversion logic to handle the new type safely and ensure compatibility with associated functionality.

ffi/wgpu.h

@@ -271,7 +271,7 @@ uint32_t wgpuGetVersion(void);
 
 void wgpuRenderPassEncoderSetPushConstants(WGPURenderPassEncoder encoder, WGPUShaderStage stages, uint32_t offset, uint32_t sizeBytes, void const * data);
 void wgpuComputePassEncoderSetPushConstants(WGPUComputePassEncoder encoder, uint32_t offset, uint32_t sizeBytes, void const * data);
-void wgpuRenderBundleEncoderSetPushConstants(WGPURenderBundleEncoder encoder, WGPUShaderStageFlags stages, uint32_t offset, uint32_t sizeBytes, void const * data);
+void wgpuRenderBundleEncoderSetPushConstants(WGPURenderBundleEncoder encoder, WGPUShaderStage stages, uint32_t offset, uint32_t sizeBytes, void const * data);
 
 void wgpuRenderPassEncoderMultiDrawIndirect(WGPURenderPassEncoder encoder, WGPUBuffer buffer, uint64_t offset, uint32_t count);
 void wgpuRenderPassEncoderMultiDrawIndexedIndirect(WGPURenderPassEncoder encoder, WGPUBuffer buffer, uint64_t offset, uint32_t count);

src/lib.rs

@@ -4361,7 +4361,7 @@ pub unsafe extern "C" fn wgpuComputePassEncoderSetPushConstants(
 #[no_mangle]
 pub unsafe extern "C" fn wgpuRenderBundleEncoderSetPushConstants(
     bundle: native::WGPURenderBundleEncoder,
-    stages: native::WGPUShaderStageFlags,
+    stages: native::WGPUShaderStage,
     offset: u32,
     size_bytes: u32,
     data: *const u8,
@@ -4373,7 +4373,7 @@ pub unsafe extern "C" fn wgpuRenderBundleEncoderSetPushConstants(
 
     bundle_ffi::wgpu_render_bundle_set_push_constants(
         encoder,
-        wgt::ShaderStages::from_bits(stages).expect("invalid shader stage"),
+        wgt::ShaderStages::from_bits(stages.try_into().unwrap()).expect("invalid shader stage"),
         offset,
         size_bytes,
         data,