spv.subgroupClustered.comp 7.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. #version 450
  2. #extension GL_KHR_shader_subgroup_clustered: enable
  3. layout (local_size_x = 8) in;
  4. layout(binding = 0) buffer Buffers
  5. {
  6. vec4 f4;
  7. ivec4 i4;
  8. uvec4 u4;
  9. dvec4 d4;
  10. } data[4];
  11. void main()
  12. {
  13. uint invocation = (gl_SubgroupInvocationID + gl_SubgroupSize) % 4;
  14. data[invocation].f4.x = subgroupClusteredAdd(data[0].f4.x, 1);
  15. data[invocation].f4.xy = subgroupClusteredAdd(data[1].f4.xy, 1);
  16. data[invocation].f4.xyz = subgroupClusteredAdd(data[2].f4.xyz, 1);
  17. data[invocation].f4 = subgroupClusteredAdd(data[3].f4, 1);
  18. data[invocation].i4.x = subgroupClusteredAdd(data[0].i4.x, 1);
  19. data[invocation].i4.xy = subgroupClusteredAdd(data[1].i4.xy, 1);
  20. data[invocation].i4.xyz = subgroupClusteredAdd(data[2].i4.xyz, 1);
  21. data[invocation].i4 = subgroupClusteredAdd(data[3].i4, 1);
  22. data[invocation].u4.x = subgroupClusteredAdd(data[0].u4.x, 1);
  23. data[invocation].u4.xy = subgroupClusteredAdd(data[1].u4.xy, 1);
  24. data[invocation].u4.xyz = subgroupClusteredAdd(data[2].u4.xyz, 1);
  25. data[invocation].u4 = subgroupClusteredAdd(data[3].u4, 1);
  26. data[invocation].d4.x = subgroupClusteredAdd(data[0].d4.x, 1);
  27. data[invocation].d4.xy = subgroupClusteredAdd(data[1].d4.xy, 1);
  28. data[invocation].d4.xyz = subgroupClusteredAdd(data[2].d4.xyz, 1);
  29. data[invocation].d4 = subgroupClusteredAdd(data[3].d4, 1);
  30. data[invocation].f4.x = subgroupClusteredMul(data[0].f4.x, 1);
  31. data[invocation].f4.xy = subgroupClusteredMul(data[1].f4.xy, 1);
  32. data[invocation].f4.xyz = subgroupClusteredMul(data[2].f4.xyz, 1);
  33. data[invocation].f4 = subgroupClusteredMul(data[3].f4, 1);
  34. data[invocation].i4.x = subgroupClusteredMul(data[0].i4.x, 1);
  35. data[invocation].i4.xy = subgroupClusteredMul(data[1].i4.xy, 1);
  36. data[invocation].i4.xyz = subgroupClusteredMul(data[2].i4.xyz, 1);
  37. data[invocation].i4 = subgroupClusteredMul(data[3].i4, 1);
  38. data[invocation].u4.x = subgroupClusteredMul(data[0].u4.x, 1);
  39. data[invocation].u4.xy = subgroupClusteredMul(data[1].u4.xy, 1);
  40. data[invocation].u4.xyz = subgroupClusteredMul(data[2].u4.xyz, 1);
  41. data[invocation].u4 = subgroupClusteredMul(data[3].u4, 1);
  42. data[invocation].d4.x = subgroupClusteredMul(data[0].d4.x, 1);
  43. data[invocation].d4.xy = subgroupClusteredMul(data[1].d4.xy, 1);
  44. data[invocation].d4.xyz = subgroupClusteredMul(data[2].d4.xyz, 1);
  45. data[invocation].d4 = subgroupClusteredMul(data[3].d4, 1);
  46. data[invocation].f4.x = subgroupClusteredMin(data[0].f4.x, 1);
  47. data[invocation].f4.xy = subgroupClusteredMin(data[1].f4.xy, 1);
  48. data[invocation].f4.xyz = subgroupClusteredMin(data[2].f4.xyz, 1);
  49. data[invocation].f4 = subgroupClusteredMin(data[3].f4, 1);
  50. data[invocation].i4.x = subgroupClusteredMin(data[0].i4.x, 1);
  51. data[invocation].i4.xy = subgroupClusteredMin(data[1].i4.xy, 1);
  52. data[invocation].i4.xyz = subgroupClusteredMin(data[2].i4.xyz, 1);
  53. data[invocation].i4 = subgroupClusteredMin(data[3].i4, 1);
  54. data[invocation].u4.x = subgroupClusteredMin(data[0].u4.x, 1);
  55. data[invocation].u4.xy = subgroupClusteredMin(data[1].u4.xy, 1);
  56. data[invocation].u4.xyz = subgroupClusteredMin(data[2].u4.xyz, 1);
  57. data[invocation].u4 = subgroupClusteredMin(data[3].u4, 1);
  58. data[invocation].d4.x = subgroupClusteredMin(data[0].d4.x, 1);
  59. data[invocation].d4.xy = subgroupClusteredMin(data[1].d4.xy, 1);
  60. data[invocation].d4.xyz = subgroupClusteredMin(data[2].d4.xyz, 1);
  61. data[invocation].d4 = subgroupClusteredMin(data[3].d4, 1);
  62. data[invocation].f4.x = subgroupClusteredMax(data[0].f4.x, 1);
  63. data[invocation].f4.xy = subgroupClusteredMax(data[1].f4.xy, 1);
  64. data[invocation].f4.xyz = subgroupClusteredMax(data[2].f4.xyz, 1);
  65. data[invocation].f4 = subgroupClusteredMax(data[3].f4, 1);
  66. data[invocation].i4.x = subgroupClusteredMax(data[0].i4.x, 1);
  67. data[invocation].i4.xy = subgroupClusteredMax(data[1].i4.xy, 1);
  68. data[invocation].i4.xyz = subgroupClusteredMax(data[2].i4.xyz, 1);
  69. data[invocation].i4 = subgroupClusteredMax(data[3].i4, 1);
  70. data[invocation].u4.x = subgroupClusteredMax(data[0].u4.x, 1);
  71. data[invocation].u4.xy = subgroupClusteredMax(data[1].u4.xy, 1);
  72. data[invocation].u4.xyz = subgroupClusteredMax(data[2].u4.xyz, 1);
  73. data[invocation].u4 = subgroupClusteredMax(data[3].u4, 1);
  74. data[invocation].d4.x = subgroupClusteredMax(data[0].d4.x, 1);
  75. data[invocation].d4.xy = subgroupClusteredMax(data[1].d4.xy, 1);
  76. data[invocation].d4.xyz = subgroupClusteredMax(data[2].d4.xyz, 1);
  77. data[invocation].d4 = subgroupClusteredMax(data[3].d4, 1);
  78. data[invocation].i4.x = subgroupClusteredAnd(data[0].i4.x, 1);
  79. data[invocation].i4.xy = subgroupClusteredAnd(data[1].i4.xy, 1);
  80. data[invocation].i4.xyz = subgroupClusteredAnd(data[2].i4.xyz, 1);
  81. data[invocation].i4 = subgroupClusteredAnd(data[3].i4, 1);
  82. data[invocation].u4.x = subgroupClusteredAnd(data[0].u4.x, 1);
  83. data[invocation].u4.xy = subgroupClusteredAnd(data[1].u4.xy, 1);
  84. data[invocation].u4.xyz = subgroupClusteredAnd(data[2].u4.xyz, 1);
  85. data[invocation].u4 = subgroupClusteredAnd(data[3].u4, 1);
  86. data[invocation].i4.x = int(subgroupClusteredAnd(data[0].i4.x < 0, 1));
  87. data[invocation].i4.xy = ivec2(subgroupClusteredAnd(lessThan(data[1].i4.xy, ivec2(0)), 1));
  88. data[invocation].i4.xyz = ivec3(subgroupClusteredAnd(lessThan(data[1].i4.xyz, ivec3(0)), 1));
  89. data[invocation].i4 = ivec4(subgroupClusteredAnd(lessThan(data[1].i4, ivec4(0)), 1));
  90. data[invocation].i4.x = subgroupClusteredOr(data[0].i4.x, 1);
  91. data[invocation].i4.xy = subgroupClusteredOr(data[1].i4.xy, 1);
  92. data[invocation].i4.xyz = subgroupClusteredOr(data[2].i4.xyz, 1);
  93. data[invocation].i4 = subgroupClusteredOr(data[3].i4, 1);
  94. data[invocation].u4.x = subgroupClusteredOr(data[0].u4.x, 1);
  95. data[invocation].u4.xy = subgroupClusteredOr(data[1].u4.xy, 1);
  96. data[invocation].u4.xyz = subgroupClusteredOr(data[2].u4.xyz, 1);
  97. data[invocation].u4 = subgroupClusteredOr(data[3].u4, 1);
  98. data[invocation].i4.x = int(subgroupClusteredOr(data[0].i4.x < 0, 1));
  99. data[invocation].i4.xy = ivec2(subgroupClusteredOr(lessThan(data[1].i4.xy, ivec2(0)), 1));
  100. data[invocation].i4.xyz = ivec3(subgroupClusteredOr(lessThan(data[1].i4.xyz, ivec3(0)), 1));
  101. data[invocation].i4 = ivec4(subgroupClusteredOr(lessThan(data[1].i4, ivec4(0)), 1));
  102. data[invocation].i4.x = subgroupClusteredXor(data[0].i4.x, 1);
  103. data[invocation].i4.xy = subgroupClusteredXor(data[1].i4.xy, 1);
  104. data[invocation].i4.xyz = subgroupClusteredXor(data[2].i4.xyz, 1);
  105. data[invocation].i4 = subgroupClusteredXor(data[3].i4, 1);
  106. data[invocation].u4.x = subgroupClusteredXor(data[0].u4.x, 1);
  107. data[invocation].u4.xy = subgroupClusteredXor(data[1].u4.xy, 1);
  108. data[invocation].u4.xyz = subgroupClusteredXor(data[2].u4.xyz, 1);
  109. data[invocation].u4 = subgroupClusteredXor(data[3].u4, 1);
  110. data[invocation].i4.x = int(subgroupClusteredXor(data[0].i4.x < 0, 1));
  111. data[invocation].i4.xy = ivec2(subgroupClusteredXor(lessThan(data[1].i4.xy, ivec2(0)), 1));
  112. data[invocation].i4.xyz = ivec3(subgroupClusteredXor(lessThan(data[1].i4.xyz, ivec3(0)), 1));
  113. data[invocation].i4 = ivec4(subgroupClusteredXor(lessThan(data[1].i4, ivec4(0)), 1));
  114. }