ir_emitter.cpp 69 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058
  1. // SPDX-FileCopyrightText: Copyright 2021 yuzu Emulator Project
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  3. #include "common/bit_cast.h"
  4. #include "shader_recompiler/frontend/ir/ir_emitter.h"
  5. #include "shader_recompiler/frontend/ir/value.h"
  6. namespace Shader::IR {
  7. namespace {
  8. [[noreturn]] void ThrowInvalidType(Type type) {
  9. throw InvalidArgument("Invalid type {}", type);
  10. }
  11. Value MakeLodClampPair(IREmitter& ir, const F32& bias_lod, const F32& lod_clamp) {
  12. if (!bias_lod.IsEmpty() && !lod_clamp.IsEmpty()) {
  13. return ir.CompositeConstruct(bias_lod, lod_clamp);
  14. } else if (!bias_lod.IsEmpty()) {
  15. return bias_lod;
  16. } else if (!lod_clamp.IsEmpty()) {
  17. return lod_clamp;
  18. } else {
  19. return Value{};
  20. }
  21. }
  22. } // Anonymous namespace
  23. U1 IREmitter::Imm1(bool value) const {
  24. return U1{Value{value}};
  25. }
  26. U8 IREmitter::Imm8(u8 value) const {
  27. return U8{Value{value}};
  28. }
  29. U16 IREmitter::Imm16(u16 value) const {
  30. return U16{Value{value}};
  31. }
  32. U32 IREmitter::Imm32(u32 value) const {
  33. return U32{Value{value}};
  34. }
  35. U32 IREmitter::Imm32(s32 value) const {
  36. return U32{Value{static_cast<u32>(value)}};
  37. }
  38. F32 IREmitter::Imm32(f32 value) const {
  39. return F32{Value{value}};
  40. }
  41. U64 IREmitter::Imm64(u64 value) const {
  42. return U64{Value{value}};
  43. }
  44. U64 IREmitter::Imm64(s64 value) const {
  45. return U64{Value{static_cast<u64>(value)}};
  46. }
  47. F64 IREmitter::Imm64(f64 value) const {
  48. return F64{Value{value}};
  49. }
  50. U1 IREmitter::ConditionRef(const U1& value) {
  51. return Inst<U1>(Opcode::ConditionRef, value);
  52. }
  53. void IREmitter::Reference(const Value& value) {
  54. Inst(Opcode::Reference, value);
  55. }
  56. void IREmitter::PhiMove(IR::Inst& phi, const Value& value) {
  57. Inst(Opcode::PhiMove, Value{&phi}, value);
  58. }
  59. void IREmitter::Prologue() {
  60. Inst(Opcode::Prologue);
  61. }
  62. void IREmitter::Epilogue() {
  63. Inst(Opcode::Epilogue);
  64. }
  65. void IREmitter::DemoteToHelperInvocation() {
  66. Inst(Opcode::DemoteToHelperInvocation);
  67. }
  68. void IREmitter::EmitVertex(const U32& stream) {
  69. Inst(Opcode::EmitVertex, stream);
  70. }
  71. void IREmitter::EndPrimitive(const U32& stream) {
  72. Inst(Opcode::EndPrimitive, stream);
  73. }
  74. void IREmitter::Barrier() {
  75. Inst(Opcode::Barrier);
  76. }
  77. void IREmitter::WorkgroupMemoryBarrier() {
  78. Inst(Opcode::WorkgroupMemoryBarrier);
  79. }
  80. void IREmitter::DeviceMemoryBarrier() {
  81. Inst(Opcode::DeviceMemoryBarrier);
  82. }
  83. U32 IREmitter::GetReg(IR::Reg reg) {
  84. return Inst<U32>(Opcode::GetRegister, reg);
  85. }
  86. void IREmitter::SetReg(IR::Reg reg, const U32& value) {
  87. Inst(Opcode::SetRegister, reg, value);
  88. }
  89. U1 IREmitter::GetPred(IR::Pred pred, bool is_negated) {
  90. if (pred == Pred::PT) {
  91. return Imm1(!is_negated);
  92. }
  93. const U1 value{Inst<U1>(Opcode::GetPred, pred)};
  94. if (is_negated) {
  95. return Inst<U1>(Opcode::LogicalNot, value);
  96. } else {
  97. return value;
  98. }
  99. }
  100. void IREmitter::SetPred(IR::Pred pred, const U1& value) {
  101. if (pred != IR::Pred::PT) {
  102. Inst(Opcode::SetPred, pred, value);
  103. }
  104. }
  105. U1 IREmitter::GetGotoVariable(u32 id) {
  106. return Inst<U1>(Opcode::GetGotoVariable, id);
  107. }
  108. void IREmitter::SetGotoVariable(u32 id, const U1& value) {
  109. Inst(Opcode::SetGotoVariable, id, value);
  110. }
  111. U32 IREmitter::GetIndirectBranchVariable() {
  112. return Inst<U32>(Opcode::GetIndirectBranchVariable);
  113. }
  114. void IREmitter::SetIndirectBranchVariable(const U32& value) {
  115. Inst(Opcode::SetIndirectBranchVariable, value);
  116. }
  117. U32 IREmitter::GetCbuf(const U32& binding, const U32& byte_offset) {
  118. return Inst<U32>(Opcode::GetCbufU32, binding, byte_offset);
  119. }
  120. Value IREmitter::GetCbuf(const U32& binding, const U32& byte_offset, size_t bitsize,
  121. bool is_signed) {
  122. switch (bitsize) {
  123. case 8:
  124. return Inst<U32>(is_signed ? Opcode::GetCbufS8 : Opcode::GetCbufU8, binding, byte_offset);
  125. case 16:
  126. return Inst<U32>(is_signed ? Opcode::GetCbufS16 : Opcode::GetCbufU16, binding, byte_offset);
  127. case 32:
  128. return Inst<U32>(Opcode::GetCbufU32, binding, byte_offset);
  129. case 64:
  130. return Inst(Opcode::GetCbufU32x2, binding, byte_offset);
  131. default:
  132. throw InvalidArgument("Invalid bit size {}", bitsize);
  133. }
  134. }
  135. F32 IREmitter::GetFloatCbuf(const U32& binding, const U32& byte_offset) {
  136. return Inst<F32>(Opcode::GetCbufF32, binding, byte_offset);
  137. }
  138. U1 IREmitter::GetZFlag() {
  139. return Inst<U1>(Opcode::GetZFlag);
  140. }
  141. U1 IREmitter::GetSFlag() {
  142. return Inst<U1>(Opcode::GetSFlag);
  143. }
  144. U1 IREmitter::GetCFlag() {
  145. return Inst<U1>(Opcode::GetCFlag);
  146. }
  147. U1 IREmitter::GetOFlag() {
  148. return Inst<U1>(Opcode::GetOFlag);
  149. }
  150. void IREmitter::SetZFlag(const U1& value) {
  151. Inst(Opcode::SetZFlag, value);
  152. }
  153. void IREmitter::SetSFlag(const U1& value) {
  154. Inst(Opcode::SetSFlag, value);
  155. }
  156. void IREmitter::SetCFlag(const U1& value) {
  157. Inst(Opcode::SetCFlag, value);
  158. }
  159. void IREmitter::SetOFlag(const U1& value) {
  160. Inst(Opcode::SetOFlag, value);
  161. }
  162. static U1 GetFlowTest(IREmitter& ir, FlowTest flow_test) {
  163. switch (flow_test) {
  164. case FlowTest::F:
  165. return ir.Imm1(false);
  166. case FlowTest::LT:
  167. return ir.LogicalXor(ir.LogicalAnd(ir.GetSFlag(), ir.LogicalNot(ir.GetZFlag())),
  168. ir.GetOFlag());
  169. case FlowTest::EQ:
  170. return ir.LogicalAnd(ir.LogicalNot(ir.GetSFlag()), ir.GetZFlag());
  171. case FlowTest::LE:
  172. return ir.LogicalXor(ir.GetSFlag(), ir.LogicalOr(ir.GetZFlag(), ir.GetOFlag()));
  173. case FlowTest::GT:
  174. return ir.LogicalAnd(ir.LogicalXor(ir.LogicalNot(ir.GetSFlag()), ir.GetOFlag()),
  175. ir.LogicalNot(ir.GetZFlag()));
  176. case FlowTest::NE:
  177. return ir.LogicalNot(ir.GetZFlag());
  178. case FlowTest::GE:
  179. return ir.LogicalNot(ir.LogicalXor(ir.GetSFlag(), ir.GetOFlag()));
  180. case FlowTest::NUM:
  181. return ir.LogicalOr(ir.LogicalNot(ir.GetSFlag()), ir.LogicalNot(ir.GetZFlag()));
  182. case FlowTest::NaN:
  183. return ir.LogicalAnd(ir.GetSFlag(), ir.GetZFlag());
  184. case FlowTest::LTU:
  185. return ir.LogicalXor(ir.GetSFlag(), ir.GetOFlag());
  186. case FlowTest::EQU:
  187. return ir.GetZFlag();
  188. case FlowTest::LEU:
  189. return ir.LogicalOr(ir.LogicalXor(ir.GetSFlag(), ir.GetOFlag()), ir.GetZFlag());
  190. case FlowTest::GTU:
  191. return ir.LogicalXor(ir.LogicalNot(ir.GetSFlag()),
  192. ir.LogicalOr(ir.GetZFlag(), ir.GetOFlag()));
  193. case FlowTest::NEU:
  194. return ir.LogicalOr(ir.GetSFlag(), ir.LogicalNot(ir.GetZFlag()));
  195. case FlowTest::GEU:
  196. return ir.LogicalXor(ir.LogicalOr(ir.LogicalNot(ir.GetSFlag()), ir.GetZFlag()),
  197. ir.GetOFlag());
  198. case FlowTest::T:
  199. return ir.Imm1(true);
  200. case FlowTest::OFF:
  201. return ir.LogicalNot(ir.GetOFlag());
  202. case FlowTest::LO:
  203. return ir.LogicalNot(ir.GetCFlag());
  204. case FlowTest::SFF:
  205. return ir.LogicalNot(ir.GetSFlag());
  206. case FlowTest::LS:
  207. return ir.LogicalOr(ir.GetZFlag(), ir.LogicalNot(ir.GetCFlag()));
  208. case FlowTest::HI:
  209. return ir.LogicalAnd(ir.GetCFlag(), ir.LogicalNot(ir.GetZFlag()));
  210. case FlowTest::SFT:
  211. return ir.GetSFlag();
  212. case FlowTest::HS:
  213. return ir.GetCFlag();
  214. case FlowTest::OFT:
  215. return ir.GetOFlag();
  216. case FlowTest::RLE:
  217. return ir.LogicalOr(ir.GetSFlag(), ir.GetZFlag());
  218. case FlowTest::RGT:
  219. return ir.LogicalAnd(ir.LogicalNot(ir.GetSFlag()), ir.LogicalNot(ir.GetZFlag()));
  220. case FlowTest::FCSM_TR:
  221. LOG_WARNING(Shader, "(STUBBED) FCSM_TR");
  222. return ir.Imm1(false);
  223. case FlowTest::CSM_TA:
  224. case FlowTest::CSM_TR:
  225. case FlowTest::CSM_MX:
  226. case FlowTest::FCSM_TA:
  227. case FlowTest::FCSM_MX:
  228. default:
  229. throw NotImplementedException("Flow test {}", flow_test);
  230. }
  231. }
  232. U1 IREmitter::Condition(IR::Condition cond) {
  233. const FlowTest flow_test{cond.GetFlowTest()};
  234. const auto [pred, is_negated]{cond.GetPred()};
  235. if (flow_test == FlowTest::T) {
  236. return GetPred(pred, is_negated);
  237. }
  238. return LogicalAnd(GetPred(pred, is_negated), GetFlowTest(*this, flow_test));
  239. }
  240. U1 IREmitter::GetFlowTestResult(FlowTest test) {
  241. return GetFlowTest(*this, test);
  242. }
  243. F32 IREmitter::GetAttribute(IR::Attribute attribute) {
  244. return GetAttribute(attribute, Imm32(0));
  245. }
  246. F32 IREmitter::GetAttribute(IR::Attribute attribute, const U32& vertex) {
  247. return Inst<F32>(Opcode::GetAttribute, attribute, vertex);
  248. }
  249. U32 IREmitter::GetAttributeU32(IR::Attribute attribute) {
  250. return GetAttributeU32(attribute, Imm32(0));
  251. }
  252. U32 IREmitter::GetAttributeU32(IR::Attribute attribute, const U32& vertex) {
  253. return Inst<U32>(Opcode::GetAttributeU32, attribute, vertex);
  254. }
  255. void IREmitter::SetAttribute(IR::Attribute attribute, const F32& value, const U32& vertex) {
  256. Inst(Opcode::SetAttribute, attribute, value, vertex);
  257. }
  258. F32 IREmitter::GetAttributeIndexed(const U32& phys_address) {
  259. return GetAttributeIndexed(phys_address, Imm32(0));
  260. }
  261. F32 IREmitter::GetAttributeIndexed(const U32& phys_address, const U32& vertex) {
  262. return Inst<F32>(Opcode::GetAttributeIndexed, phys_address, vertex);
  263. }
  264. void IREmitter::SetAttributeIndexed(const U32& phys_address, const F32& value, const U32& vertex) {
  265. Inst(Opcode::SetAttributeIndexed, phys_address, value, vertex);
  266. }
  267. F32 IREmitter::GetPatch(Patch patch) {
  268. return Inst<F32>(Opcode::GetPatch, patch);
  269. }
  270. void IREmitter::SetPatch(Patch patch, const F32& value) {
  271. Inst(Opcode::SetPatch, patch, value);
  272. }
  273. void IREmitter::SetFragColor(u32 index, u32 component, const F32& value) {
  274. Inst(Opcode::SetFragColor, Imm32(index), Imm32(component), value);
  275. }
  276. void IREmitter::SetSampleMask(const U32& value) {
  277. Inst(Opcode::SetSampleMask, value);
  278. }
  279. void IREmitter::SetFragDepth(const F32& value) {
  280. Inst(Opcode::SetFragDepth, value);
  281. }
  282. U32 IREmitter::WorkgroupIdX() {
  283. return U32{CompositeExtract(Inst(Opcode::WorkgroupId), 0)};
  284. }
  285. U32 IREmitter::WorkgroupIdY() {
  286. return U32{CompositeExtract(Inst(Opcode::WorkgroupId), 1)};
  287. }
  288. U32 IREmitter::WorkgroupIdZ() {
  289. return U32{CompositeExtract(Inst(Opcode::WorkgroupId), 2)};
  290. }
  291. Value IREmitter::LocalInvocationId() {
  292. return Inst(Opcode::LocalInvocationId);
  293. }
  294. U32 IREmitter::LocalInvocationIdX() {
  295. return U32{CompositeExtract(Inst(Opcode::LocalInvocationId), 0)};
  296. }
  297. U32 IREmitter::LocalInvocationIdY() {
  298. return U32{CompositeExtract(Inst(Opcode::LocalInvocationId), 1)};
  299. }
  300. U32 IREmitter::LocalInvocationIdZ() {
  301. return U32{CompositeExtract(Inst(Opcode::LocalInvocationId), 2)};
  302. }
  303. U32 IREmitter::InvocationId() {
  304. return Inst<U32>(Opcode::InvocationId);
  305. }
  306. U32 IREmitter::InvocationInfo() {
  307. return Inst<U32>(Opcode::InvocationInfo);
  308. }
  309. U32 IREmitter::SampleId() {
  310. return Inst<U32>(Opcode::SampleId);
  311. }
  312. U1 IREmitter::IsHelperInvocation() {
  313. return Inst<U1>(Opcode::IsHelperInvocation);
  314. }
  315. F32 IREmitter::YDirection() {
  316. return Inst<F32>(Opcode::YDirection);
  317. }
  318. F32 IREmitter::ResolutionDownFactor() {
  319. return Inst<F32>(Opcode::ResolutionDownFactor);
  320. }
  321. F32 IREmitter::RenderAreaWidth() {
  322. return F32(CompositeExtract(Inst<Value>(Opcode::RenderArea), 0));
  323. }
  324. F32 IREmitter::RenderAreaHeight() {
  325. return F32(CompositeExtract(Inst<Value>(Opcode::RenderArea), 1));
  326. }
  327. U32 IREmitter::LaneId() {
  328. return Inst<U32>(Opcode::LaneId);
  329. }
  330. U32 IREmitter::LoadGlobalU8(const U64& address) {
  331. return Inst<U32>(Opcode::LoadGlobalU8, address);
  332. }
  333. U32 IREmitter::LoadGlobalS8(const U64& address) {
  334. return Inst<U32>(Opcode::LoadGlobalS8, address);
  335. }
  336. U32 IREmitter::LoadGlobalU16(const U64& address) {
  337. return Inst<U32>(Opcode::LoadGlobalU16, address);
  338. }
  339. U32 IREmitter::LoadGlobalS16(const U64& address) {
  340. return Inst<U32>(Opcode::LoadGlobalS16, address);
  341. }
  342. U32 IREmitter::LoadGlobal32(const U64& address) {
  343. return Inst<U32>(Opcode::LoadGlobal32, address);
  344. }
  345. Value IREmitter::LoadGlobal64(const U64& address) {
  346. return Inst<Value>(Opcode::LoadGlobal64, address);
  347. }
  348. Value IREmitter::LoadGlobal128(const U64& address) {
  349. return Inst<Value>(Opcode::LoadGlobal128, address);
  350. }
  351. void IREmitter::WriteGlobalU8(const U64& address, const U32& value) {
  352. Inst(Opcode::WriteGlobalU8, address, value);
  353. }
  354. void IREmitter::WriteGlobalS8(const U64& address, const U32& value) {
  355. Inst(Opcode::WriteGlobalS8, address, value);
  356. }
  357. void IREmitter::WriteGlobalU16(const U64& address, const U32& value) {
  358. Inst(Opcode::WriteGlobalU16, address, value);
  359. }
  360. void IREmitter::WriteGlobalS16(const U64& address, const U32& value) {
  361. Inst(Opcode::WriteGlobalS16, address, value);
  362. }
  363. void IREmitter::WriteGlobal32(const U64& address, const U32& value) {
  364. Inst(Opcode::WriteGlobal32, address, value);
  365. }
  366. void IREmitter::WriteGlobal64(const U64& address, const IR::Value& vector) {
  367. Inst(Opcode::WriteGlobal64, address, vector);
  368. }
  369. void IREmitter::WriteGlobal128(const U64& address, const IR::Value& vector) {
  370. Inst(Opcode::WriteGlobal128, address, vector);
  371. }
  372. U32 IREmitter::LoadLocal(const IR::U32& word_offset) {
  373. return Inst<U32>(Opcode::LoadLocal, word_offset);
  374. }
  375. void IREmitter::WriteLocal(const IR::U32& word_offset, const IR::U32& value) {
  376. Inst(Opcode::WriteLocal, word_offset, value);
  377. }
  378. Value IREmitter::LoadShared(int bit_size, bool is_signed, const IR::U32& offset) {
  379. switch (bit_size) {
  380. case 8:
  381. return Inst(is_signed ? Opcode::LoadSharedS8 : Opcode::LoadSharedU8, offset);
  382. case 16:
  383. return Inst(is_signed ? Opcode::LoadSharedS16 : Opcode::LoadSharedU16, offset);
  384. case 32:
  385. return Inst(Opcode::LoadSharedU32, offset);
  386. case 64:
  387. return Inst(Opcode::LoadSharedU64, offset);
  388. case 128:
  389. return Inst(Opcode::LoadSharedU128, offset);
  390. }
  391. throw InvalidArgument("Invalid bit size {}", bit_size);
  392. }
  393. void IREmitter::WriteShared(int bit_size, const IR::U32& offset, const IR::Value& value) {
  394. switch (bit_size) {
  395. case 8:
  396. Inst(Opcode::WriteSharedU8, offset, value);
  397. break;
  398. case 16:
  399. Inst(Opcode::WriteSharedU16, offset, value);
  400. break;
  401. case 32:
  402. Inst(Opcode::WriteSharedU32, offset, value);
  403. break;
  404. case 64:
  405. Inst(Opcode::WriteSharedU64, offset, value);
  406. break;
  407. case 128:
  408. Inst(Opcode::WriteSharedU128, offset, value);
  409. break;
  410. default:
  411. throw InvalidArgument("Invalid bit size {}", bit_size);
  412. }
  413. }
  414. U1 IREmitter::GetZeroFromOp(const Value& op) {
  415. return Inst<U1>(Opcode::GetZeroFromOp, op);
  416. }
  417. U1 IREmitter::GetSignFromOp(const Value& op) {
  418. return Inst<U1>(Opcode::GetSignFromOp, op);
  419. }
  420. U1 IREmitter::GetCarryFromOp(const Value& op) {
  421. return Inst<U1>(Opcode::GetCarryFromOp, op);
  422. }
  423. U1 IREmitter::GetOverflowFromOp(const Value& op) {
  424. return Inst<U1>(Opcode::GetOverflowFromOp, op);
  425. }
  426. U1 IREmitter::GetSparseFromOp(const Value& op) {
  427. return Inst<U1>(Opcode::GetSparseFromOp, op);
  428. }
  429. U1 IREmitter::GetInBoundsFromOp(const Value& op) {
  430. return Inst<U1>(Opcode::GetInBoundsFromOp, op);
  431. }
  432. F16F32F64 IREmitter::FPAdd(const F16F32F64& a, const F16F32F64& b, FpControl control) {
  433. if (a.Type() != b.Type()) {
  434. throw InvalidArgument("Mismatching types {} and {}", a.Type(), b.Type());
  435. }
  436. switch (a.Type()) {
  437. case Type::F16:
  438. return Inst<F16>(Opcode::FPAdd16, Flags{control}, a, b);
  439. case Type::F32:
  440. return Inst<F32>(Opcode::FPAdd32, Flags{control}, a, b);
  441. case Type::F64:
  442. return Inst<F64>(Opcode::FPAdd64, Flags{control}, a, b);
  443. default:
  444. ThrowInvalidType(a.Type());
  445. }
  446. }
  447. Value IREmitter::CompositeConstruct(const Value& e1, const Value& e2) {
  448. if (e1.Type() != e2.Type()) {
  449. throw InvalidArgument("Mismatching types {} and {}", e1.Type(), e2.Type());
  450. }
  451. switch (e1.Type()) {
  452. case Type::U32:
  453. return Inst(Opcode::CompositeConstructU32x2, e1, e2);
  454. case Type::F16:
  455. return Inst(Opcode::CompositeConstructF16x2, e1, e2);
  456. case Type::F32:
  457. return Inst(Opcode::CompositeConstructF32x2, e1, e2);
  458. case Type::F64:
  459. return Inst(Opcode::CompositeConstructF64x2, e1, e2);
  460. default:
  461. ThrowInvalidType(e1.Type());
  462. }
  463. }
  464. Value IREmitter::CompositeConstruct(const Value& e1, const Value& e2, const Value& e3) {
  465. if (e1.Type() != e2.Type() || e1.Type() != e3.Type()) {
  466. throw InvalidArgument("Mismatching types {}, {}, and {}", e1.Type(), e2.Type(), e3.Type());
  467. }
  468. switch (e1.Type()) {
  469. case Type::U32:
  470. return Inst(Opcode::CompositeConstructU32x3, e1, e2, e3);
  471. case Type::F16:
  472. return Inst(Opcode::CompositeConstructF16x3, e1, e2, e3);
  473. case Type::F32:
  474. return Inst(Opcode::CompositeConstructF32x3, e1, e2, e3);
  475. case Type::F64:
  476. return Inst(Opcode::CompositeConstructF64x3, e1, e2, e3);
  477. default:
  478. ThrowInvalidType(e1.Type());
  479. }
  480. }
  481. Value IREmitter::CompositeConstruct(const Value& e1, const Value& e2, const Value& e3,
  482. const Value& e4) {
  483. if (e1.Type() != e2.Type() || e1.Type() != e3.Type() || e1.Type() != e4.Type()) {
  484. throw InvalidArgument("Mismatching types {}, {}, {}, and {}", e1.Type(), e2.Type(),
  485. e3.Type(), e4.Type());
  486. }
  487. switch (e1.Type()) {
  488. case Type::U32:
  489. return Inst(Opcode::CompositeConstructU32x4, e1, e2, e3, e4);
  490. case Type::F16:
  491. return Inst(Opcode::CompositeConstructF16x4, e1, e2, e3, e4);
  492. case Type::F32:
  493. return Inst(Opcode::CompositeConstructF32x4, e1, e2, e3, e4);
  494. case Type::F64:
  495. return Inst(Opcode::CompositeConstructF64x4, e1, e2, e3, e4);
  496. default:
  497. ThrowInvalidType(e1.Type());
  498. }
  499. }
  500. Value IREmitter::CompositeExtract(const Value& vector, size_t element) {
  501. const auto read{[&](Opcode opcode, size_t limit) -> Value {
  502. if (element >= limit) {
  503. throw InvalidArgument("Out of bounds element {}", element);
  504. }
  505. return Inst(opcode, vector, Value{static_cast<u32>(element)});
  506. }};
  507. switch (vector.Type()) {
  508. case Type::U32x2:
  509. return read(Opcode::CompositeExtractU32x2, 2);
  510. case Type::U32x3:
  511. return read(Opcode::CompositeExtractU32x3, 3);
  512. case Type::U32x4:
  513. return read(Opcode::CompositeExtractU32x4, 4);
  514. case Type::F16x2:
  515. return read(Opcode::CompositeExtractF16x2, 2);
  516. case Type::F16x3:
  517. return read(Opcode::CompositeExtractF16x3, 3);
  518. case Type::F16x4:
  519. return read(Opcode::CompositeExtractF16x4, 4);
  520. case Type::F32x2:
  521. return read(Opcode::CompositeExtractF32x2, 2);
  522. case Type::F32x3:
  523. return read(Opcode::CompositeExtractF32x3, 3);
  524. case Type::F32x4:
  525. return read(Opcode::CompositeExtractF32x4, 4);
  526. case Type::F64x2:
  527. return read(Opcode::CompositeExtractF64x2, 2);
  528. case Type::F64x3:
  529. return read(Opcode::CompositeExtractF64x3, 3);
  530. case Type::F64x4:
  531. return read(Opcode::CompositeExtractF64x4, 4);
  532. default:
  533. ThrowInvalidType(vector.Type());
  534. }
  535. }
  536. Value IREmitter::CompositeInsert(const Value& vector, const Value& object, size_t element) {
  537. const auto insert{[&](Opcode opcode, size_t limit) {
  538. if (element >= limit) {
  539. throw InvalidArgument("Out of bounds element {}", element);
  540. }
  541. return Inst(opcode, vector, object, Value{static_cast<u32>(element)});
  542. }};
  543. switch (vector.Type()) {
  544. case Type::U32x2:
  545. return insert(Opcode::CompositeInsertU32x2, 2);
  546. case Type::U32x3:
  547. return insert(Opcode::CompositeInsertU32x3, 3);
  548. case Type::U32x4:
  549. return insert(Opcode::CompositeInsertU32x4, 4);
  550. case Type::F16x2:
  551. return insert(Opcode::CompositeInsertF16x2, 2);
  552. case Type::F16x3:
  553. return insert(Opcode::CompositeInsertF16x3, 3);
  554. case Type::F16x4:
  555. return insert(Opcode::CompositeInsertF16x4, 4);
  556. case Type::F32x2:
  557. return insert(Opcode::CompositeInsertF32x2, 2);
  558. case Type::F32x3:
  559. return insert(Opcode::CompositeInsertF32x3, 3);
  560. case Type::F32x4:
  561. return insert(Opcode::CompositeInsertF32x4, 4);
  562. case Type::F64x2:
  563. return insert(Opcode::CompositeInsertF64x2, 2);
  564. case Type::F64x3:
  565. return insert(Opcode::CompositeInsertF64x3, 3);
  566. case Type::F64x4:
  567. return insert(Opcode::CompositeInsertF64x4, 4);
  568. default:
  569. ThrowInvalidType(vector.Type());
  570. }
  571. }
  572. Value IREmitter::Select(const U1& condition, const Value& true_value, const Value& false_value) {
  573. if (true_value.Type() != false_value.Type()) {
  574. throw InvalidArgument("Mismatching types {} and {}", true_value.Type(), false_value.Type());
  575. }
  576. switch (true_value.Type()) {
  577. case Type::U1:
  578. return Inst(Opcode::SelectU1, condition, true_value, false_value);
  579. case Type::U8:
  580. return Inst(Opcode::SelectU8, condition, true_value, false_value);
  581. case Type::U16:
  582. return Inst(Opcode::SelectU16, condition, true_value, false_value);
  583. case Type::U32:
  584. return Inst(Opcode::SelectU32, condition, true_value, false_value);
  585. case Type::U64:
  586. return Inst(Opcode::SelectU64, condition, true_value, false_value);
  587. case Type::F32:
  588. return Inst(Opcode::SelectF32, condition, true_value, false_value);
  589. case Type::F64:
  590. return Inst(Opcode::SelectF64, condition, true_value, false_value);
  591. default:
  592. throw InvalidArgument("Invalid type {}", true_value.Type());
  593. }
  594. }
  595. template <>
  596. IR::U32 IREmitter::BitCast<IR::U32, IR::F32>(const IR::F32& value) {
  597. return Inst<IR::U32>(Opcode::BitCastU32F32, value);
  598. }
  599. template <>
  600. IR::F32 IREmitter::BitCast<IR::F32, IR::U32>(const IR::U32& value) {
  601. return Inst<IR::F32>(Opcode::BitCastF32U32, value);
  602. }
  603. template <>
  604. IR::U16 IREmitter::BitCast<IR::U16, IR::F16>(const IR::F16& value) {
  605. return Inst<IR::U16>(Opcode::BitCastU16F16, value);
  606. }
  607. template <>
  608. IR::F16 IREmitter::BitCast<IR::F16, IR::U16>(const IR::U16& value) {
  609. return Inst<IR::F16>(Opcode::BitCastF16U16, value);
  610. }
  611. template <>
  612. IR::U64 IREmitter::BitCast<IR::U64, IR::F64>(const IR::F64& value) {
  613. return Inst<IR::U64>(Opcode::BitCastU64F64, value);
  614. }
  615. template <>
  616. IR::F64 IREmitter::BitCast<IR::F64, IR::U64>(const IR::U64& value) {
  617. return Inst<IR::F64>(Opcode::BitCastF64U64, value);
  618. }
  619. U64 IREmitter::PackUint2x32(const Value& vector) {
  620. return Inst<U64>(Opcode::PackUint2x32, vector);
  621. }
  622. Value IREmitter::UnpackUint2x32(const U64& value) {
  623. return Inst<Value>(Opcode::UnpackUint2x32, value);
  624. }
  625. U32 IREmitter::PackFloat2x16(const Value& vector) {
  626. return Inst<U32>(Opcode::PackFloat2x16, vector);
  627. }
  628. Value IREmitter::UnpackFloat2x16(const U32& value) {
  629. return Inst(Opcode::UnpackFloat2x16, value);
  630. }
  631. U32 IREmitter::PackHalf2x16(const Value& vector) {
  632. return Inst<U32>(Opcode::PackHalf2x16, vector);
  633. }
  634. Value IREmitter::UnpackHalf2x16(const U32& value) {
  635. return Inst(Opcode::UnpackHalf2x16, value);
  636. }
  637. F64 IREmitter::PackDouble2x32(const Value& vector) {
  638. return Inst<F64>(Opcode::PackDouble2x32, vector);
  639. }
  640. Value IREmitter::UnpackDouble2x32(const F64& value) {
  641. return Inst<Value>(Opcode::UnpackDouble2x32, value);
  642. }
  643. F16F32F64 IREmitter::FPMul(const F16F32F64& a, const F16F32F64& b, FpControl control) {
  644. if (a.Type() != b.Type()) {
  645. throw InvalidArgument("Mismatching types {} and {}", a.Type(), b.Type());
  646. }
  647. switch (a.Type()) {
  648. case Type::F16:
  649. return Inst<F16>(Opcode::FPMul16, Flags{control}, a, b);
  650. case Type::F32:
  651. return Inst<F32>(Opcode::FPMul32, Flags{control}, a, b);
  652. case Type::F64:
  653. return Inst<F64>(Opcode::FPMul64, Flags{control}, a, b);
  654. default:
  655. ThrowInvalidType(a.Type());
  656. }
  657. }
  658. F16F32F64 IREmitter::FPFma(const F16F32F64& a, const F16F32F64& b, const F16F32F64& c,
  659. FpControl control) {
  660. if (a.Type() != b.Type() || a.Type() != c.Type()) {
  661. throw InvalidArgument("Mismatching types {}, {}, and {}", a.Type(), b.Type(), c.Type());
  662. }
  663. switch (a.Type()) {
  664. case Type::F16:
  665. return Inst<F16>(Opcode::FPFma16, Flags{control}, a, b, c);
  666. case Type::F32:
  667. return Inst<F32>(Opcode::FPFma32, Flags{control}, a, b, c);
  668. case Type::F64:
  669. return Inst<F64>(Opcode::FPFma64, Flags{control}, a, b, c);
  670. default:
  671. ThrowInvalidType(a.Type());
  672. }
  673. }
  674. F16F32F64 IREmitter::FPAbs(const F16F32F64& value) {
  675. switch (value.Type()) {
  676. case Type::F16:
  677. return Inst<F16>(Opcode::FPAbs16, value);
  678. case Type::F32:
  679. return Inst<F32>(Opcode::FPAbs32, value);
  680. case Type::F64:
  681. return Inst<F64>(Opcode::FPAbs64, value);
  682. default:
  683. ThrowInvalidType(value.Type());
  684. }
  685. }
  686. F16F32F64 IREmitter::FPNeg(const F16F32F64& value) {
  687. switch (value.Type()) {
  688. case Type::F16:
  689. return Inst<F16>(Opcode::FPNeg16, value);
  690. case Type::F32:
  691. return Inst<F32>(Opcode::FPNeg32, value);
  692. case Type::F64:
  693. return Inst<F64>(Opcode::FPNeg64, value);
  694. default:
  695. ThrowInvalidType(value.Type());
  696. }
  697. }
  698. F16F32F64 IREmitter::FPAbsNeg(const F16F32F64& value, bool abs, bool neg) {
  699. F16F32F64 result{value};
  700. if (abs) {
  701. result = FPAbs(result);
  702. }
  703. if (neg) {
  704. result = FPNeg(result);
  705. }
  706. return result;
  707. }
  708. F32 IREmitter::FPCos(const F32& value) {
  709. return Inst<F32>(Opcode::FPCos, value);
  710. }
  711. F32 IREmitter::FPSin(const F32& value) {
  712. return Inst<F32>(Opcode::FPSin, value);
  713. }
  714. F32 IREmitter::FPExp2(const F32& value) {
  715. return Inst<F32>(Opcode::FPExp2, value);
  716. }
  717. F32 IREmitter::FPLog2(const F32& value) {
  718. return Inst<F32>(Opcode::FPLog2, value);
  719. }
  720. F32F64 IREmitter::FPRecip(const F32F64& value) {
  721. switch (value.Type()) {
  722. case Type::F32:
  723. return Inst<F32>(Opcode::FPRecip32, value);
  724. case Type::F64:
  725. return Inst<F64>(Opcode::FPRecip64, value);
  726. default:
  727. ThrowInvalidType(value.Type());
  728. }
  729. }
  730. F32F64 IREmitter::FPRecipSqrt(const F32F64& value) {
  731. switch (value.Type()) {
  732. case Type::F32:
  733. return Inst<F32>(Opcode::FPRecipSqrt32, value);
  734. case Type::F64:
  735. return Inst<F64>(Opcode::FPRecipSqrt64, value);
  736. default:
  737. ThrowInvalidType(value.Type());
  738. }
  739. }
  740. F32 IREmitter::FPSqrt(const F32& value) {
  741. return Inst<F32>(Opcode::FPSqrt, value);
  742. }
  743. F16F32F64 IREmitter::FPSaturate(const F16F32F64& value) {
  744. switch (value.Type()) {
  745. case Type::F16:
  746. return Inst<F16>(Opcode::FPSaturate16, value);
  747. case Type::F32:
  748. return Inst<F32>(Opcode::FPSaturate32, value);
  749. case Type::F64:
  750. return Inst<F64>(Opcode::FPSaturate64, value);
  751. default:
  752. ThrowInvalidType(value.Type());
  753. }
  754. }
  755. F16F32F64 IREmitter::FPClamp(const F16F32F64& value, const F16F32F64& min_value,
  756. const F16F32F64& max_value) {
  757. if (value.Type() != min_value.Type() || value.Type() != max_value.Type()) {
  758. throw InvalidArgument("Mismatching types {}, {}, and {}", value.Type(), min_value.Type(),
  759. max_value.Type());
  760. }
  761. switch (value.Type()) {
  762. case Type::F16:
  763. return Inst<F16>(Opcode::FPClamp16, value, min_value, max_value);
  764. case Type::F32:
  765. return Inst<F32>(Opcode::FPClamp32, value, min_value, max_value);
  766. case Type::F64:
  767. return Inst<F64>(Opcode::FPClamp64, value, min_value, max_value);
  768. default:
  769. ThrowInvalidType(value.Type());
  770. }
  771. }
  772. F16F32F64 IREmitter::FPRoundEven(const F16F32F64& value, FpControl control) {
  773. switch (value.Type()) {
  774. case Type::F16:
  775. return Inst<F16>(Opcode::FPRoundEven16, Flags{control}, value);
  776. case Type::F32:
  777. return Inst<F32>(Opcode::FPRoundEven32, Flags{control}, value);
  778. case Type::F64:
  779. return Inst<F64>(Opcode::FPRoundEven64, Flags{control}, value);
  780. default:
  781. ThrowInvalidType(value.Type());
  782. }
  783. }
  784. F16F32F64 IREmitter::FPFloor(const F16F32F64& value, FpControl control) {
  785. switch (value.Type()) {
  786. case Type::F16:
  787. return Inst<F16>(Opcode::FPFloor16, Flags{control}, value);
  788. case Type::F32:
  789. return Inst<F32>(Opcode::FPFloor32, Flags{control}, value);
  790. case Type::F64:
  791. return Inst<F64>(Opcode::FPFloor64, Flags{control}, value);
  792. default:
  793. ThrowInvalidType(value.Type());
  794. }
  795. }
  796. F16F32F64 IREmitter::FPCeil(const F16F32F64& value, FpControl control) {
  797. switch (value.Type()) {
  798. case Type::F16:
  799. return Inst<F16>(Opcode::FPCeil16, Flags{control}, value);
  800. case Type::F32:
  801. return Inst<F32>(Opcode::FPCeil32, Flags{control}, value);
  802. case Type::F64:
  803. return Inst<F64>(Opcode::FPCeil64, Flags{control}, value);
  804. default:
  805. ThrowInvalidType(value.Type());
  806. }
  807. }
  808. F16F32F64 IREmitter::FPTrunc(const F16F32F64& value, FpControl control) {
  809. switch (value.Type()) {
  810. case Type::F16:
  811. return Inst<F16>(Opcode::FPTrunc16, Flags{control}, value);
  812. case Type::F32:
  813. return Inst<F32>(Opcode::FPTrunc32, Flags{control}, value);
  814. case Type::F64:
  815. return Inst<F64>(Opcode::FPTrunc64, Flags{control}, value);
  816. default:
  817. ThrowInvalidType(value.Type());
  818. }
  819. }
  820. U1 IREmitter::FPEqual(const F16F32F64& lhs, const F16F32F64& rhs, FpControl control, bool ordered) {
  821. if (lhs.Type() != rhs.Type()) {
  822. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  823. }
  824. switch (lhs.Type()) {
  825. case Type::F16:
  826. return Inst<U1>(ordered ? Opcode::FPOrdEqual16 : Opcode::FPUnordEqual16, Flags{control},
  827. lhs, rhs);
  828. case Type::F32:
  829. return Inst<U1>(ordered ? Opcode::FPOrdEqual32 : Opcode::FPUnordEqual32, Flags{control},
  830. lhs, rhs);
  831. case Type::F64:
  832. return Inst<U1>(ordered ? Opcode::FPOrdEqual64 : Opcode::FPUnordEqual64, Flags{control},
  833. lhs, rhs);
  834. default:
  835. ThrowInvalidType(lhs.Type());
  836. }
  837. }
  838. U1 IREmitter::FPNotEqual(const F16F32F64& lhs, const F16F32F64& rhs, FpControl control,
  839. bool ordered) {
  840. if (lhs.Type() != rhs.Type()) {
  841. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  842. }
  843. switch (lhs.Type()) {
  844. case Type::F16:
  845. return Inst<U1>(ordered ? Opcode::FPOrdNotEqual16 : Opcode::FPUnordNotEqual16,
  846. Flags{control}, lhs, rhs);
  847. case Type::F32:
  848. return Inst<U1>(ordered ? Opcode::FPOrdNotEqual32 : Opcode::FPUnordNotEqual32,
  849. Flags{control}, lhs, rhs);
  850. case Type::F64:
  851. return Inst<U1>(ordered ? Opcode::FPOrdNotEqual64 : Opcode::FPUnordNotEqual64,
  852. Flags{control}, lhs, rhs);
  853. default:
  854. ThrowInvalidType(lhs.Type());
  855. }
  856. }
  857. U1 IREmitter::FPLessThan(const F16F32F64& lhs, const F16F32F64& rhs, FpControl control,
  858. bool ordered) {
  859. if (lhs.Type() != rhs.Type()) {
  860. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  861. }
  862. switch (lhs.Type()) {
  863. case Type::F16:
  864. return Inst<U1>(ordered ? Opcode::FPOrdLessThan16 : Opcode::FPUnordLessThan16,
  865. Flags{control}, lhs, rhs);
  866. case Type::F32:
  867. return Inst<U1>(ordered ? Opcode::FPOrdLessThan32 : Opcode::FPUnordLessThan32,
  868. Flags{control}, lhs, rhs);
  869. case Type::F64:
  870. return Inst<U1>(ordered ? Opcode::FPOrdLessThan64 : Opcode::FPUnordLessThan64,
  871. Flags{control}, lhs, rhs);
  872. default:
  873. ThrowInvalidType(lhs.Type());
  874. }
  875. }
  876. U1 IREmitter::FPGreaterThan(const F16F32F64& lhs, const F16F32F64& rhs, FpControl control,
  877. bool ordered) {
  878. if (lhs.Type() != rhs.Type()) {
  879. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  880. }
  881. switch (lhs.Type()) {
  882. case Type::F16:
  883. return Inst<U1>(ordered ? Opcode::FPOrdGreaterThan16 : Opcode::FPUnordGreaterThan16,
  884. Flags{control}, lhs, rhs);
  885. case Type::F32:
  886. return Inst<U1>(ordered ? Opcode::FPOrdGreaterThan32 : Opcode::FPUnordGreaterThan32,
  887. Flags{control}, lhs, rhs);
  888. case Type::F64:
  889. return Inst<U1>(ordered ? Opcode::FPOrdGreaterThan64 : Opcode::FPUnordGreaterThan64,
  890. Flags{control}, lhs, rhs);
  891. default:
  892. ThrowInvalidType(lhs.Type());
  893. }
  894. }
  895. U1 IREmitter::FPLessThanEqual(const F16F32F64& lhs, const F16F32F64& rhs, FpControl control,
  896. bool ordered) {
  897. if (lhs.Type() != rhs.Type()) {
  898. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  899. }
  900. switch (lhs.Type()) {
  901. case Type::F16:
  902. return Inst<U1>(ordered ? Opcode::FPOrdLessThanEqual16 : Opcode::FPUnordLessThanEqual16,
  903. Flags{control}, lhs, rhs);
  904. case Type::F32:
  905. return Inst<U1>(ordered ? Opcode::FPOrdLessThanEqual32 : Opcode::FPUnordLessThanEqual32,
  906. Flags{control}, lhs, rhs);
  907. case Type::F64:
  908. return Inst<U1>(ordered ? Opcode::FPOrdLessThanEqual64 : Opcode::FPUnordLessThanEqual64,
  909. Flags{control}, lhs, rhs);
  910. default:
  911. ThrowInvalidType(lhs.Type());
  912. }
  913. }
  914. U1 IREmitter::FPGreaterThanEqual(const F16F32F64& lhs, const F16F32F64& rhs, FpControl control,
  915. bool ordered) {
  916. if (lhs.Type() != rhs.Type()) {
  917. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  918. }
  919. switch (lhs.Type()) {
  920. case Type::F16:
  921. return Inst<U1>(ordered ? Opcode::FPOrdGreaterThanEqual16
  922. : Opcode::FPUnordGreaterThanEqual16,
  923. Flags{control}, lhs, rhs);
  924. case Type::F32:
  925. return Inst<U1>(ordered ? Opcode::FPOrdGreaterThanEqual32
  926. : Opcode::FPUnordGreaterThanEqual32,
  927. Flags{control}, lhs, rhs);
  928. case Type::F64:
  929. return Inst<U1>(ordered ? Opcode::FPOrdGreaterThanEqual64
  930. : Opcode::FPUnordGreaterThanEqual64,
  931. Flags{control}, lhs, rhs);
  932. default:
  933. ThrowInvalidType(lhs.Type());
  934. }
  935. }
  936. U1 IREmitter::FPIsNan(const F16F32F64& value) {
  937. switch (value.Type()) {
  938. case Type::F16:
  939. return Inst<U1>(Opcode::FPIsNan16, value);
  940. case Type::F32:
  941. return Inst<U1>(Opcode::FPIsNan32, value);
  942. case Type::F64:
  943. return Inst<U1>(Opcode::FPIsNan64, value);
  944. default:
  945. ThrowInvalidType(value.Type());
  946. }
  947. }
  948. U1 IREmitter::FPOrdered(const F16F32F64& lhs, const F16F32F64& rhs) {
  949. if (lhs.Type() != rhs.Type()) {
  950. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  951. }
  952. return LogicalAnd(LogicalNot(FPIsNan(lhs)), LogicalNot(FPIsNan(rhs)));
  953. }
  954. U1 IREmitter::FPUnordered(const F16F32F64& lhs, const F16F32F64& rhs) {
  955. if (lhs.Type() != rhs.Type()) {
  956. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  957. }
  958. return LogicalOr(FPIsNan(lhs), FPIsNan(rhs));
  959. }
  960. F32F64 IREmitter::FPMax(const F32F64& lhs, const F32F64& rhs, FpControl control) {
  961. if (lhs.Type() != rhs.Type()) {
  962. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  963. }
  964. switch (lhs.Type()) {
  965. case Type::F32:
  966. return Inst<F32>(Opcode::FPMax32, Flags{control}, lhs, rhs);
  967. case Type::F64:
  968. return Inst<F64>(Opcode::FPMax64, Flags{control}, lhs, rhs);
  969. default:
  970. ThrowInvalidType(lhs.Type());
  971. }
  972. }
  973. F32F64 IREmitter::FPMin(const F32F64& lhs, const F32F64& rhs, FpControl control) {
  974. if (lhs.Type() != rhs.Type()) {
  975. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  976. }
  977. switch (lhs.Type()) {
  978. case Type::F32:
  979. return Inst<F32>(Opcode::FPMin32, Flags{control}, lhs, rhs);
  980. case Type::F64:
  981. return Inst<F64>(Opcode::FPMin64, Flags{control}, lhs, rhs);
  982. default:
  983. ThrowInvalidType(lhs.Type());
  984. }
  985. }
  986. U32U64 IREmitter::IAdd(const U32U64& a, const U32U64& b) {
  987. if (a.Type() != b.Type()) {
  988. throw InvalidArgument("Mismatching types {} and {}", a.Type(), b.Type());
  989. }
  990. switch (a.Type()) {
  991. case Type::U32:
  992. return Inst<U32>(Opcode::IAdd32, a, b);
  993. case Type::U64:
  994. return Inst<U64>(Opcode::IAdd64, a, b);
  995. default:
  996. ThrowInvalidType(a.Type());
  997. }
  998. }
  999. U32U64 IREmitter::ISub(const U32U64& a, const U32U64& b) {
  1000. if (a.Type() != b.Type()) {
  1001. throw InvalidArgument("Mismatching types {} and {}", a.Type(), b.Type());
  1002. }
  1003. switch (a.Type()) {
  1004. case Type::U32:
  1005. return Inst<U32>(Opcode::ISub32, a, b);
  1006. case Type::U64:
  1007. return Inst<U64>(Opcode::ISub64, a, b);
  1008. default:
  1009. ThrowInvalidType(a.Type());
  1010. }
  1011. }
  1012. U32 IREmitter::IMul(const U32& a, const U32& b) {
  1013. return Inst<U32>(Opcode::IMul32, a, b);
  1014. }
  1015. U32 IREmitter::IDiv(const U32& a, const U32& b, bool is_signed) {
  1016. return Inst<U32>(is_signed ? Opcode::SDiv32 : Opcode::UDiv32, a, b);
  1017. }
  1018. U32U64 IREmitter::INeg(const U32U64& value) {
  1019. switch (value.Type()) {
  1020. case Type::U32:
  1021. return Inst<U32>(Opcode::INeg32, value);
  1022. case Type::U64:
  1023. return Inst<U64>(Opcode::INeg64, value);
  1024. default:
  1025. ThrowInvalidType(value.Type());
  1026. }
  1027. }
  1028. U32 IREmitter::IAbs(const U32& value) {
  1029. return Inst<U32>(Opcode::IAbs32, value);
  1030. }
  1031. U32U64 IREmitter::ShiftLeftLogical(const U32U64& base, const U32& shift) {
  1032. switch (base.Type()) {
  1033. case Type::U32:
  1034. return Inst<U32>(Opcode::ShiftLeftLogical32, base, shift);
  1035. case Type::U64:
  1036. return Inst<U64>(Opcode::ShiftLeftLogical64, base, shift);
  1037. default:
  1038. ThrowInvalidType(base.Type());
  1039. }
  1040. }
  1041. U32U64 IREmitter::ShiftRightLogical(const U32U64& base, const U32& shift) {
  1042. switch (base.Type()) {
  1043. case Type::U32:
  1044. return Inst<U32>(Opcode::ShiftRightLogical32, base, shift);
  1045. case Type::U64:
  1046. return Inst<U64>(Opcode::ShiftRightLogical64, base, shift);
  1047. default:
  1048. ThrowInvalidType(base.Type());
  1049. }
  1050. }
  1051. U32U64 IREmitter::ShiftRightArithmetic(const U32U64& base, const U32& shift) {
  1052. switch (base.Type()) {
  1053. case Type::U32:
  1054. return Inst<U32>(Opcode::ShiftRightArithmetic32, base, shift);
  1055. case Type::U64:
  1056. return Inst<U64>(Opcode::ShiftRightArithmetic64, base, shift);
  1057. default:
  1058. ThrowInvalidType(base.Type());
  1059. }
  1060. }
  1061. U32 IREmitter::BitwiseAnd(const U32& a, const U32& b) {
  1062. return Inst<U32>(Opcode::BitwiseAnd32, a, b);
  1063. }
  1064. U32 IREmitter::BitwiseOr(const U32& a, const U32& b) {
  1065. return Inst<U32>(Opcode::BitwiseOr32, a, b);
  1066. }
  1067. U32 IREmitter::BitwiseXor(const U32& a, const U32& b) {
  1068. return Inst<U32>(Opcode::BitwiseXor32, a, b);
  1069. }
  1070. U32 IREmitter::BitFieldInsert(const U32& base, const U32& insert, const U32& offset,
  1071. const U32& count) {
  1072. return Inst<U32>(Opcode::BitFieldInsert, base, insert, offset, count);
  1073. }
  1074. U32 IREmitter::BitFieldExtract(const U32& base, const U32& offset, const U32& count,
  1075. bool is_signed) {
  1076. return Inst<U32>(is_signed ? Opcode::BitFieldSExtract : Opcode::BitFieldUExtract, base, offset,
  1077. count);
  1078. }
  1079. U32 IREmitter::BitReverse(const U32& value) {
  1080. return Inst<U32>(Opcode::BitReverse32, value);
  1081. }
  1082. U32 IREmitter::BitCount(const U32& value) {
  1083. return Inst<U32>(Opcode::BitCount32, value);
  1084. }
  1085. U32 IREmitter::BitwiseNot(const U32& value) {
  1086. return Inst<U32>(Opcode::BitwiseNot32, value);
  1087. }
  1088. U32 IREmitter::FindSMsb(const U32& value) {
  1089. return Inst<U32>(Opcode::FindSMsb32, value);
  1090. }
  1091. U32 IREmitter::FindUMsb(const U32& value) {
  1092. return Inst<U32>(Opcode::FindUMsb32, value);
  1093. }
  1094. U32 IREmitter::SMin(const U32& a, const U32& b) {
  1095. return Inst<U32>(Opcode::SMin32, a, b);
  1096. }
  1097. U32 IREmitter::UMin(const U32& a, const U32& b) {
  1098. return Inst<U32>(Opcode::UMin32, a, b);
  1099. }
  1100. U32 IREmitter::IMin(const U32& a, const U32& b, bool is_signed) {
  1101. return is_signed ? SMin(a, b) : UMin(a, b);
  1102. }
  1103. U32 IREmitter::SMax(const U32& a, const U32& b) {
  1104. return Inst<U32>(Opcode::SMax32, a, b);
  1105. }
  1106. U32 IREmitter::UMax(const U32& a, const U32& b) {
  1107. return Inst<U32>(Opcode::UMax32, a, b);
  1108. }
  1109. U32 IREmitter::IMax(const U32& a, const U32& b, bool is_signed) {
  1110. return is_signed ? SMax(a, b) : UMax(a, b);
  1111. }
  1112. U32 IREmitter::SClamp(const U32& value, const U32& min, const U32& max) {
  1113. return Inst<U32>(Opcode::SClamp32, value, min, max);
  1114. }
  1115. U32 IREmitter::UClamp(const U32& value, const U32& min, const U32& max) {
  1116. return Inst<U32>(Opcode::UClamp32, value, min, max);
  1117. }
  1118. U1 IREmitter::ILessThan(const U32& lhs, const U32& rhs, bool is_signed) {
  1119. return Inst<U1>(is_signed ? Opcode::SLessThan : Opcode::ULessThan, lhs, rhs);
  1120. }
  1121. U1 IREmitter::IEqual(const U32U64& lhs, const U32U64& rhs) {
  1122. if (lhs.Type() != rhs.Type()) {
  1123. throw InvalidArgument("Mismatching types {} and {}", lhs.Type(), rhs.Type());
  1124. }
  1125. switch (lhs.Type()) {
  1126. case Type::U32:
  1127. return Inst<U1>(Opcode::IEqual, lhs, rhs);
  1128. case Type::U64: {
  1129. // Manually compare the unpacked values
  1130. const Value lhs_vector{UnpackUint2x32(lhs)};
  1131. const Value rhs_vector{UnpackUint2x32(rhs)};
  1132. return LogicalAnd(IEqual(IR::U32{CompositeExtract(lhs_vector, 0)},
  1133. IR::U32{CompositeExtract(rhs_vector, 0)}),
  1134. IEqual(IR::U32{CompositeExtract(lhs_vector, 1)},
  1135. IR::U32{CompositeExtract(rhs_vector, 1)}));
  1136. }
  1137. default:
  1138. ThrowInvalidType(lhs.Type());
  1139. }
  1140. }
  1141. U1 IREmitter::ILessThanEqual(const U32& lhs, const U32& rhs, bool is_signed) {
  1142. return Inst<U1>(is_signed ? Opcode::SLessThanEqual : Opcode::ULessThanEqual, lhs, rhs);
  1143. }
  1144. U1 IREmitter::IGreaterThan(const U32& lhs, const U32& rhs, bool is_signed) {
  1145. return Inst<U1>(is_signed ? Opcode::SGreaterThan : Opcode::UGreaterThan, lhs, rhs);
  1146. }
  1147. U1 IREmitter::INotEqual(const U32& lhs, const U32& rhs) {
  1148. return Inst<U1>(Opcode::INotEqual, lhs, rhs);
  1149. }
  1150. U1 IREmitter::IGreaterThanEqual(const U32& lhs, const U32& rhs, bool is_signed) {
  1151. return Inst<U1>(is_signed ? Opcode::SGreaterThanEqual : Opcode::UGreaterThanEqual, lhs, rhs);
  1152. }
  1153. U32 IREmitter::SharedAtomicIAdd(const U32& pointer_offset, const U32& value) {
  1154. return Inst<U32>(Opcode::SharedAtomicIAdd32, pointer_offset, value);
  1155. }
  1156. U32 IREmitter::SharedAtomicSMin(const U32& pointer_offset, const U32& value) {
  1157. return Inst<U32>(Opcode::SharedAtomicSMin32, pointer_offset, value);
  1158. }
  1159. U32 IREmitter::SharedAtomicUMin(const U32& pointer_offset, const U32& value) {
  1160. return Inst<U32>(Opcode::SharedAtomicUMin32, pointer_offset, value);
  1161. }
  1162. U32 IREmitter::SharedAtomicIMin(const U32& pointer_offset, const U32& value, bool is_signed) {
  1163. return is_signed ? SharedAtomicSMin(pointer_offset, value)
  1164. : SharedAtomicUMin(pointer_offset, value);
  1165. }
  1166. U32 IREmitter::SharedAtomicSMax(const U32& pointer_offset, const U32& value) {
  1167. return Inst<U32>(Opcode::SharedAtomicSMax32, pointer_offset, value);
  1168. }
  1169. U32 IREmitter::SharedAtomicUMax(const U32& pointer_offset, const U32& value) {
  1170. return Inst<U32>(Opcode::SharedAtomicUMax32, pointer_offset, value);
  1171. }
  1172. U32 IREmitter::SharedAtomicIMax(const U32& pointer_offset, const U32& value, bool is_signed) {
  1173. return is_signed ? SharedAtomicSMax(pointer_offset, value)
  1174. : SharedAtomicUMax(pointer_offset, value);
  1175. }
  1176. U32 IREmitter::SharedAtomicInc(const U32& pointer_offset, const U32& value) {
  1177. return Inst<U32>(Opcode::SharedAtomicInc32, pointer_offset, value);
  1178. }
  1179. U32 IREmitter::SharedAtomicDec(const U32& pointer_offset, const U32& value) {
  1180. return Inst<U32>(Opcode::SharedAtomicDec32, pointer_offset, value);
  1181. }
  1182. U32 IREmitter::SharedAtomicAnd(const U32& pointer_offset, const U32& value) {
  1183. return Inst<U32>(Opcode::SharedAtomicAnd32, pointer_offset, value);
  1184. }
  1185. U32 IREmitter::SharedAtomicOr(const U32& pointer_offset, const U32& value) {
  1186. return Inst<U32>(Opcode::SharedAtomicOr32, pointer_offset, value);
  1187. }
  1188. U32 IREmitter::SharedAtomicXor(const U32& pointer_offset, const U32& value) {
  1189. return Inst<U32>(Opcode::SharedAtomicXor32, pointer_offset, value);
  1190. }
  1191. U32U64 IREmitter::SharedAtomicExchange(const U32& pointer_offset, const U32U64& value) {
  1192. switch (value.Type()) {
  1193. case Type::U32:
  1194. return Inst<U32>(Opcode::SharedAtomicExchange32, pointer_offset, value);
  1195. case Type::U64:
  1196. return Inst<U64>(Opcode::SharedAtomicExchange64, pointer_offset, value);
  1197. default:
  1198. ThrowInvalidType(pointer_offset.Type());
  1199. }
  1200. }
  1201. U32U64 IREmitter::GlobalAtomicIAdd(const U64& pointer_offset, const U32U64& value) {
  1202. switch (value.Type()) {
  1203. case Type::U32:
  1204. return Inst<U32>(Opcode::GlobalAtomicIAdd32, pointer_offset, value);
  1205. case Type::U64:
  1206. return Inst<U64>(Opcode::GlobalAtomicIAdd64, pointer_offset, value);
  1207. default:
  1208. ThrowInvalidType(value.Type());
  1209. }
  1210. }
  1211. U32U64 IREmitter::GlobalAtomicSMin(const U64& pointer_offset, const U32U64& value) {
  1212. switch (value.Type()) {
  1213. case Type::U32:
  1214. return Inst<U32>(Opcode::GlobalAtomicSMin32, pointer_offset, value);
  1215. case Type::U64:
  1216. return Inst<U64>(Opcode::GlobalAtomicSMin64, pointer_offset, value);
  1217. default:
  1218. ThrowInvalidType(value.Type());
  1219. }
  1220. }
  1221. U32U64 IREmitter::GlobalAtomicUMin(const U64& pointer_offset, const U32U64& value) {
  1222. switch (value.Type()) {
  1223. case Type::U32:
  1224. return Inst<U32>(Opcode::GlobalAtomicUMin32, pointer_offset, value);
  1225. case Type::U64:
  1226. return Inst<U64>(Opcode::GlobalAtomicUMin64, pointer_offset, value);
  1227. default:
  1228. ThrowInvalidType(value.Type());
  1229. }
  1230. }
  1231. U32U64 IREmitter::GlobalAtomicIMin(const U64& pointer_offset, const U32U64& value, bool is_signed) {
  1232. return is_signed ? GlobalAtomicSMin(pointer_offset, value)
  1233. : GlobalAtomicUMin(pointer_offset, value);
  1234. }
  1235. U32U64 IREmitter::GlobalAtomicSMax(const U64& pointer_offset, const U32U64& value) {
  1236. switch (value.Type()) {
  1237. case Type::U32:
  1238. return Inst<U32>(Opcode::GlobalAtomicSMax32, pointer_offset, value);
  1239. case Type::U64:
  1240. return Inst<U64>(Opcode::GlobalAtomicSMax64, pointer_offset, value);
  1241. default:
  1242. ThrowInvalidType(value.Type());
  1243. }
  1244. }
  1245. U32U64 IREmitter::GlobalAtomicUMax(const U64& pointer_offset, const U32U64& value) {
  1246. switch (value.Type()) {
  1247. case Type::U32:
  1248. return Inst<U32>(Opcode::GlobalAtomicUMax32, pointer_offset, value);
  1249. case Type::U64:
  1250. return Inst<U64>(Opcode::GlobalAtomicUMax64, pointer_offset, value);
  1251. default:
  1252. ThrowInvalidType(value.Type());
  1253. }
  1254. }
  1255. U32U64 IREmitter::GlobalAtomicIMax(const U64& pointer_offset, const U32U64& value, bool is_signed) {
  1256. return is_signed ? GlobalAtomicSMax(pointer_offset, value)
  1257. : GlobalAtomicUMax(pointer_offset, value);
  1258. }
  1259. U32 IREmitter::GlobalAtomicInc(const U64& pointer_offset, const U32& value) {
  1260. return Inst<U32>(Opcode::GlobalAtomicInc32, pointer_offset, value);
  1261. }
  1262. U32 IREmitter::GlobalAtomicDec(const U64& pointer_offset, const U32& value) {
  1263. return Inst<U32>(Opcode::GlobalAtomicDec32, pointer_offset, value);
  1264. }
  1265. U32U64 IREmitter::GlobalAtomicAnd(const U64& pointer_offset, const U32U64& value) {
  1266. switch (value.Type()) {
  1267. case Type::U32:
  1268. return Inst<U32>(Opcode::GlobalAtomicAnd32, pointer_offset, value);
  1269. case Type::U64:
  1270. return Inst<U64>(Opcode::GlobalAtomicAnd64, pointer_offset, value);
  1271. default:
  1272. ThrowInvalidType(value.Type());
  1273. }
  1274. }
  1275. U32U64 IREmitter::GlobalAtomicOr(const U64& pointer_offset, const U32U64& value) {
  1276. switch (value.Type()) {
  1277. case Type::U32:
  1278. return Inst<U32>(Opcode::GlobalAtomicOr32, pointer_offset, value);
  1279. case Type::U64:
  1280. return Inst<U64>(Opcode::GlobalAtomicOr64, pointer_offset, value);
  1281. default:
  1282. ThrowInvalidType(value.Type());
  1283. }
  1284. }
  1285. U32U64 IREmitter::GlobalAtomicXor(const U64& pointer_offset, const U32U64& value) {
  1286. switch (value.Type()) {
  1287. case Type::U32:
  1288. return Inst<U32>(Opcode::GlobalAtomicXor32, pointer_offset, value);
  1289. case Type::U64:
  1290. return Inst<U64>(Opcode::GlobalAtomicXor64, pointer_offset, value);
  1291. default:
  1292. ThrowInvalidType(value.Type());
  1293. }
  1294. }
  1295. U32U64 IREmitter::GlobalAtomicExchange(const U64& pointer_offset, const U32U64& value) {
  1296. switch (value.Type()) {
  1297. case Type::U32:
  1298. return Inst<U32>(Opcode::GlobalAtomicExchange32, pointer_offset, value);
  1299. case Type::U64:
  1300. return Inst<U64>(Opcode::GlobalAtomicExchange64, pointer_offset, value);
  1301. default:
  1302. ThrowInvalidType(pointer_offset.Type());
  1303. }
  1304. }
  1305. F32 IREmitter::GlobalAtomicF32Add(const U64& pointer_offset, const Value& value,
  1306. const FpControl control) {
  1307. return Inst<F32>(Opcode::GlobalAtomicAddF32, Flags{control}, pointer_offset, value);
  1308. }
  1309. Value IREmitter::GlobalAtomicF16x2Add(const U64& pointer_offset, const Value& value,
  1310. const FpControl control) {
  1311. return Inst(Opcode::GlobalAtomicAddF16x2, Flags{control}, pointer_offset, value);
  1312. }
  1313. Value IREmitter::GlobalAtomicF16x2Min(const U64& pointer_offset, const Value& value,
  1314. const FpControl control) {
  1315. return Inst(Opcode::GlobalAtomicMinF16x2, Flags{control}, pointer_offset, value);
  1316. }
  1317. Value IREmitter::GlobalAtomicF16x2Max(const U64& pointer_offset, const Value& value,
  1318. const FpControl control) {
  1319. return Inst(Opcode::GlobalAtomicMaxF16x2, Flags{control}, pointer_offset, value);
  1320. }
  1321. U1 IREmitter::LogicalOr(const U1& a, const U1& b) {
  1322. return Inst<U1>(Opcode::LogicalOr, a, b);
  1323. }
  1324. U1 IREmitter::LogicalAnd(const U1& a, const U1& b) {
  1325. return Inst<U1>(Opcode::LogicalAnd, a, b);
  1326. }
  1327. U1 IREmitter::LogicalXor(const U1& a, const U1& b) {
  1328. return Inst<U1>(Opcode::LogicalXor, a, b);
  1329. }
  1330. U1 IREmitter::LogicalNot(const U1& value) {
  1331. return Inst<U1>(Opcode::LogicalNot, value);
  1332. }
  1333. U32U64 IREmitter::ConvertFToS(size_t bitsize, const F16F32F64& value) {
  1334. switch (bitsize) {
  1335. case 16:
  1336. switch (value.Type()) {
  1337. case Type::F16:
  1338. return Inst<U32>(Opcode::ConvertS16F16, value);
  1339. case Type::F32:
  1340. return Inst<U32>(Opcode::ConvertS16F32, value);
  1341. case Type::F64:
  1342. return Inst<U32>(Opcode::ConvertS16F64, value);
  1343. default:
  1344. ThrowInvalidType(value.Type());
  1345. }
  1346. case 32:
  1347. switch (value.Type()) {
  1348. case Type::F16:
  1349. return Inst<U32>(Opcode::ConvertS32F16, value);
  1350. case Type::F32:
  1351. return Inst<U32>(Opcode::ConvertS32F32, value);
  1352. case Type::F64:
  1353. return Inst<U32>(Opcode::ConvertS32F64, value);
  1354. default:
  1355. ThrowInvalidType(value.Type());
  1356. }
  1357. case 64:
  1358. switch (value.Type()) {
  1359. case Type::F16:
  1360. return Inst<U64>(Opcode::ConvertS64F16, value);
  1361. case Type::F32:
  1362. return Inst<U64>(Opcode::ConvertS64F32, value);
  1363. case Type::F64:
  1364. return Inst<U64>(Opcode::ConvertS64F64, value);
  1365. default:
  1366. ThrowInvalidType(value.Type());
  1367. }
  1368. default:
  1369. throw InvalidArgument("Invalid destination bitsize {}", bitsize);
  1370. }
  1371. }
  1372. U32U64 IREmitter::ConvertFToU(size_t bitsize, const F16F32F64& value) {
  1373. switch (bitsize) {
  1374. case 16:
  1375. switch (value.Type()) {
  1376. case Type::F16:
  1377. return Inst<U32>(Opcode::ConvertU16F16, value);
  1378. case Type::F32:
  1379. return Inst<U32>(Opcode::ConvertU16F32, value);
  1380. case Type::F64:
  1381. return Inst<U32>(Opcode::ConvertU16F64, value);
  1382. default:
  1383. ThrowInvalidType(value.Type());
  1384. }
  1385. case 32:
  1386. switch (value.Type()) {
  1387. case Type::F16:
  1388. return Inst<U32>(Opcode::ConvertU32F16, value);
  1389. case Type::F32:
  1390. return Inst<U32>(Opcode::ConvertU32F32, value);
  1391. case Type::F64:
  1392. return Inst<U32>(Opcode::ConvertU32F64, value);
  1393. default:
  1394. ThrowInvalidType(value.Type());
  1395. }
  1396. case 64:
  1397. switch (value.Type()) {
  1398. case Type::F16:
  1399. return Inst<U64>(Opcode::ConvertU64F16, value);
  1400. case Type::F32:
  1401. return Inst<U64>(Opcode::ConvertU64F32, value);
  1402. case Type::F64:
  1403. return Inst<U64>(Opcode::ConvertU64F64, value);
  1404. default:
  1405. ThrowInvalidType(value.Type());
  1406. }
  1407. default:
  1408. throw InvalidArgument("Invalid destination bitsize {}", bitsize);
  1409. }
  1410. }
  1411. U32U64 IREmitter::ConvertFToI(size_t bitsize, bool is_signed, const F16F32F64& value) {
  1412. return is_signed ? ConvertFToS(bitsize, value) : ConvertFToU(bitsize, value);
  1413. }
  1414. F16F32F64 IREmitter::ConvertSToF(size_t dest_bitsize, size_t src_bitsize, const Value& value,
  1415. FpControl control) {
  1416. switch (dest_bitsize) {
  1417. case 16:
  1418. switch (src_bitsize) {
  1419. case 8:
  1420. return Inst<F16>(Opcode::ConvertF16S8, Flags{control}, value);
  1421. case 16:
  1422. return Inst<F16>(Opcode::ConvertF16S16, Flags{control}, value);
  1423. case 32:
  1424. return Inst<F16>(Opcode::ConvertF16S32, Flags{control}, value);
  1425. case 64:
  1426. return Inst<F16>(Opcode::ConvertF16S64, Flags{control}, value);
  1427. }
  1428. break;
  1429. case 32:
  1430. switch (src_bitsize) {
  1431. case 8:
  1432. return Inst<F32>(Opcode::ConvertF32S8, Flags{control}, value);
  1433. case 16:
  1434. return Inst<F32>(Opcode::ConvertF32S16, Flags{control}, value);
  1435. case 32:
  1436. return Inst<F32>(Opcode::ConvertF32S32, Flags{control}, value);
  1437. case 64:
  1438. return Inst<F32>(Opcode::ConvertF32S64, Flags{control}, value);
  1439. }
  1440. break;
  1441. case 64:
  1442. switch (src_bitsize) {
  1443. case 8:
  1444. return Inst<F64>(Opcode::ConvertF64S8, Flags{control}, value);
  1445. case 16:
  1446. return Inst<F64>(Opcode::ConvertF64S16, Flags{control}, value);
  1447. case 32:
  1448. return Inst<F64>(Opcode::ConvertF64S32, Flags{control}, value);
  1449. case 64:
  1450. return Inst<F64>(Opcode::ConvertF64S64, Flags{control}, value);
  1451. }
  1452. break;
  1453. }
  1454. throw InvalidArgument("Invalid bit size combination dst={} src={}", dest_bitsize, src_bitsize);
  1455. }
  1456. F16F32F64 IREmitter::ConvertUToF(size_t dest_bitsize, size_t src_bitsize, const Value& value,
  1457. FpControl control) {
  1458. switch (dest_bitsize) {
  1459. case 16:
  1460. switch (src_bitsize) {
  1461. case 8:
  1462. return Inst<F16>(Opcode::ConvertF16U8, Flags{control}, value);
  1463. case 16:
  1464. return Inst<F16>(Opcode::ConvertF16U16, Flags{control}, value);
  1465. case 32:
  1466. return Inst<F16>(Opcode::ConvertF16U32, Flags{control}, value);
  1467. case 64:
  1468. return Inst<F16>(Opcode::ConvertF16U64, Flags{control}, value);
  1469. }
  1470. break;
  1471. case 32:
  1472. switch (src_bitsize) {
  1473. case 8:
  1474. return Inst<F32>(Opcode::ConvertF32U8, Flags{control}, value);
  1475. case 16:
  1476. return Inst<F32>(Opcode::ConvertF32U16, Flags{control}, value);
  1477. case 32:
  1478. return Inst<F32>(Opcode::ConvertF32U32, Flags{control}, value);
  1479. case 64:
  1480. return Inst<F32>(Opcode::ConvertF32U64, Flags{control}, value);
  1481. }
  1482. break;
  1483. case 64:
  1484. switch (src_bitsize) {
  1485. case 8:
  1486. return Inst<F64>(Opcode::ConvertF64U8, Flags{control}, value);
  1487. case 16:
  1488. return Inst<F64>(Opcode::ConvertF64U16, Flags{control}, value);
  1489. case 32:
  1490. return Inst<F64>(Opcode::ConvertF64U32, Flags{control}, value);
  1491. case 64:
  1492. return Inst<F64>(Opcode::ConvertF64U64, Flags{control}, value);
  1493. }
  1494. break;
  1495. }
  1496. throw InvalidArgument("Invalid bit size combination dst={} src={}", dest_bitsize, src_bitsize);
  1497. }
  1498. F16F32F64 IREmitter::ConvertIToF(size_t dest_bitsize, size_t src_bitsize, bool is_signed,
  1499. const Value& value, FpControl control) {
  1500. return is_signed ? ConvertSToF(dest_bitsize, src_bitsize, value, control)
  1501. : ConvertUToF(dest_bitsize, src_bitsize, value, control);
  1502. }
  1503. U32U64 IREmitter::UConvert(size_t result_bitsize, const U32U64& value) {
  1504. switch (result_bitsize) {
  1505. case 32:
  1506. switch (value.Type()) {
  1507. case Type::U32:
  1508. // Nothing to do
  1509. return value;
  1510. case Type::U64:
  1511. return Inst<U32>(Opcode::ConvertU32U64, value);
  1512. default:
  1513. break;
  1514. }
  1515. break;
  1516. case 64:
  1517. switch (value.Type()) {
  1518. case Type::U32:
  1519. return Inst<U64>(Opcode::ConvertU64U32, value);
  1520. case Type::U64:
  1521. // Nothing to do
  1522. return value;
  1523. default:
  1524. break;
  1525. }
  1526. }
  1527. throw NotImplementedException("Conversion from {} to {} bits", value.Type(), result_bitsize);
  1528. }
  1529. F16F32F64 IREmitter::FPConvert(size_t result_bitsize, const F16F32F64& value, FpControl control) {
  1530. switch (result_bitsize) {
  1531. case 16:
  1532. switch (value.Type()) {
  1533. case Type::F16:
  1534. // Nothing to do
  1535. return value;
  1536. case Type::F32:
  1537. return Inst<F16>(Opcode::ConvertF16F32, Flags{control}, value);
  1538. case Type::F64:
  1539. throw LogicError("Illegal conversion from F64 to F16");
  1540. default:
  1541. break;
  1542. }
  1543. break;
  1544. case 32:
  1545. switch (value.Type()) {
  1546. case Type::F16:
  1547. return Inst<F32>(Opcode::ConvertF32F16, Flags{control}, value);
  1548. case Type::F32:
  1549. // Nothing to do
  1550. return value;
  1551. case Type::F64:
  1552. return Inst<F32>(Opcode::ConvertF32F64, Flags{control}, value);
  1553. default:
  1554. break;
  1555. }
  1556. break;
  1557. case 64:
  1558. switch (value.Type()) {
  1559. case Type::F16:
  1560. throw LogicError("Illegal conversion from F16 to F64");
  1561. case Type::F32:
  1562. return Inst<F64>(Opcode::ConvertF64F32, Flags{control}, value);
  1563. case Type::F64:
  1564. // Nothing to do
  1565. return value;
  1566. default:
  1567. break;
  1568. }
  1569. break;
  1570. }
  1571. throw NotImplementedException("Conversion from {} to {} bits", value.Type(), result_bitsize);
  1572. }
  1573. Value IREmitter::ImageSampleImplicitLod(const Value& handle, const Value& coords, const F32& bias,
  1574. const Value& offset, const F32& lod_clamp,
  1575. TextureInstInfo info) {
  1576. const Value bias_lc{MakeLodClampPair(*this, bias, lod_clamp)};
  1577. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageSampleImplicitLod
  1578. : Opcode::BindlessImageSampleImplicitLod};
  1579. return Inst(op, Flags{info}, handle, coords, bias_lc, offset);
  1580. }
  1581. Value IREmitter::ImageSampleExplicitLod(const Value& handle, const Value& coords, const F32& lod,
  1582. const Value& offset, TextureInstInfo info) {
  1583. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageSampleExplicitLod
  1584. : Opcode::BindlessImageSampleExplicitLod};
  1585. return Inst(op, Flags{info}, handle, coords, lod, offset);
  1586. }
  1587. F32 IREmitter::ImageSampleDrefImplicitLod(const Value& handle, const Value& coords, const F32& dref,
  1588. const F32& bias, const Value& offset,
  1589. const F32& lod_clamp, TextureInstInfo info) {
  1590. const Value bias_lc{MakeLodClampPair(*this, bias, lod_clamp)};
  1591. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageSampleDrefImplicitLod
  1592. : Opcode::BindlessImageSampleDrefImplicitLod};
  1593. return Inst<F32>(op, Flags{info}, handle, coords, dref, bias_lc, offset);
  1594. }
  1595. F32 IREmitter::ImageSampleDrefExplicitLod(const Value& handle, const Value& coords, const F32& dref,
  1596. const F32& lod, const Value& offset,
  1597. TextureInstInfo info) {
  1598. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageSampleDrefExplicitLod
  1599. : Opcode::BindlessImageSampleDrefExplicitLod};
  1600. return Inst<F32>(op, Flags{info}, handle, coords, dref, lod, offset);
  1601. }
  1602. Value IREmitter::ImageGather(const Value& handle, const Value& coords, const Value& offset,
  1603. const Value& offset2, TextureInstInfo info) {
  1604. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageGather : Opcode::BindlessImageGather};
  1605. return Inst(op, Flags{info}, handle, coords, offset, offset2);
  1606. }
  1607. Value IREmitter::ImageGatherDref(const Value& handle, const Value& coords, const Value& offset,
  1608. const Value& offset2, const F32& dref, TextureInstInfo info) {
  1609. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageGatherDref
  1610. : Opcode::BindlessImageGatherDref};
  1611. return Inst(op, Flags{info}, handle, coords, offset, offset2, dref);
  1612. }
  1613. Value IREmitter::ImageFetch(const Value& handle, const Value& coords, const Value& offset,
  1614. const U32& lod, const U32& multisampling, TextureInstInfo info) {
  1615. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageFetch : Opcode::BindlessImageFetch};
  1616. return Inst(op, Flags{info}, handle, coords, offset, lod, multisampling);
  1617. }
  1618. Value IREmitter::ImageQueryDimension(const Value& handle, const IR::U32& lod,
  1619. const IR::U1& skip_mips) {
  1620. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageQueryDimensions
  1621. : Opcode::BindlessImageQueryDimensions};
  1622. return Inst(op, handle, lod, skip_mips);
  1623. }
  1624. Value IREmitter::ImageQueryDimension(const Value& handle, const IR::U32& lod,
  1625. const IR::U1& skip_mips, TextureInstInfo info) {
  1626. return Inst(Opcode::ImageQueryDimensions, Flags{info}, handle, lod, skip_mips);
  1627. }
  1628. Value IREmitter::ImageQueryLod(const Value& handle, const Value& coords, TextureInstInfo info) {
  1629. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageQueryLod
  1630. : Opcode::BindlessImageQueryLod};
  1631. return Inst(op, Flags{info}, handle, coords);
  1632. }
  1633. Value IREmitter::ImageGradient(const Value& handle, const Value& coords, const Value& derivates,
  1634. const Value& offset, const F32& lod_clamp, TextureInstInfo info) {
  1635. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageGradient
  1636. : Opcode::BindlessImageGradient};
  1637. return Inst(op, Flags{info}, handle, coords, derivates, offset, lod_clamp);
  1638. }
  1639. Value IREmitter::ImageRead(const Value& handle, const Value& coords, TextureInstInfo info) {
  1640. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageRead : Opcode::BindlessImageRead};
  1641. return Inst(op, Flags{info}, handle, coords);
  1642. }
  1643. void IREmitter::ImageWrite(const Value& handle, const Value& coords, const Value& color,
  1644. TextureInstInfo info) {
  1645. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageWrite : Opcode::BindlessImageWrite};
  1646. Inst(op, Flags{info}, handle, coords, color);
  1647. }
  1648. Value IREmitter::ImageAtomicIAdd(const Value& handle, const Value& coords, const Value& value,
  1649. TextureInstInfo info) {
  1650. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicIAdd32
  1651. : Opcode::BindlessImageAtomicIAdd32};
  1652. return Inst(op, Flags{info}, handle, coords, value);
  1653. }
  1654. Value IREmitter::ImageAtomicSMin(const Value& handle, const Value& coords, const Value& value,
  1655. TextureInstInfo info) {
  1656. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicSMin32
  1657. : Opcode::BindlessImageAtomicSMin32};
  1658. return Inst(op, Flags{info}, handle, coords, value);
  1659. }
  1660. Value IREmitter::ImageAtomicUMin(const Value& handle, const Value& coords, const Value& value,
  1661. TextureInstInfo info) {
  1662. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicUMin32
  1663. : Opcode::BindlessImageAtomicUMin32};
  1664. return Inst(op, Flags{info}, handle, coords, value);
  1665. }
  1666. Value IREmitter::ImageAtomicIMin(const Value& handle, const Value& coords, const Value& value,
  1667. bool is_signed, TextureInstInfo info) {
  1668. return is_signed ? ImageAtomicSMin(handle, coords, value, info)
  1669. : ImageAtomicUMin(handle, coords, value, info);
  1670. }
  1671. Value IREmitter::ImageAtomicSMax(const Value& handle, const Value& coords, const Value& value,
  1672. TextureInstInfo info) {
  1673. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicSMax32
  1674. : Opcode::BindlessImageAtomicSMax32};
  1675. return Inst(op, Flags{info}, handle, coords, value);
  1676. }
  1677. Value IREmitter::ImageAtomicUMax(const Value& handle, const Value& coords, const Value& value,
  1678. TextureInstInfo info) {
  1679. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicUMax32
  1680. : Opcode::BindlessImageAtomicUMax32};
  1681. return Inst(op, Flags{info}, handle, coords, value);
  1682. }
  1683. Value IREmitter::ImageAtomicIMax(const Value& handle, const Value& coords, const Value& value,
  1684. bool is_signed, TextureInstInfo info) {
  1685. return is_signed ? ImageAtomicSMax(handle, coords, value, info)
  1686. : ImageAtomicUMax(handle, coords, value, info);
  1687. }
  1688. Value IREmitter::ImageAtomicInc(const Value& handle, const Value& coords, const Value& value,
  1689. TextureInstInfo info) {
  1690. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicInc32
  1691. : Opcode::BindlessImageAtomicInc32};
  1692. return Inst(op, Flags{info}, handle, coords, value);
  1693. }
  1694. Value IREmitter::ImageAtomicDec(const Value& handle, const Value& coords, const Value& value,
  1695. TextureInstInfo info) {
  1696. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicDec32
  1697. : Opcode::BindlessImageAtomicDec32};
  1698. return Inst(op, Flags{info}, handle, coords, value);
  1699. }
  1700. Value IREmitter::ImageAtomicAnd(const Value& handle, const Value& coords, const Value& value,
  1701. TextureInstInfo info) {
  1702. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicAnd32
  1703. : Opcode::BindlessImageAtomicAnd32};
  1704. return Inst(op, Flags{info}, handle, coords, value);
  1705. }
  1706. Value IREmitter::ImageAtomicOr(const Value& handle, const Value& coords, const Value& value,
  1707. TextureInstInfo info) {
  1708. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicOr32
  1709. : Opcode::BindlessImageAtomicOr32};
  1710. return Inst(op, Flags{info}, handle, coords, value);
  1711. }
  1712. Value IREmitter::ImageAtomicXor(const Value& handle, const Value& coords, const Value& value,
  1713. TextureInstInfo info) {
  1714. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicXor32
  1715. : Opcode::BindlessImageAtomicXor32};
  1716. return Inst(op, Flags{info}, handle, coords, value);
  1717. }
  1718. Value IREmitter::ImageAtomicExchange(const Value& handle, const Value& coords, const Value& value,
  1719. TextureInstInfo info) {
  1720. const Opcode op{handle.IsImmediate() ? Opcode::BoundImageAtomicExchange32
  1721. : Opcode::BindlessImageAtomicExchange32};
  1722. return Inst(op, Flags{info}, handle, coords, value);
  1723. }
  1724. U1 IREmitter::IsTextureScaled(const U32& index) {
  1725. return Inst<U1>(Opcode::IsTextureScaled, index);
  1726. }
  1727. U1 IREmitter::IsImageScaled(const U32& index) {
  1728. return Inst<U1>(Opcode::IsImageScaled, index);
  1729. }
  1730. U1 IREmitter::VoteAll(const U1& value) {
  1731. return Inst<U1>(Opcode::VoteAll, value);
  1732. }
  1733. U1 IREmitter::VoteAny(const U1& value) {
  1734. return Inst<U1>(Opcode::VoteAny, value);
  1735. }
  1736. U1 IREmitter::VoteEqual(const U1& value) {
  1737. return Inst<U1>(Opcode::VoteEqual, value);
  1738. }
  1739. U32 IREmitter::SubgroupBallot(const U1& value) {
  1740. return Inst<U32>(Opcode::SubgroupBallot, value);
  1741. }
  1742. U32 IREmitter::SubgroupEqMask() {
  1743. return Inst<U32>(Opcode::SubgroupEqMask);
  1744. }
  1745. U32 IREmitter::SubgroupLtMask() {
  1746. return Inst<U32>(Opcode::SubgroupLtMask);
  1747. }
  1748. U32 IREmitter::SubgroupLeMask() {
  1749. return Inst<U32>(Opcode::SubgroupLeMask);
  1750. }
  1751. U32 IREmitter::SubgroupGtMask() {
  1752. return Inst<U32>(Opcode::SubgroupGtMask);
  1753. }
  1754. U32 IREmitter::SubgroupGeMask() {
  1755. return Inst<U32>(Opcode::SubgroupGeMask);
  1756. }
  1757. U32 IREmitter::ShuffleIndex(const IR::U32& value, const IR::U32& index, const IR::U32& clamp,
  1758. const IR::U32& seg_mask) {
  1759. return Inst<U32>(Opcode::ShuffleIndex, value, index, clamp, seg_mask);
  1760. }
  1761. U32 IREmitter::ShuffleUp(const IR::U32& value, const IR::U32& index, const IR::U32& clamp,
  1762. const IR::U32& seg_mask) {
  1763. return Inst<U32>(Opcode::ShuffleUp, value, index, clamp, seg_mask);
  1764. }
  1765. U32 IREmitter::ShuffleDown(const IR::U32& value, const IR::U32& index, const IR::U32& clamp,
  1766. const IR::U32& seg_mask) {
  1767. return Inst<U32>(Opcode::ShuffleDown, value, index, clamp, seg_mask);
  1768. }
  1769. U32 IREmitter::ShuffleButterfly(const IR::U32& value, const IR::U32& index, const IR::U32& clamp,
  1770. const IR::U32& seg_mask) {
  1771. return Inst<U32>(Opcode::ShuffleButterfly, value, index, clamp, seg_mask);
  1772. }
  1773. F32 IREmitter::FSwizzleAdd(const F32& a, const F32& b, const U32& swizzle, FpControl control) {
  1774. return Inst<F32>(Opcode::FSwizzleAdd, Flags{control}, a, b, swizzle);
  1775. }
  1776. F32 IREmitter::DPdxFine(const F32& a) {
  1777. return Inst<F32>(Opcode::DPdxFine, a);
  1778. }
  1779. F32 IREmitter::DPdyFine(const F32& a) {
  1780. return Inst<F32>(Opcode::DPdyFine, a);
  1781. }
  1782. F32 IREmitter::DPdxCoarse(const F32& a) {
  1783. return Inst<F32>(Opcode::DPdxCoarse, a);
  1784. }
  1785. F32 IREmitter::DPdyCoarse(const F32& a) {
  1786. return Inst<F32>(Opcode::DPdyCoarse, a);
  1787. }
  1788. } // namespace Shader::IR