Compile error when function declare after '@compute @workgroup_size'

[haoyunfeix]

Failed shader:

  @compute @workgroup_size(256, 1, 1)
        fn A(index : i32)
       {
        }
        fn main(@builtin(local_invocation_id) LocalId : vec3<u32>,
                  @builtin(global_invocation_id) GlobalId : vec3<u32>,
                  @builtin(num_workgroups) NumWorkgroups : vec3<u32>) {
          var localId = LocalId;
          var globalId = GlobalId;
          var numWorkgroups = NumWorkgroups;
          A(getGlobalIndex());

naga output:

$ naga after.wgsl
error: unknown local function `A`
   鈹屸攢 after.wgsl:11:11
   鈹?[0m
11 鈹?[0m           A(getGlobalIndex());
   鈹?[0m           ^ unknown local function

Could not parse WGSL

This could be fixed after moved A() declaration before @compute @workgroup_size(256, 1, 1)

[haoyunfeix]

Dup with #2072

[vicary]

Thanks @haoyunfeix!