spv.int16.amd.frag 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315
  1. #version 450 core
  2. #extension GL_ARB_gpu_shader_int64: enable
  3. #extension GL_AMD_gpu_shader_half_float: enable
  4. #extension GL_AMD_gpu_shader_int16: enable
  5. layout(binding = 0) uniform Uniforms
  6. {
  7. uint i;
  8. };
  9. // int16/uint16 in block
  10. layout(std140, binding = 1) uniform Block
  11. {
  12. i16vec3 i16v;
  13. uint16_t u16;
  14. } block;
  15. // int16/uint16 for input
  16. layout(location = 0) in flat u16vec3 iu16v;
  17. layout(location = 1) in flat int16_t ii16;
  18. void literal()
  19. {
  20. const int16_t i16c[3] =
  21. {
  22. 0x111S, // Hex
  23. -2s, // Dec
  24. 0400s, // Oct
  25. };
  26. const uint16_t u16c[] =
  27. {
  28. 0xFFFFus, // Hex
  29. 65535US, // Dec
  30. 0177777us, // Oct
  31. };
  32. uint16_t u16 = i16c[i] + u16c[i];
  33. }
  34. void operators()
  35. {
  36. u16vec3 u16v;
  37. int16_t i16;
  38. uint16_t u16;
  39. int i;
  40. uint u;
  41. bool b;
  42. // Unary
  43. u16v++;
  44. i16--;
  45. ++i16;
  46. --u16v;
  47. u16v = ~u16v;
  48. i16 = +i16;
  49. u16v = -u16v;
  50. // Arithmetic
  51. u16 += i16;
  52. u16v -= u16v;
  53. i16 *= i16;
  54. u16v /= u16v;
  55. u16v %= i16;
  56. u16v = u16v + u16v;
  57. u16 = i16 - u16;
  58. u16v = u16v * i16;
  59. i16 = i16 * i16;
  60. i16 = i16 % i16;
  61. // Shift
  62. u16v <<= i16;
  63. i16 >>= u16v.y;
  64. i16 = i16 << u16v.z;
  65. u16v = u16v << i16;
  66. // Relational
  67. b = (u16v.x != i16);
  68. b = (i16 == u16v.x);
  69. b = (u16v.x > u16v.y);
  70. b = (i16 < u);
  71. b = (u16v.y >= u16v.x);
  72. b = (i16 <= i);
  73. // Bitwise
  74. u16v |= i16;
  75. u16 = i16 | u16;
  76. i16 &= i16;
  77. u16v = u16v & u16v;
  78. u16v ^= i16;
  79. u16v = u16v ^ i16;
  80. }
  81. void typeCast()
  82. {
  83. bvec2 bv;
  84. ivec2 iv;
  85. uvec2 uv;
  86. vec2 fv;
  87. dvec2 dv;
  88. f16vec2 f16v;
  89. i64vec2 i64v;
  90. u64vec2 u64v;
  91. i16vec2 i16v;
  92. u16vec2 u16v;
  93. i16v = i16vec2(bv); // bool -> int16
  94. u16v = u16vec2(bv); // bool -> uint16
  95. bv = bvec2(i16v); // int16 -> bool
  96. bv = bvec2(u16v); // uint16 -> bool
  97. i16v = i16vec2(iv); // int -> int16
  98. u16v = u16vec2(iv); // int -> uint16
  99. iv = i16v; // int16 -> int
  100. iv = ivec2(u16v); // uint16 -> int
  101. i16v = i16vec2(uv); // uint -> int16
  102. u16v = u16vec2(uv); // uint -> uint16
  103. uv = i16v; // int16 -> uint
  104. uv = u16v; // uint16 -> uint
  105. i16v = i16vec2(fv); // float -> int16
  106. u16v = u16vec2(fv); // float -> uint16
  107. fv = i16v; // int16 -> float
  108. fv = u16v; // uint16 -> float
  109. i16v = i16vec2(dv); // double -> int16
  110. u16v = u16vec2(dv); // double -> uint16
  111. dv = i16v; // int16 -> double
  112. dv = u16v; // uint16 -> double
  113. i16v = i16vec2(f16v); // float16 -> int16
  114. u16v = u16vec2(f16v); // float16 -> uint16
  115. f16v = i16v; // int16 -> float16
  116. f16v = u16v; // uint16 -> float16
  117. i16v = i16vec2(i64v); // int64 -> int16
  118. u16v = u16vec2(i64v); // int64 -> uint16
  119. i64v = i16v; // int16 -> int64
  120. i64v = i64vec2(u16v); // uint16 -> int64
  121. i16v = i16vec2(u64v); // uint64 -> int16
  122. u16v = u16vec2(u64v); // uint64 -> uint16
  123. u64v = i16v; // int16 -> uint64
  124. u64v = u16v; // uint16 -> uint64
  125. i16v = i16vec2(u16v); // uint16 -> int16
  126. u16v = i16v; // int16 -> uint16
  127. }
  128. void builtinFuncs()
  129. {
  130. i16vec2 i16v;
  131. u16vec3 u16v;
  132. f16vec3 f16v;
  133. bvec3 bv;
  134. int16_t i16;
  135. uint16_t u16;
  136. // abs()
  137. i16v = abs(i16v);
  138. // sign()
  139. i16v = sign(i16v);
  140. // min()
  141. i16v = min(i16v, i16);
  142. i16v = min(i16v, i16vec2(-1s));
  143. u16v = min(u16v, u16);
  144. u16v = min(u16v, u16vec3(0us));
  145. // max()
  146. i16v = max(i16v, i16);
  147. i16v = max(i16v, i16vec2(-1s));
  148. u16v = max(u16v, u16);
  149. u16v = max(u16v, u16vec3(0us));
  150. // clamp()
  151. i16v = clamp(i16v, -i16, i16);
  152. i16v = clamp(i16v, -i16v, i16v);
  153. u16v = clamp(u16v, -u16, u16);
  154. u16v = clamp(u16v, -u16v, u16v);
  155. // mix()
  156. i16 = mix(i16v.x, i16v.y, true);
  157. i16v = mix(i16vec2(i16), i16vec2(-i16), bvec2(false));
  158. u16 = mix(u16v.x, u16v.y, true);
  159. u16v = mix(u16vec3(u16), u16vec3(-u16), bvec3(false));
  160. // frexp()
  161. i16vec3 exp;
  162. f16v = frexp(f16v, exp);
  163. // ldexp()
  164. f16v = ldexp(f16v, exp);
  165. // float16BitsToInt16()
  166. i16v = float16BitsToInt16(f16v.xy);
  167. // float16BitsToUint16()
  168. u16v.x = float16BitsToUint16(f16v.z);
  169. // int16BitsToFloat16()
  170. f16v.xy = int16BitsToFloat16(i16v);
  171. // uint16BitsToFloat16()
  172. f16v = uint16BitsToFloat16(u16v);
  173. // packInt2x16()
  174. int packi = packInt2x16(i16v);
  175. // unpackInt2x16()
  176. i16v = unpackInt2x16(packi);
  177. // packUint2x16()
  178. uint packu = packUint2x16(u16v.xy);
  179. // unpackUint2x16()
  180. u16v.xy = unpackUint2x16(packu);
  181. // packInt4x16()
  182. int64_t packi64 = packInt4x16(i16vec4(i16));
  183. // unpackInt4x16()
  184. i16v = unpackInt4x16(packi64).xy;
  185. // packUint4x16()
  186. uint64_t packu64 = packUint4x16(u16vec4(u16));
  187. // unpackUint4x16()
  188. u16v = unpackUint4x16(packu64).xyz;
  189. // lessThan()
  190. bv = lessThan(u16v, u16vec3(u16));
  191. bv.xy = lessThan(i16v, i16vec2(i16));
  192. // lessThanEqual()
  193. bv = lessThanEqual(u16v, u16vec3(u16));
  194. bv.xy = lessThanEqual(i16v, i16vec2(i16));
  195. // greaterThan()
  196. bv = greaterThan(u16v, u16vec3(u16));
  197. bv.xy = greaterThan(i16v, i16vec2(i16));
  198. // greaterThanEqual()
  199. bv = greaterThanEqual(u16v, u16vec3(u16));
  200. bv.xy = greaterThanEqual(i16v, i16vec2(i16));
  201. // equal()
  202. bv = equal(u16v, u16vec3(u16));
  203. bv.xy = equal(i16v, i16vec2(i16));
  204. // notEqual()
  205. bv = notEqual(u16v, u16vec3(u16));
  206. bv.xy = notEqual(i16v, i16vec2(i16));
  207. }
  208. // Type conversion for specialization constant
  209. layout(constant_id = 100) const int64_t si64 = -10L;
  210. layout(constant_id = 101) const uint64_t su64 = 20UL;
  211. layout(constant_id = 102) const int si = -5;
  212. layout(constant_id = 103) const uint su = 4;
  213. layout(constant_id = 104) const bool sb = true;
  214. layout(constant_id = 105) const int16_t si16 = -5S;
  215. layout(constant_id = 106) const uint16_t su16 = 4US;
  216. // bool <-> int16/uint16
  217. const bool i16_to_b = bool(si16);
  218. const bool u16_to_b = bool(su16);
  219. const int16_t b_to_i16 = int16_t(sb);
  220. const uint16_t b_to_u16 = uint16_t(sb);
  221. // int <-> int16/uint16
  222. const int i16_to_i = int(si16);
  223. const int u16_to_i = int(su16);
  224. const int16_t i_to_i16 = int16_t(si);
  225. const uint16_t i_to_u16 = uint16_t(si);
  226. // uint <-> int16/uint16
  227. const uint i16_to_u = uint(si16);
  228. const uint u16_to_u = uint(su16);
  229. const int16_t u_to_i16 = int16_t(su);
  230. const uint16_t u_to_u16 = uint16_t(su);
  231. // int64 <-> int16/uint16
  232. const int64_t i16_to_i64 = int64_t(si16);
  233. const int64_t u16_to_i64 = int64_t(su16);
  234. const int16_t i64_to_i16 = int16_t(si64);
  235. const uint16_t i64_to_u16 = uint16_t(si64);
  236. // uint64 <-> int16/uint16
  237. const uint64_t i16_to_u64 = uint64_t(si16);
  238. const uint64_t u16_to_u64 = uint64_t(su16);
  239. const int16_t u64_to_i16 = int16_t(su64);
  240. const uint16_t u64_to_u16 = uint16_t(su64);
  241. // int16 <-> uint16
  242. const uint16_t i16_to_u16 = uint16_t(si16);
  243. const int16_t u16_to_i16 = int16_t(su16);
  244. void main()
  245. {
  246. literal();
  247. operators();
  248. typeCast();
  249. builtinFuncs();
  250. }