opcode.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849
  1. // Copyright (c) 2015-2022 The Khronos Group Inc.
  2. // Modifications Copyright (C) 2020-2024 Advanced Micro Devices, Inc. All
  3. // rights reserved.
  4. //
  5. // Licensed under the Apache License, Version 2.0 (the "License");
  6. // you may not use this file except in compliance with the License.
  7. // You may obtain a copy of the License at
  8. //
  9. // http://www.apache.org/licenses/LICENSE-2.0
  10. //
  11. // Unless required by applicable law or agreed to in writing, software
  12. // distributed under the License is distributed on an "AS IS" BASIS,
  13. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. // See the License for the specific language governing permissions and
  15. // limitations under the License.
  16. #include "source/opcode.h"
  17. #include <assert.h>
  18. #include <string.h>
  19. #include <algorithm>
  20. #include <cstdlib>
  21. #include "source/instruction.h"
  22. #include "source/macro.h"
  23. #include "source/spirv_constant.h"
  24. #include "source/spirv_endian.h"
  25. #include "source/spirv_target_env.h"
  26. #include "spirv-tools/libspirv.h"
  27. namespace {
  28. struct OpcodeDescPtrLen {
  29. const spv_opcode_desc_t* ptr;
  30. uint32_t len;
  31. };
  32. #include "core.insts-unified1.inc"
  33. static const spv_opcode_table_t kOpcodeTable = {ARRAY_SIZE(kOpcodeTableEntries),
  34. kOpcodeTableEntries};
  35. // Represents a vendor tool entry in the SPIR-V XML Registry.
  36. struct VendorTool {
  37. uint32_t value;
  38. const char* vendor;
  39. const char* tool; // Might be empty string.
  40. const char* vendor_tool; // Combination of vendor and tool.
  41. };
  42. const VendorTool vendor_tools[] = {
  43. #include "generators.inc"
  44. };
  45. } // anonymous namespace
  46. // TODO(dneto): Move this to another file. It doesn't belong with opcode
  47. // processing.
  48. const char* spvGeneratorStr(uint32_t generator) {
  49. auto where = std::find_if(
  50. std::begin(vendor_tools), std::end(vendor_tools),
  51. [generator](const VendorTool& vt) { return generator == vt.value; });
  52. if (where != std::end(vendor_tools)) return where->vendor_tool;
  53. return "Unknown";
  54. }
  55. uint32_t spvOpcodeMake(uint16_t wordCount, spv::Op opcode) {
  56. return ((uint32_t)opcode) | (((uint32_t)wordCount) << 16);
  57. }
  58. void spvOpcodeSplit(const uint32_t word, uint16_t* pWordCount,
  59. uint16_t* pOpcode) {
  60. if (pWordCount) {
  61. *pWordCount = (uint16_t)((0xffff0000 & word) >> 16);
  62. }
  63. if (pOpcode) {
  64. *pOpcode = 0x0000ffff & word;
  65. }
  66. }
  67. spv_result_t spvOpcodeTableGet(spv_opcode_table* pInstTable, spv_target_env) {
  68. if (!pInstTable) return SPV_ERROR_INVALID_POINTER;
  69. // Descriptions of each opcode. Each entry describes the format of the
  70. // instruction that follows a particular opcode.
  71. *pInstTable = &kOpcodeTable;
  72. return SPV_SUCCESS;
  73. }
  74. spv_result_t spvOpcodeTableNameLookup(spv_target_env env,
  75. const spv_opcode_table table,
  76. const char* name,
  77. spv_opcode_desc* pEntry) {
  78. if (!name || !pEntry) return SPV_ERROR_INVALID_POINTER;
  79. if (!table) return SPV_ERROR_INVALID_TABLE;
  80. // TODO: This lookup of the Opcode table is suboptimal! Binary sort would be
  81. // preferable but the table requires sorting on the Opcode name, but it's
  82. // static const initialized and matches the order of the spec.
  83. const size_t nameLength = strlen(name);
  84. const auto version = spvVersionForTargetEnv(env);
  85. for (uint64_t opcodeIndex = 0; opcodeIndex < table->count; ++opcodeIndex) {
  86. const spv_opcode_desc_t& entry = table->entries[opcodeIndex];
  87. // We consider the current opcode as available as long as
  88. // 1. The target environment satisfies the minimal requirement of the
  89. // opcode; or
  90. // 2. There is at least one extension enabling this opcode.
  91. //
  92. // Note that the second rule assumes the extension enabling this instruction
  93. // is indeed requested in the SPIR-V code; checking that should be
  94. // validator's work.
  95. if ((version >= entry.minVersion && version <= entry.lastVersion) ||
  96. entry.numExtensions > 0u || entry.numCapabilities > 0u) {
  97. // Exact match case.
  98. if (nameLength == strlen(entry.name) &&
  99. !strncmp(name, entry.name, nameLength)) {
  100. *pEntry = &entry;
  101. return SPV_SUCCESS;
  102. }
  103. // Lack of binary search really hurts here. There isn't an easy filter to
  104. // apply before checking aliases since we need to handle promotion from
  105. // vendor to KHR/EXT and KHR/EXT to core. It would require a sure-fire way
  106. // of dropping suffices. Fortunately, most lookup are based on token
  107. // value.
  108. //
  109. // If this was a binary search we could iterate between the lower and
  110. // upper bounds.
  111. if (entry.numAliases > 0) {
  112. for (uint32_t aliasIndex = 0; aliasIndex < entry.numAliases;
  113. aliasIndex++) {
  114. // Skip Op prefix. Should this be encoded in the table instead?
  115. const auto alias = entry.aliases[aliasIndex] + 2;
  116. const size_t aliasLength = strlen(alias);
  117. if (nameLength == aliasLength && !strncmp(name, alias, nameLength)) {
  118. *pEntry = &entry;
  119. return SPV_SUCCESS;
  120. }
  121. }
  122. }
  123. }
  124. }
  125. return SPV_ERROR_INVALID_LOOKUP;
  126. }
  127. spv_result_t spvOpcodeTableValueLookup(spv_target_env env,
  128. const spv_opcode_table table,
  129. const spv::Op opcode,
  130. spv_opcode_desc* pEntry) {
  131. if (!table) return SPV_ERROR_INVALID_TABLE;
  132. if (!pEntry) return SPV_ERROR_INVALID_POINTER;
  133. const auto beg = table->entries;
  134. const auto end = table->entries + table->count;
  135. spv_opcode_desc_t needle = {"", opcode, 0, nullptr, 0, {}, 0,
  136. {}, false, false, 0, nullptr, ~0u, ~0u};
  137. auto comp = [](const spv_opcode_desc_t& lhs, const spv_opcode_desc_t& rhs) {
  138. return lhs.opcode < rhs.opcode;
  139. };
  140. // We need to loop here because there can exist multiple symbols for the same
  141. // opcode value, and they can be introduced in different target environments,
  142. // which means they can have different minimal version requirements.
  143. // Assumes the underlying table is already sorted ascendingly according to
  144. // opcode value.
  145. const auto version = spvVersionForTargetEnv(env);
  146. for (auto it = std::lower_bound(beg, end, needle, comp);
  147. it != end && it->opcode == opcode; ++it) {
  148. // We considers the current opcode as available as long as
  149. // 1. The target environment satisfies the minimal requirement of the
  150. // opcode; or
  151. // 2. There is at least one extension enabling this opcode.
  152. //
  153. // Note that the second rule assumes the extension enabling this instruction
  154. // is indeed requested in the SPIR-V code; checking that should be
  155. // validator's work.
  156. if ((version >= it->minVersion && version <= it->lastVersion) ||
  157. it->numExtensions > 0u || it->numCapabilities > 0u) {
  158. *pEntry = it;
  159. return SPV_SUCCESS;
  160. }
  161. }
  162. return SPV_ERROR_INVALID_LOOKUP;
  163. }
  164. void spvInstructionCopy(const uint32_t* words, const spv::Op opcode,
  165. const uint16_t wordCount, const spv_endianness_t endian,
  166. spv_instruction_t* pInst) {
  167. pInst->opcode = opcode;
  168. pInst->words.resize(wordCount);
  169. for (uint16_t wordIndex = 0; wordIndex < wordCount; ++wordIndex) {
  170. pInst->words[wordIndex] = spvFixWord(words[wordIndex], endian);
  171. if (!wordIndex) {
  172. uint16_t thisWordCount;
  173. uint16_t thisOpcode;
  174. spvOpcodeSplit(pInst->words[wordIndex], &thisWordCount, &thisOpcode);
  175. assert(opcode == static_cast<spv::Op>(thisOpcode) &&
  176. wordCount == thisWordCount && "Endianness failed!");
  177. }
  178. }
  179. }
  180. const char* spvOpcodeString(const uint32_t opcode) {
  181. const auto beg = kOpcodeTableEntries;
  182. const auto end = kOpcodeTableEntries + ARRAY_SIZE(kOpcodeTableEntries);
  183. spv_opcode_desc_t needle = {"", static_cast<spv::Op>(opcode),
  184. 0, nullptr,
  185. 0, {},
  186. 0, {},
  187. false, false,
  188. 0, nullptr,
  189. ~0u, ~0u};
  190. auto comp = [](const spv_opcode_desc_t& lhs, const spv_opcode_desc_t& rhs) {
  191. return lhs.opcode < rhs.opcode;
  192. };
  193. auto it = std::lower_bound(beg, end, needle, comp);
  194. if (it != end && it->opcode == spv::Op(opcode)) {
  195. return it->name;
  196. }
  197. assert(0 && "Unreachable!");
  198. return "unknown";
  199. }
  200. const char* spvOpcodeString(const spv::Op opcode) {
  201. return spvOpcodeString(static_cast<uint32_t>(opcode));
  202. }
  203. int32_t spvOpcodeIsScalarType(const spv::Op opcode) {
  204. switch (opcode) {
  205. case spv::Op::OpTypeInt:
  206. case spv::Op::OpTypeFloat:
  207. case spv::Op::OpTypeBool:
  208. return true;
  209. default:
  210. return false;
  211. }
  212. }
  213. int32_t spvOpcodeIsSpecConstant(const spv::Op opcode) {
  214. switch (opcode) {
  215. case spv::Op::OpSpecConstantTrue:
  216. case spv::Op::OpSpecConstantFalse:
  217. case spv::Op::OpSpecConstant:
  218. case spv::Op::OpSpecConstantComposite:
  219. case spv::Op::OpSpecConstantCompositeReplicateEXT:
  220. case spv::Op::OpSpecConstantOp:
  221. return true;
  222. default:
  223. return false;
  224. }
  225. }
  226. int32_t spvOpcodeIsConstant(const spv::Op opcode) {
  227. switch (opcode) {
  228. case spv::Op::OpConstantTrue:
  229. case spv::Op::OpConstantFalse:
  230. case spv::Op::OpConstant:
  231. case spv::Op::OpConstantComposite:
  232. case spv::Op::OpConstantCompositeReplicateEXT:
  233. case spv::Op::OpConstantSampler:
  234. case spv::Op::OpConstantNull:
  235. case spv::Op::OpConstantFunctionPointerINTEL:
  236. case spv::Op::OpConstantStringAMDX:
  237. case spv::Op::OpSpecConstantTrue:
  238. case spv::Op::OpSpecConstantFalse:
  239. case spv::Op::OpSpecConstant:
  240. case spv::Op::OpSpecConstantComposite:
  241. case spv::Op::OpSpecConstantCompositeReplicateEXT:
  242. case spv::Op::OpSpecConstantOp:
  243. case spv::Op::OpSpecConstantStringAMDX:
  244. return true;
  245. default:
  246. return false;
  247. }
  248. }
  249. bool spvOpcodeIsConstantOrUndef(const spv::Op opcode) {
  250. return opcode == spv::Op::OpUndef || spvOpcodeIsConstant(opcode);
  251. }
  252. bool spvOpcodeIsScalarSpecConstant(const spv::Op opcode) {
  253. switch (opcode) {
  254. case spv::Op::OpSpecConstantTrue:
  255. case spv::Op::OpSpecConstantFalse:
  256. case spv::Op::OpSpecConstant:
  257. return true;
  258. default:
  259. return false;
  260. }
  261. }
  262. int32_t spvOpcodeIsComposite(const spv::Op opcode) {
  263. switch (opcode) {
  264. case spv::Op::OpTypeVector:
  265. case spv::Op::OpTypeMatrix:
  266. case spv::Op::OpTypeArray:
  267. case spv::Op::OpTypeStruct:
  268. case spv::Op::OpTypeRuntimeArray:
  269. case spv::Op::OpTypeCooperativeMatrixNV:
  270. case spv::Op::OpTypeCooperativeMatrixKHR:
  271. case spv::Op::OpTypeCooperativeVectorNV:
  272. return true;
  273. default:
  274. return false;
  275. }
  276. }
  277. bool spvOpcodeReturnsLogicalVariablePointer(const spv::Op opcode) {
  278. switch (opcode) {
  279. case spv::Op::OpVariable:
  280. case spv::Op::OpUntypedVariableKHR:
  281. case spv::Op::OpAccessChain:
  282. case spv::Op::OpInBoundsAccessChain:
  283. case spv::Op::OpUntypedAccessChainKHR:
  284. case spv::Op::OpUntypedInBoundsAccessChainKHR:
  285. case spv::Op::OpFunctionParameter:
  286. case spv::Op::OpImageTexelPointer:
  287. case spv::Op::OpCopyObject:
  288. case spv::Op::OpAllocateNodePayloadsAMDX:
  289. case spv::Op::OpSelect:
  290. case spv::Op::OpPhi:
  291. case spv::Op::OpFunctionCall:
  292. case spv::Op::OpPtrAccessChain:
  293. case spv::Op::OpUntypedPtrAccessChainKHR:
  294. case spv::Op::OpLoad:
  295. case spv::Op::OpConstantNull:
  296. case spv::Op::OpRawAccessChainNV:
  297. return true;
  298. default:
  299. return false;
  300. }
  301. }
  302. int32_t spvOpcodeReturnsLogicalPointer(const spv::Op opcode) {
  303. switch (opcode) {
  304. case spv::Op::OpVariable:
  305. case spv::Op::OpUntypedVariableKHR:
  306. case spv::Op::OpAccessChain:
  307. case spv::Op::OpInBoundsAccessChain:
  308. case spv::Op::OpUntypedAccessChainKHR:
  309. case spv::Op::OpUntypedInBoundsAccessChainKHR:
  310. case spv::Op::OpFunctionParameter:
  311. case spv::Op::OpImageTexelPointer:
  312. case spv::Op::OpCopyObject:
  313. case spv::Op::OpRawAccessChainNV:
  314. case spv::Op::OpAllocateNodePayloadsAMDX:
  315. return true;
  316. default:
  317. return false;
  318. }
  319. }
  320. int32_t spvOpcodeGeneratesType(spv::Op op) {
  321. switch (op) {
  322. case spv::Op::OpTypeVoid:
  323. case spv::Op::OpTypeBool:
  324. case spv::Op::OpTypeInt:
  325. case spv::Op::OpTypeFloat:
  326. case spv::Op::OpTypeVector:
  327. case spv::Op::OpTypeMatrix:
  328. case spv::Op::OpTypeImage:
  329. case spv::Op::OpTypeSampler:
  330. case spv::Op::OpTypeSampledImage:
  331. case spv::Op::OpTypeArray:
  332. case spv::Op::OpTypeRuntimeArray:
  333. case spv::Op::OpTypeStruct:
  334. case spv::Op::OpTypeOpaque:
  335. case spv::Op::OpTypePointer:
  336. case spv::Op::OpTypeFunction:
  337. case spv::Op::OpTypeEvent:
  338. case spv::Op::OpTypeDeviceEvent:
  339. case spv::Op::OpTypeReserveId:
  340. case spv::Op::OpTypeQueue:
  341. case spv::Op::OpTypePipe:
  342. case spv::Op::OpTypePipeStorage:
  343. case spv::Op::OpTypeNamedBarrier:
  344. case spv::Op::OpTypeAccelerationStructureNV:
  345. case spv::Op::OpTypeCooperativeMatrixNV:
  346. case spv::Op::OpTypeCooperativeMatrixKHR:
  347. case spv::Op::OpTypeCooperativeVectorNV:
  348. // case spv::Op::OpTypeAccelerationStructureKHR: covered by
  349. // spv::Op::OpTypeAccelerationStructureNV
  350. case spv::Op::OpTypeRayQueryKHR:
  351. case spv::Op::OpTypeHitObjectNV:
  352. case spv::Op::OpTypeUntypedPointerKHR:
  353. case spv::Op::OpTypeNodePayloadArrayAMDX:
  354. case spv::Op::OpTypeTensorLayoutNV:
  355. case spv::Op::OpTypeTensorViewNV:
  356. return true;
  357. default:
  358. // In particular, OpTypeForwardPointer does not generate a type,
  359. // but declares a storage class for a pointer type generated
  360. // by a different instruction.
  361. break;
  362. }
  363. return 0;
  364. }
  365. bool spvOpcodeIsDecoration(const spv::Op opcode) {
  366. switch (opcode) {
  367. case spv::Op::OpDecorate:
  368. case spv::Op::OpDecorateId:
  369. case spv::Op::OpMemberDecorate:
  370. case spv::Op::OpGroupDecorate:
  371. case spv::Op::OpGroupMemberDecorate:
  372. case spv::Op::OpDecorateStringGOOGLE:
  373. case spv::Op::OpMemberDecorateStringGOOGLE:
  374. return true;
  375. default:
  376. break;
  377. }
  378. return false;
  379. }
  380. bool spvOpcodeIsLoad(const spv::Op opcode) {
  381. switch (opcode) {
  382. case spv::Op::OpLoad:
  383. case spv::Op::OpImageSampleExplicitLod:
  384. case spv::Op::OpImageSampleImplicitLod:
  385. case spv::Op::OpImageSampleDrefImplicitLod:
  386. case spv::Op::OpImageSampleDrefExplicitLod:
  387. case spv::Op::OpImageSampleProjImplicitLod:
  388. case spv::Op::OpImageSampleProjExplicitLod:
  389. case spv::Op::OpImageSampleProjDrefImplicitLod:
  390. case spv::Op::OpImageSampleProjDrefExplicitLod:
  391. case spv::Op::OpImageSampleFootprintNV:
  392. case spv::Op::OpImageFetch:
  393. case spv::Op::OpImageGather:
  394. case spv::Op::OpImageDrefGather:
  395. case spv::Op::OpImageRead:
  396. case spv::Op::OpImageSparseSampleImplicitLod:
  397. case spv::Op::OpImageSparseSampleExplicitLod:
  398. case spv::Op::OpImageSparseSampleDrefExplicitLod:
  399. case spv::Op::OpImageSparseSampleDrefImplicitLod:
  400. case spv::Op::OpImageSparseFetch:
  401. case spv::Op::OpImageSparseGather:
  402. case spv::Op::OpImageSparseDrefGather:
  403. case spv::Op::OpImageSparseRead:
  404. return true;
  405. default:
  406. return false;
  407. }
  408. }
  409. bool spvOpcodeIsBranch(spv::Op opcode) {
  410. switch (opcode) {
  411. case spv::Op::OpBranch:
  412. case spv::Op::OpBranchConditional:
  413. case spv::Op::OpSwitch:
  414. return true;
  415. default:
  416. return false;
  417. }
  418. }
  419. bool spvOpcodeIsAtomicWithLoad(const spv::Op opcode) {
  420. switch (opcode) {
  421. case spv::Op::OpAtomicLoad:
  422. case spv::Op::OpAtomicExchange:
  423. case spv::Op::OpAtomicCompareExchange:
  424. case spv::Op::OpAtomicCompareExchangeWeak:
  425. case spv::Op::OpAtomicIIncrement:
  426. case spv::Op::OpAtomicIDecrement:
  427. case spv::Op::OpAtomicIAdd:
  428. case spv::Op::OpAtomicFAddEXT:
  429. case spv::Op::OpAtomicISub:
  430. case spv::Op::OpAtomicSMin:
  431. case spv::Op::OpAtomicUMin:
  432. case spv::Op::OpAtomicFMinEXT:
  433. case spv::Op::OpAtomicSMax:
  434. case spv::Op::OpAtomicUMax:
  435. case spv::Op::OpAtomicFMaxEXT:
  436. case spv::Op::OpAtomicAnd:
  437. case spv::Op::OpAtomicOr:
  438. case spv::Op::OpAtomicXor:
  439. case spv::Op::OpAtomicFlagTestAndSet:
  440. return true;
  441. default:
  442. return false;
  443. }
  444. }
  445. bool spvOpcodeIsAtomicOp(const spv::Op opcode) {
  446. return (spvOpcodeIsAtomicWithLoad(opcode) ||
  447. opcode == spv::Op::OpAtomicStore ||
  448. opcode == spv::Op::OpAtomicFlagClear);
  449. }
  450. bool spvOpcodeIsReturn(spv::Op opcode) {
  451. switch (opcode) {
  452. case spv::Op::OpReturn:
  453. case spv::Op::OpReturnValue:
  454. return true;
  455. default:
  456. return false;
  457. }
  458. }
  459. bool spvOpcodeIsAbort(spv::Op opcode) {
  460. switch (opcode) {
  461. case spv::Op::OpKill:
  462. case spv::Op::OpUnreachable:
  463. case spv::Op::OpTerminateInvocation:
  464. case spv::Op::OpTerminateRayKHR:
  465. case spv::Op::OpIgnoreIntersectionKHR:
  466. case spv::Op::OpEmitMeshTasksEXT:
  467. return true;
  468. default:
  469. return false;
  470. }
  471. }
  472. bool spvOpcodeIsReturnOrAbort(spv::Op opcode) {
  473. return spvOpcodeIsReturn(opcode) || spvOpcodeIsAbort(opcode);
  474. }
  475. bool spvOpcodeIsBlockTerminator(spv::Op opcode) {
  476. return spvOpcodeIsBranch(opcode) || spvOpcodeIsReturnOrAbort(opcode);
  477. }
  478. bool spvOpcodeIsBaseOpaqueType(spv::Op opcode) {
  479. switch (opcode) {
  480. case spv::Op::OpTypeImage:
  481. case spv::Op::OpTypeSampler:
  482. case spv::Op::OpTypeSampledImage:
  483. case spv::Op::OpTypeOpaque:
  484. case spv::Op::OpTypeEvent:
  485. case spv::Op::OpTypeDeviceEvent:
  486. case spv::Op::OpTypeReserveId:
  487. case spv::Op::OpTypeQueue:
  488. case spv::Op::OpTypePipe:
  489. case spv::Op::OpTypeForwardPointer:
  490. case spv::Op::OpTypePipeStorage:
  491. case spv::Op::OpTypeNamedBarrier:
  492. return true;
  493. default:
  494. return false;
  495. }
  496. }
  497. bool spvOpcodeIsNonUniformGroupOperation(spv::Op opcode) {
  498. switch (opcode) {
  499. case spv::Op::OpGroupNonUniformElect:
  500. case spv::Op::OpGroupNonUniformAll:
  501. case spv::Op::OpGroupNonUniformAny:
  502. case spv::Op::OpGroupNonUniformAllEqual:
  503. case spv::Op::OpGroupNonUniformBroadcast:
  504. case spv::Op::OpGroupNonUniformBroadcastFirst:
  505. case spv::Op::OpGroupNonUniformBallot:
  506. case spv::Op::OpGroupNonUniformInverseBallot:
  507. case spv::Op::OpGroupNonUniformBallotBitExtract:
  508. case spv::Op::OpGroupNonUniformBallotBitCount:
  509. case spv::Op::OpGroupNonUniformBallotFindLSB:
  510. case spv::Op::OpGroupNonUniformBallotFindMSB:
  511. case spv::Op::OpGroupNonUniformShuffle:
  512. case spv::Op::OpGroupNonUniformShuffleXor:
  513. case spv::Op::OpGroupNonUniformShuffleUp:
  514. case spv::Op::OpGroupNonUniformShuffleDown:
  515. case spv::Op::OpGroupNonUniformIAdd:
  516. case spv::Op::OpGroupNonUniformFAdd:
  517. case spv::Op::OpGroupNonUniformIMul:
  518. case spv::Op::OpGroupNonUniformFMul:
  519. case spv::Op::OpGroupNonUniformSMin:
  520. case spv::Op::OpGroupNonUniformUMin:
  521. case spv::Op::OpGroupNonUniformFMin:
  522. case spv::Op::OpGroupNonUniformSMax:
  523. case spv::Op::OpGroupNonUniformUMax:
  524. case spv::Op::OpGroupNonUniformFMax:
  525. case spv::Op::OpGroupNonUniformBitwiseAnd:
  526. case spv::Op::OpGroupNonUniformBitwiseOr:
  527. case spv::Op::OpGroupNonUniformBitwiseXor:
  528. case spv::Op::OpGroupNonUniformLogicalAnd:
  529. case spv::Op::OpGroupNonUniformLogicalOr:
  530. case spv::Op::OpGroupNonUniformLogicalXor:
  531. case spv::Op::OpGroupNonUniformQuadBroadcast:
  532. case spv::Op::OpGroupNonUniformQuadSwap:
  533. case spv::Op::OpGroupNonUniformRotateKHR:
  534. case spv::Op::OpGroupNonUniformQuadAllKHR:
  535. case spv::Op::OpGroupNonUniformQuadAnyKHR:
  536. return true;
  537. default:
  538. return false;
  539. }
  540. }
  541. bool spvOpcodeIsScalarizable(spv::Op opcode) {
  542. switch (opcode) {
  543. case spv::Op::OpPhi:
  544. case spv::Op::OpCopyObject:
  545. case spv::Op::OpConvertFToU:
  546. case spv::Op::OpConvertFToS:
  547. case spv::Op::OpConvertSToF:
  548. case spv::Op::OpConvertUToF:
  549. case spv::Op::OpUConvert:
  550. case spv::Op::OpSConvert:
  551. case spv::Op::OpFConvert:
  552. case spv::Op::OpQuantizeToF16:
  553. case spv::Op::OpVectorInsertDynamic:
  554. case spv::Op::OpSNegate:
  555. case spv::Op::OpFNegate:
  556. case spv::Op::OpIAdd:
  557. case spv::Op::OpFAdd:
  558. case spv::Op::OpISub:
  559. case spv::Op::OpFSub:
  560. case spv::Op::OpIMul:
  561. case spv::Op::OpFMul:
  562. case spv::Op::OpUDiv:
  563. case spv::Op::OpSDiv:
  564. case spv::Op::OpFDiv:
  565. case spv::Op::OpUMod:
  566. case spv::Op::OpSRem:
  567. case spv::Op::OpSMod:
  568. case spv::Op::OpFRem:
  569. case spv::Op::OpFMod:
  570. case spv::Op::OpVectorTimesScalar:
  571. case spv::Op::OpIAddCarry:
  572. case spv::Op::OpISubBorrow:
  573. case spv::Op::OpUMulExtended:
  574. case spv::Op::OpSMulExtended:
  575. case spv::Op::OpShiftRightLogical:
  576. case spv::Op::OpShiftRightArithmetic:
  577. case spv::Op::OpShiftLeftLogical:
  578. case spv::Op::OpBitwiseOr:
  579. case spv::Op::OpBitwiseAnd:
  580. case spv::Op::OpNot:
  581. case spv::Op::OpBitFieldInsert:
  582. case spv::Op::OpBitFieldSExtract:
  583. case spv::Op::OpBitFieldUExtract:
  584. case spv::Op::OpBitReverse:
  585. case spv::Op::OpBitCount:
  586. case spv::Op::OpIsNan:
  587. case spv::Op::OpIsInf:
  588. case spv::Op::OpIsFinite:
  589. case spv::Op::OpIsNormal:
  590. case spv::Op::OpSignBitSet:
  591. case spv::Op::OpLessOrGreater:
  592. case spv::Op::OpOrdered:
  593. case spv::Op::OpUnordered:
  594. case spv::Op::OpLogicalEqual:
  595. case spv::Op::OpLogicalNotEqual:
  596. case spv::Op::OpLogicalOr:
  597. case spv::Op::OpLogicalAnd:
  598. case spv::Op::OpLogicalNot:
  599. case spv::Op::OpSelect:
  600. case spv::Op::OpIEqual:
  601. case spv::Op::OpINotEqual:
  602. case spv::Op::OpUGreaterThan:
  603. case spv::Op::OpSGreaterThan:
  604. case spv::Op::OpUGreaterThanEqual:
  605. case spv::Op::OpSGreaterThanEqual:
  606. case spv::Op::OpULessThan:
  607. case spv::Op::OpSLessThan:
  608. case spv::Op::OpULessThanEqual:
  609. case spv::Op::OpSLessThanEqual:
  610. case spv::Op::OpFOrdEqual:
  611. case spv::Op::OpFUnordEqual:
  612. case spv::Op::OpFOrdNotEqual:
  613. case spv::Op::OpFUnordNotEqual:
  614. case spv::Op::OpFOrdLessThan:
  615. case spv::Op::OpFUnordLessThan:
  616. case spv::Op::OpFOrdGreaterThan:
  617. case spv::Op::OpFUnordGreaterThan:
  618. case spv::Op::OpFOrdLessThanEqual:
  619. case spv::Op::OpFUnordLessThanEqual:
  620. case spv::Op::OpFOrdGreaterThanEqual:
  621. case spv::Op::OpFUnordGreaterThanEqual:
  622. return true;
  623. default:
  624. return false;
  625. }
  626. }
  627. bool spvOpcodeIsDebug(spv::Op opcode) {
  628. switch (opcode) {
  629. case spv::Op::OpName:
  630. case spv::Op::OpMemberName:
  631. case spv::Op::OpSource:
  632. case spv::Op::OpSourceContinued:
  633. case spv::Op::OpSourceExtension:
  634. case spv::Op::OpString:
  635. case spv::Op::OpLine:
  636. case spv::Op::OpNoLine:
  637. case spv::Op::OpModuleProcessed:
  638. return true;
  639. default:
  640. return false;
  641. }
  642. }
  643. bool spvOpcodeIsCommutativeBinaryOperator(spv::Op opcode) {
  644. switch (opcode) {
  645. case spv::Op::OpPtrEqual:
  646. case spv::Op::OpPtrNotEqual:
  647. case spv::Op::OpIAdd:
  648. case spv::Op::OpFAdd:
  649. case spv::Op::OpIMul:
  650. case spv::Op::OpFMul:
  651. case spv::Op::OpDot:
  652. case spv::Op::OpIAddCarry:
  653. case spv::Op::OpUMulExtended:
  654. case spv::Op::OpSMulExtended:
  655. case spv::Op::OpBitwiseOr:
  656. case spv::Op::OpBitwiseXor:
  657. case spv::Op::OpBitwiseAnd:
  658. case spv::Op::OpOrdered:
  659. case spv::Op::OpUnordered:
  660. case spv::Op::OpLogicalEqual:
  661. case spv::Op::OpLogicalNotEqual:
  662. case spv::Op::OpLogicalOr:
  663. case spv::Op::OpLogicalAnd:
  664. case spv::Op::OpIEqual:
  665. case spv::Op::OpINotEqual:
  666. case spv::Op::OpFOrdEqual:
  667. case spv::Op::OpFUnordEqual:
  668. case spv::Op::OpFOrdNotEqual:
  669. case spv::Op::OpFUnordNotEqual:
  670. return true;
  671. default:
  672. return false;
  673. }
  674. }
  675. bool spvOpcodeIsLinearAlgebra(spv::Op opcode) {
  676. switch (opcode) {
  677. case spv::Op::OpTranspose:
  678. case spv::Op::OpVectorTimesScalar:
  679. case spv::Op::OpMatrixTimesScalar:
  680. case spv::Op::OpVectorTimesMatrix:
  681. case spv::Op::OpMatrixTimesVector:
  682. case spv::Op::OpMatrixTimesMatrix:
  683. case spv::Op::OpOuterProduct:
  684. case spv::Op::OpDot:
  685. return true;
  686. default:
  687. return false;
  688. }
  689. }
  690. bool spvOpcodeIsImageSample(const spv::Op opcode) {
  691. switch (opcode) {
  692. case spv::Op::OpImageSampleImplicitLod:
  693. case spv::Op::OpImageSampleExplicitLod:
  694. case spv::Op::OpImageSampleDrefImplicitLod:
  695. case spv::Op::OpImageSampleDrefExplicitLod:
  696. case spv::Op::OpImageSampleProjImplicitLod:
  697. case spv::Op::OpImageSampleProjExplicitLod:
  698. case spv::Op::OpImageSampleProjDrefImplicitLod:
  699. case spv::Op::OpImageSampleProjDrefExplicitLod:
  700. case spv::Op::OpImageSparseSampleImplicitLod:
  701. case spv::Op::OpImageSparseSampleExplicitLod:
  702. case spv::Op::OpImageSparseSampleDrefImplicitLod:
  703. case spv::Op::OpImageSparseSampleDrefExplicitLod:
  704. case spv::Op::OpImageSampleFootprintNV:
  705. return true;
  706. default:
  707. return false;
  708. }
  709. }
  710. bool spvIsExtendedInstruction(const spv::Op opcode) {
  711. switch (opcode) {
  712. case spv::Op::OpExtInst:
  713. case spv::Op::OpExtInstWithForwardRefsKHR:
  714. return true;
  715. default:
  716. return false;
  717. }
  718. }
  719. std::vector<uint32_t> spvOpcodeMemorySemanticsOperandIndices(spv::Op opcode) {
  720. switch (opcode) {
  721. case spv::Op::OpMemoryBarrier:
  722. return {1};
  723. case spv::Op::OpAtomicStore:
  724. case spv::Op::OpControlBarrier:
  725. case spv::Op::OpAtomicFlagClear:
  726. case spv::Op::OpMemoryNamedBarrier:
  727. return {2};
  728. case spv::Op::OpAtomicLoad:
  729. case spv::Op::OpAtomicExchange:
  730. case spv::Op::OpAtomicIIncrement:
  731. case spv::Op::OpAtomicIDecrement:
  732. case spv::Op::OpAtomicIAdd:
  733. case spv::Op::OpAtomicFAddEXT:
  734. case spv::Op::OpAtomicISub:
  735. case spv::Op::OpAtomicSMin:
  736. case spv::Op::OpAtomicUMin:
  737. case spv::Op::OpAtomicSMax:
  738. case spv::Op::OpAtomicUMax:
  739. case spv::Op::OpAtomicAnd:
  740. case spv::Op::OpAtomicOr:
  741. case spv::Op::OpAtomicXor:
  742. case spv::Op::OpAtomicFlagTestAndSet:
  743. return {4};
  744. case spv::Op::OpAtomicCompareExchange:
  745. case spv::Op::OpAtomicCompareExchangeWeak:
  746. return {4, 5};
  747. default:
  748. return {};
  749. }
  750. }
  751. bool spvOpcodeIsAccessChain(spv::Op opcode) {
  752. switch (opcode) {
  753. case spv::Op::OpAccessChain:
  754. case spv::Op::OpInBoundsAccessChain:
  755. case spv::Op::OpPtrAccessChain:
  756. case spv::Op::OpInBoundsPtrAccessChain:
  757. case spv::Op::OpRawAccessChainNV:
  758. return true;
  759. default:
  760. return false;
  761. }
  762. }
  763. bool spvOpcodeIsBit(spv::Op opcode) {
  764. switch (opcode) {
  765. case spv::Op::OpShiftRightLogical:
  766. case spv::Op::OpShiftRightArithmetic:
  767. case spv::Op::OpShiftLeftLogical:
  768. case spv::Op::OpBitwiseOr:
  769. case spv::Op::OpBitwiseXor:
  770. case spv::Op::OpBitwiseAnd:
  771. case spv::Op::OpNot:
  772. case spv::Op::OpBitReverse:
  773. case spv::Op::OpBitCount:
  774. return true;
  775. default:
  776. return false;
  777. }
  778. }
  779. bool spvOpcodeGeneratesUntypedPointer(spv::Op opcode) {
  780. switch (opcode) {
  781. case spv::Op::OpUntypedVariableKHR:
  782. case spv::Op::OpUntypedAccessChainKHR:
  783. case spv::Op::OpUntypedInBoundsAccessChainKHR:
  784. case spv::Op::OpUntypedPtrAccessChainKHR:
  785. case spv::Op::OpUntypedInBoundsPtrAccessChainKHR:
  786. return true;
  787. default:
  788. return false;
  789. }
  790. }