ir_emitter.cpp 68 KB

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