Fix SPIR-V code generation for PTX special registers (vosen#21)

We currently directly map PTX special registers: %ntid, %tid, etc. to SPIR-V builtins with type OpTypeVector %uint 4.
This is wrong and leads to a silent corruption, which fails e.g. Depth of Field in GeekBench

ptx/src/test/spirv_run/ntid.spvtxt

@@ -12,16 +12,16 @@
                OpEntryPoint Kernel %1 "ntid" %gl_WorkGroupSize
                OpDecorate %gl_WorkGroupSize BuiltIn WorkgroupSize
        %void = OpTypeVoid
-       %uint = OpTypeInt 32 0
-     %v4uint = OpTypeVector %uint 4
-%_ptr_UniformConstant_v4uint = OpTypePointer UniformConstant %v4uint
-%gl_WorkGroupSize = OpVariable %_ptr_UniformConstant_v4uint UniformConstant
       %ulong = OpTypeInt 64 0
-         %33 = OpTypeFunction %void %ulong %ulong
+    %v3ulong = OpTypeVector %ulong 3
+%_ptr_Input_v3ulong = OpTypePointer Input %v3ulong
+%gl_WorkGroupSize = OpVariable %_ptr_Input_v3ulong Input
+         %32 = OpTypeFunction %void %ulong %ulong
 %_ptr_Function_ulong = OpTypePointer Function %ulong
+       %uint = OpTypeInt 32 0
 %_ptr_Function_uint = OpTypePointer Function %uint
 %_ptr_Generic_uint = OpTypePointer Generic %uint
-          %1 = OpFunction %void None %33
+          %1 = OpFunction %void None %32
           %9 = OpFunctionParameter %ulong
          %10 = OpFunctionParameter %ulong
          %25 = OpLabel
@@ -41,9 +41,11 @@
          %23 = OpConvertUToPtr %_ptr_Generic_uint %14
          %13 = OpLoad %uint %23
                OpStore %6 %13
-         %16 = OpLoad %v4uint %gl_WorkGroupSize
-         %22 = OpCompositeExtract %uint %16 0
-         %15 = OpCopyObject %uint %22
+         %37 = OpLoad %v3ulong %gl_WorkGroupSize
+         %22 = OpCompositeExtract %ulong %37 0
+         %38 = OpBitcast %ulong %22
+         %16 = OpUConvert %uint %38
+         %15 = OpCopyObject %uint %16
                OpStore %7 %15
          %18 = OpLoad %uint %6
          %19 = OpLoad %uint %7

ptx/src/test/spirv_run/stateful_ld_st_ntid.spvtxt

@@ -12,16 +12,16 @@
                OpEntryPoint Kernel %1 "stateful_ld_st_ntid" %gl_LocalInvocationID
                OpDecorate %gl_LocalInvocationID BuiltIn LocalInvocationId
        %void = OpTypeVoid
-       %uint = OpTypeInt 32 0
-     %v4uint = OpTypeVector %uint 4
-%_ptr_Input_v4uint = OpTypePointer Input %v4uint
-%gl_LocalInvocationID = OpVariable %_ptr_Input_v4uint Input
+      %ulong = OpTypeInt 64 0
+    %v3ulong = OpTypeVector %ulong 3
+%_ptr_Input_v3ulong = OpTypePointer Input %v3ulong
+%gl_LocalInvocationID = OpVariable %_ptr_Input_v3ulong Input
       %uchar = OpTypeInt 8 0
 %_ptr_CrossWorkgroup_uchar = OpTypePointer CrossWorkgroup %uchar
          %56 = OpTypeFunction %void %_ptr_CrossWorkgroup_uchar %_ptr_CrossWorkgroup_uchar
 %_ptr_Function__ptr_CrossWorkgroup_uchar = OpTypePointer Function %_ptr_CrossWorkgroup_uchar
+       %uint = OpTypeInt 32 0
 %_ptr_Function_uint = OpTypePointer Function %uint
-      %ulong = OpTypeInt 64 0
 %_ptr_Function_ulong = OpTypePointer Function %ulong
 %_ptr_CrossWorkgroup_ulong = OpTypePointer CrossWorkgroup %ulong
           %1 = OpFunction %void None %56
@@ -57,25 +57,27 @@
          %18 = OpCopyObject %ulong %19
          %27 = OpConvertUToPtr %_ptr_CrossWorkgroup_uchar %18
                OpStore %11 %27
-         %29 = OpLoad %v4uint %gl_LocalInvocationID
-         %42 = OpCompositeExtract %uint %29 0
-         %28 = OpCopyObject %uint %42
+         %61 = OpLoad %v3ulong %gl_LocalInvocationID
+         %42 = OpCompositeExtract %ulong %61 0
+         %62 = OpBitcast %ulong %42
+         %29 = OpUConvert %uint %62
+         %28 = OpCopyObject %uint %29
                OpStore %6 %28
          %31 = OpLoad %uint %6
-         %61 = OpBitcast %uint %31
-         %30 = OpUConvert %ulong %61
+         %63 = OpBitcast %uint %31
+         %30 = OpUConvert %ulong %63
                OpStore %7 %30
          %33 = OpLoad %_ptr_CrossWorkgroup_uchar %10
          %34 = OpLoad %ulong %7
-         %62 = OpBitcast %_ptr_CrossWorkgroup_uchar %33
-         %63 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %62 %34
-         %32 = OpBitcast %_ptr_CrossWorkgroup_uchar %63
+         %64 = OpBitcast %_ptr_CrossWorkgroup_uchar %33
+         %65 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %64 %34
+         %32 = OpBitcast %_ptr_CrossWorkgroup_uchar %65
                OpStore %10 %32
          %36 = OpLoad %_ptr_CrossWorkgroup_uchar %11
          %37 = OpLoad %ulong %7
-         %64 = OpBitcast %_ptr_CrossWorkgroup_uchar %36
-         %65 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %64 %37
-         %35 = OpBitcast %_ptr_CrossWorkgroup_uchar %65
+         %66 = OpBitcast %_ptr_CrossWorkgroup_uchar %36
+         %67 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %66 %37
+         %35 = OpBitcast %_ptr_CrossWorkgroup_uchar %67
                OpStore %11 %35
          %39 = OpLoad %_ptr_CrossWorkgroup_uchar %10
          %45 = OpBitcast %_ptr_CrossWorkgroup_ulong %39

ptx/src/test/spirv_run/stateful_ld_st_ntid_chain.spvtxt

@@ -12,16 +12,16 @@
                OpEntryPoint Kernel %1 "stateful_ld_st_ntid_chain" %gl_LocalInvocationID
                OpDecorate %gl_LocalInvocationID BuiltIn LocalInvocationId
        %void = OpTypeVoid
-       %uint = OpTypeInt 32 0
-     %v4uint = OpTypeVector %uint 4
-%_ptr_Input_v4uint = OpTypePointer Input %v4uint
-%gl_LocalInvocationID = OpVariable %_ptr_Input_v4uint Input
+      %ulong = OpTypeInt 64 0
+    %v3ulong = OpTypeVector %ulong 3
+%_ptr_Input_v3ulong = OpTypePointer Input %v3ulong
+%gl_LocalInvocationID = OpVariable %_ptr_Input_v3ulong Input
       %uchar = OpTypeInt 8 0
 %_ptr_CrossWorkgroup_uchar = OpTypePointer CrossWorkgroup %uchar
          %64 = OpTypeFunction %void %_ptr_CrossWorkgroup_uchar %_ptr_CrossWorkgroup_uchar
 %_ptr_Function__ptr_CrossWorkgroup_uchar = OpTypePointer Function %_ptr_CrossWorkgroup_uchar
+       %uint = OpTypeInt 32 0
 %_ptr_Function_uint = OpTypePointer Function %uint
-      %ulong = OpTypeInt 64 0
 %_ptr_Function_ulong = OpTypePointer Function %ulong
 %_ptr_CrossWorkgroup_ulong = OpTypePointer CrossWorkgroup %ulong
           %1 = OpFunction %void None %64
@@ -61,25 +61,27 @@
          %26 = OpCopyObject %ulong %27
          %35 = OpConvertUToPtr %_ptr_CrossWorkgroup_uchar %26
                OpStore %18 %35
-         %37 = OpLoad %v4uint %gl_LocalInvocationID
-         %50 = OpCompositeExtract %uint %37 0
-         %36 = OpCopyObject %uint %50
+         %69 = OpLoad %v3ulong %gl_LocalInvocationID
+         %50 = OpCompositeExtract %ulong %69 0
+         %70 = OpBitcast %ulong %50
+         %37 = OpUConvert %uint %70
+         %36 = OpCopyObject %uint %37
                OpStore %10 %36
          %39 = OpLoad %uint %10
-         %69 = OpBitcast %uint %39
-         %38 = OpUConvert %ulong %69
+         %71 = OpBitcast %uint %39
+         %38 = OpUConvert %ulong %71
                OpStore %11 %38
          %41 = OpLoad %_ptr_CrossWorkgroup_uchar %15
          %42 = OpLoad %ulong %11
-         %70 = OpBitcast %_ptr_CrossWorkgroup_uchar %41
-         %71 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %70 %42
-         %40 = OpBitcast %_ptr_CrossWorkgroup_uchar %71
+         %72 = OpBitcast %_ptr_CrossWorkgroup_uchar %41
+         %73 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %72 %42
+         %40 = OpBitcast %_ptr_CrossWorkgroup_uchar %73
                OpStore %16 %40
          %44 = OpLoad %_ptr_CrossWorkgroup_uchar %18
          %45 = OpLoad %ulong %11
-         %72 = OpBitcast %_ptr_CrossWorkgroup_uchar %44
-         %73 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %72 %45
-         %43 = OpBitcast %_ptr_CrossWorkgroup_uchar %73
+         %74 = OpBitcast %_ptr_CrossWorkgroup_uchar %44
+         %75 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %74 %45
+         %43 = OpBitcast %_ptr_CrossWorkgroup_uchar %75
                OpStore %19 %43
          %47 = OpLoad %_ptr_CrossWorkgroup_uchar %16
          %53 = OpBitcast %_ptr_CrossWorkgroup_ulong %47

ptx/src/test/spirv_run/stateful_ld_st_ntid_sub.spvtxt

@@ -12,16 +12,16 @@
                OpEntryPoint Kernel %1 "stateful_ld_st_ntid_sub" %gl_LocalInvocationID
                OpDecorate %gl_LocalInvocationID BuiltIn LocalInvocationId
        %void = OpTypeVoid
-       %uint = OpTypeInt 32 0
-     %v4uint = OpTypeVector %uint 4
-%_ptr_Input_v4uint = OpTypePointer Input %v4uint
-%gl_LocalInvocationID = OpVariable %_ptr_Input_v4uint Input
+      %ulong = OpTypeInt 64 0
+    %v3ulong = OpTypeVector %ulong 3
+%_ptr_Input_v3ulong = OpTypePointer Input %v3ulong
+%gl_LocalInvocationID = OpVariable %_ptr_Input_v3ulong Input
       %uchar = OpTypeInt 8 0
 %_ptr_CrossWorkgroup_uchar = OpTypePointer CrossWorkgroup %uchar
          %72 = OpTypeFunction %void %_ptr_CrossWorkgroup_uchar %_ptr_CrossWorkgroup_uchar
 %_ptr_Function__ptr_CrossWorkgroup_uchar = OpTypePointer Function %_ptr_CrossWorkgroup_uchar
+       %uint = OpTypeInt 32 0
 %_ptr_Function_uint = OpTypePointer Function %uint
-      %ulong = OpTypeInt 64 0
 %_ptr_Function_ulong = OpTypePointer Function %ulong
     %ulong_0 = OpConstant %ulong 0
 %_ptr_CrossWorkgroup_ulong = OpTypePointer CrossWorkgroup %ulong
@@ -63,43 +63,45 @@
          %26 = OpCopyObject %ulong %27
          %37 = OpConvertUToPtr %_ptr_CrossWorkgroup_uchar %26
                OpStore %18 %37
-         %39 = OpLoad %v4uint %gl_LocalInvocationID
-         %52 = OpCompositeExtract %uint %39 0
-         %38 = OpCopyObject %uint %52
+         %77 = OpLoad %v3ulong %gl_LocalInvocationID
+         %52 = OpCompositeExtract %ulong %77 0
+         %78 = OpBitcast %ulong %52
+         %39 = OpUConvert %uint %78
+         %38 = OpCopyObject %uint %39
                OpStore %10 %38
          %41 = OpLoad %uint %10
-         %77 = OpBitcast %uint %41
-         %40 = OpUConvert %ulong %77
+         %79 = OpBitcast %uint %41
+         %40 = OpUConvert %ulong %79
                OpStore %11 %40
          %42 = OpLoad %ulong %11
          %59 = OpCopyObject %ulong %42
          %28 = OpSNegate %ulong %59
          %44 = OpLoad %_ptr_CrossWorkgroup_uchar %15
-         %78 = OpBitcast %_ptr_CrossWorkgroup_uchar %44
-         %79 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %78 %28
-         %43 = OpBitcast %_ptr_CrossWorkgroup_uchar %79
+         %80 = OpBitcast %_ptr_CrossWorkgroup_uchar %44
+         %81 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %80 %28
+         %43 = OpBitcast %_ptr_CrossWorkgroup_uchar %81
                OpStore %16 %43
          %45 = OpLoad %ulong %11
          %60 = OpCopyObject %ulong %45
          %29 = OpSNegate %ulong %60
          %47 = OpLoad %_ptr_CrossWorkgroup_uchar %18
-         %80 = OpBitcast %_ptr_CrossWorkgroup_uchar %47
-         %81 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %80 %29
-         %46 = OpBitcast %_ptr_CrossWorkgroup_uchar %81
+         %82 = OpBitcast %_ptr_CrossWorkgroup_uchar %47
+         %83 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %82 %29
+         %46 = OpBitcast %_ptr_CrossWorkgroup_uchar %83
                OpStore %19 %46
          %49 = OpLoad %_ptr_CrossWorkgroup_uchar %16
          %61 = OpBitcast %_ptr_CrossWorkgroup_ulong %49
-         %83 = OpBitcast %_ptr_CrossWorkgroup_uchar %61
-         %84 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %83 %ulong_0
-         %54 = OpBitcast %_ptr_CrossWorkgroup_ulong %84
+         %85 = OpBitcast %_ptr_CrossWorkgroup_uchar %61
+         %86 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %85 %ulong_0
+         %54 = OpBitcast %_ptr_CrossWorkgroup_ulong %86
          %48 = OpLoad %ulong %54
                OpStore %12 %48
          %50 = OpLoad %_ptr_CrossWorkgroup_uchar %19
          %51 = OpLoad %ulong %12
          %62 = OpBitcast %_ptr_CrossWorkgroup_ulong %50
-         %85 = OpBitcast %_ptr_CrossWorkgroup_uchar %62
-         %86 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %85 %ulong_0_0
-         %56 = OpBitcast %_ptr_CrossWorkgroup_ulong %86
+         %87 = OpBitcast %_ptr_CrossWorkgroup_uchar %62
+         %88 = OpInBoundsPtrAccessChain %_ptr_CrossWorkgroup_uchar %87 %ulong_0_0
+         %56 = OpBitcast %_ptr_CrossWorkgroup_ulong %88
                OpStore %56 %51
                OpReturn
                OpFunctionEnd