ir_emitter.cpp 68 KB

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