FFT ocean surface demo non functional in firefox

Uncaptured WebGPU error: Shader module creation failed: Shader validation error
Encountered one or more errors while creating shader module "compute.shader"
WebGPU compilation info for shader module "compute.shader" (1 error(s), 0 warning(s), 0 info)
Uncaptured WebGPU error: ShaderModule with 'compute.shader' label is invalid
Encountered one or more errors while creating shader module "compute.shader"
WebGPU compilation info for shader module "compute.shader" (1 error(s), 0 warning(s), 0 info)
Uncaptured WebGPU error: In a set_pipeline command, caused by: ComputePipeline with 'compute.pipeline' label is invalid

Shader validation error: Function [2] '_vgsl_cf2a17d3__fftStages3' is invalid
   ┌─ compute.shader:59:1
   │  
59 │ ╭ fn _vgsl_cf2a17d3__fftStages3(
60 │ │   a: ptr<workgroup, array<vec2f, 256>>,
61 │ │   b: ptr<workgroup, array<vec2f, 256>>,
62 │ │   c: ptr<workgroup, array<vec2f, 256>>,
   · │
90 │ │   }
91 │ │ }
   │ ╰─^ naga::ir::Function [2]
   │  
   = Argument 'a' at index 0 is a pointer of space WorkGroup, which can't be passed into functions.