1;; Extern type definitions and constructors for the x64 `MachInst` type.
2
3;;;; `MInst` ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
4
5;; Don't build `MInst` variants directly, in general. Instead, use the
6;; instruction-emitting helpers defined further down.
7
8(type MInst nodebug
9      (enum
10       ;; =========================================
11       ;; Integer instructions.
12
13       ;; A synthetic instruction sequence used as part of the lowering of the
14       ;; `srem` instruction which returns 0 if the divisor is -1 and
15       ;; otherwise executes an `idiv` instruction.
16       ;;
17       ;; Note that this does not check for 0 as that's expected to be done
18       ;; separately. Also note that 8-bit types don't use this and use
19       ;; `CheckedSRemSeq8` instead.
20       (CheckedSRemSeq (size OperandSize)
21                       (dividend_lo Gpr)
22                       (dividend_hi Gpr)
23                       (divisor Gpr)
24                       (dst_quotient WritableGpr)
25                       (dst_remainder WritableGpr))
26
27       ;; Same as above but for 8-bit types.
28       (CheckedSRemSeq8 (dividend Gpr)
29                        (divisor Gpr)
30                        (dst WritableGpr))
31
32       ;; Like `MovRR` but with a physical register source (for implementing
33       ;; CLIF instructions like `get_stack_pointer`).
34       (MovFromPReg (src PReg)
35                    (dst WritableGpr))
36
37       ;; Like `MovRR` but with a physical register destination (for
38       ;; implementing CLIF instructions like `set_pinned_reg`).
39       (MovToPReg (src Gpr)
40                  (dst PReg))
41
42       ;; =========================================
43       ;; Conditional moves.
44
45       ;; XMM conditional move; overwrites the destination register.
46       (XmmCmove (ty Type)
47                 (cc CC)
48                 (consequent Xmm)
49                 (alternative Xmm)
50                 (dst WritableXmm))
51
52       ;; =========================================
53       ;; Stack manipulation.
54
55      ;; Emits a inline stack probe loop.
56      (StackProbeLoop (tmp WritableReg)
57                      (frame_size u32)
58                      (guard_size u32))
59
60       ;; =========================================
61       ;; Floating-point operations.
62
63       ;; Converts an unsigned int64 to a float32/float64.
64       (CvtUint64ToFloatSeq (dst_size OperandSize) ;; 4 or 8
65                            (src Gpr)
66                            (dst WritableXmm)
67                            (tmp_gpr1 WritableGpr)
68                            (tmp_gpr2 WritableGpr))
69
70       ;; Converts a scalar xmm to a signed int32/int64.
71       (CvtFloatToSintSeq (dst_size OperandSize)
72                          (src_size OperandSize)
73                          (is_saturating bool)
74                          (src Xmm)
75                          (dst WritableGpr)
76                          (tmp_gpr WritableGpr)
77                          (tmp_xmm WritableXmm))
78
79       ;; Converts a scalar xmm to an unsigned int32/int64.
80       (CvtFloatToUintSeq (dst_size OperandSize)
81                          (src_size OperandSize)
82                          (is_saturating bool)
83                          (src Xmm)
84                          (dst WritableGpr)
85                          (tmp_gpr WritableGpr)
86                          (tmp_xmm WritableXmm)
87                          (tmp_xmm2 WritableXmm))
88
89       ;; A sequence to compute min/max with the proper NaN semantics for xmm
90       ;; registers.
91       (XmmMinMaxSeq (size OperandSize)
92                     (is_min bool)
93                     (lhs Xmm)
94                     (rhs Xmm)
95                     (dst WritableXmm))
96
97       ;; =========================================
98       ;; Control flow instructions.
99
100       ;; Direct call: call simm32.
101       (CallKnown (info BoxCallInfo))
102
103       ;; Indirect call: callq (reg mem)
104       (CallUnknown (info BoxCallIndInfo))
105
106       ;; Tail call to a direct destination.
107       (ReturnCallKnown (info BoxReturnCallInfo))
108
109       ;; Tail call to an indirect destination.
110       (ReturnCallUnknown (info BoxReturnCallIndInfo))
111
112       ;; A pseudo-instruction that captures register arguments in vregs.
113       (Args
114        (args VecArgPair))
115
116       ;; A pseudo-instruction that moves vregs to return registers.
117       (Rets
118        (rets VecRetPair))
119
120       ;; Stack switching
121       (StackSwitchBasic (store_context_ptr Gpr)
122                         (load_context_ptr Gpr)
123                         (in_payload0 Gpr)
124                         (out_payload0 WritableGpr))
125
126       ;; Jump to a known target: jmp simm32.
127       (JmpKnown (dst MachLabel))
128
129       ;; Low-level one-way conditional branch: jcond cond target.
130       ;;
131       ;; This instruction is useful only for lower-level code
132       ;; generators that use the Cranelift instruction backend as an
133       ;; assembler library. The instruction is thus named after its
134       ;; primary user, Winch. This instruction *should not* be used
135       ;; by Cranelift proper and placed into VCode: it does not
136       ;; adhere to the basic-block invariant, namely that branches
137       ;; always end a block (with no fallthrough).
138       (WinchJmpIf (cc CC)
139              (taken MachLabel))
140
141       ;; Two-way conditional branch: jcond cond target target.
142       ;;
143       ;; Emitted as a compound sequence; the MachBuffer will shrink it as
144       ;; appropriate.
145       (JmpCond (cc CC)
146                (taken MachLabel)
147                (not_taken MachLabel))
148
149       ;; Two-way conditional branch with a combination of conditions:
150       ;;
151       ;;   j(cc1 or cc2) target1 target2
152       ;;
153       ;; Emitted as a compound sequence of three branches -- `jcc1
154       ;; target1`, `jcc2 target1`, `jmp target2`.
155       (JmpCondOr (cc1 CC)
156                  (cc2 CC)
157                  (taken MachLabel)
158                  (not_taken MachLabel))
159
160       ;; Jump-table sequence, as one compound instruction (see note in lower.rs
161       ;; for rationale).
162       ;;
163       ;; The generated code sequence is described in the emit's function match
164       ;; arm for this instruction.
165       ;;
166       ;; See comment on jmp_table_seq below about the temporaries signedness.
167       (JmpTableSeq (idx Reg)
168                    (tmp1 WritableReg)
169                    (tmp2 WritableReg)
170                    (default_target MachLabel)
171                    (targets BoxVecMachLabel))
172
173       ;; Traps if the condition code is set.
174       (TrapIf (cc CC)
175               (trap_code TrapCode))
176
177       ;; Traps if both of the condition codes are set.
178       (TrapIfAnd (cc1 CC)
179                  (cc2 CC)
180                  (trap_code TrapCode))
181
182       ;; Traps if either of the condition codes are set.
183       (TrapIfOr (cc1 CC)
184                 (cc2 CC)
185                 (trap_code TrapCode))
186
187       ;; Loads an external symbol in a register, with a relocation:
188       ;;
189       ;; movq $name@GOTPCREL(%rip), dst    if PIC is enabled, or
190       ;; lea $name($rip), dst              if distance is near, or
191       ;; movabsq $name, dst                otherwise.
192       (LoadExtName (dst WritableGpr)
193                    (name BoxExternalName)
194                    (offset i64)
195                    (distance RelocDistance))
196
197       ;; =========================================
198       ;; Instructions pertaining to atomic memory accesses.
199
200       ;; A synthetic instruction, based on a loop around a native `lock
201       ;; cmpxchg` instruction.
202       ;;
203       ;; This atomically modifies a value in memory and returns the old value.
204       ;; The sequence consists of an initial "normal" load from `dst`, followed
205       ;; by a loop which computes the new value and tries to compare-and-swap
206       ;; ("CAS") it into `dst`, using the native instruction `lock
207       ;; cmpxchg{b,w,l,q}`.  The loop iterates until the CAS is successful. If
208       ;; there is no contention, there will be only one pass through the loop
209       ;; body.  The sequence does *not* perform any explicit memory fence
210       ;; instructions (`mfence`/`sfence`/`lfence`).
211       ;;
212       ;; Note that the transaction is atomic in the sense that, as observed by
213       ;; some other thread, `dst` either has the initial or final value, but no
214       ;; other.  It isn't atomic in the sense of guaranteeing that no other
215       ;; thread writes to `dst` in between the initial load and the CAS -- but
216       ;; that would cause the CAS to fail unless the other thread's last write
217       ;; before the CAS wrote the same value that was already there.  In other
218       ;; words, this implementation suffers (unavoidably) from the A-B-A
219       ;; problem.
220       ;;
221       ;; This instruction sequence has fixed register uses as follows:
222       ;; - %rax  (written) the old value at `mem`
223       ;; - %rflags is written.  Do not assume anything about it after the
224       ;;   instruction.
225       (AtomicRmwSeq (ty Type) ;; I8, I16, I32, or I64
226                     (op AtomicRmwSeqOp)
227                     (mem SyntheticAmode)
228                     (operand Gpr)
229                     (temp WritableGpr)
230                     (dst_old WritableGpr))
231
232       ;; A synthetic instruction, based on a loop around a native `lock
233       ;; cmpxchg16b` instruction.
234       ;;
235       ;; This is the same as `AtomicRmwSeq`, but for 128-bit integers.
236       ;;
237       ;; For `AtomicRmwOp::Xchg`, use `Atomic128XchgSeq` instead.
238       ;;
239       ;; This requires that `mem_high` addresses 8 bytes beyond `mem_low`.
240       ;;
241       ;; This instruction sequence has fixed register uses as follows:
242       ;; - %rax (low), %rdx (high)  (written) the old value at `mem_low`
243       ;; - %rbx (low), %rcx (high)  (written) used as temp registers to hold
244       ;;   the replacement value
245       ;; - %rflags is written.  Do not assume anything about it after the
246       ;;   instruction.
247       (Atomic128RmwSeq (args BoxAtomic128RmwSeqArgs))
248
249       ;; A synthetic instruction, based on a loop around a native `lock
250       ;; cmpxchg16b` instruction.
251       ;;
252       ;; This is `Atomic128XchgSeq` but only for `AtomicRmwOp::Xchg`. As the
253       ;; replacement value is the same every time, this instruction doesn't
254       ;; require any temporary registers.
255       ;;
256       ;; This requires that `mem_high` addresses 8 bytes beyond `mem_low`.
257       ;;
258       ;; This instruction sequence has fixed register uses as follows:
259       ;; - %rax (low), %rdx (high)  (written) the old value at `mem_low`
260       ;; - %rbx (low), %rcx (high)  (read) the replacement value
261       ;; - %rflags is written.  Do not assume anything about it after the
262       ;;   instruction.
263       (Atomic128XchgSeq (args BoxAtomic128XchgSeqArgs))
264
265       ;; =========================================
266       ;; Meta-instructions generating no code.
267
268       ;; Provides a way to tell the register allocator that the upcoming
269       ;; sequence of instructions will overwrite `dst` so it should be
270       ;; considered as a `def`; use this with care.
271       ;;
272       ;; This is useful when we have a sequence of instructions whose register
273       ;; usages are nominally `mod`s, but such that the combination of
274       ;; operations creates a result that is independent of the initial
275       ;; register value. It's thus semantically a `def`, not a `mod`, when all
276       ;; the instructions are taken together, so we want to ensure the register
277       ;; is defined (its live-range starts) prior to the sequence to keep
278       ;; analyses happy.
279       ;;
280       ;; One alternative would be a compound instruction that somehow
281       ;; encapsulates the others and reports its own `def`s/`use`s/`mod`s; this
282       ;; adds complexity (the instruction list is no longer flat) and requires
283       ;; knowledge about semantics and initial-value independence anyway.
284       (XmmUninitializedValue (dst WritableXmm))
285
286       ;; See `XmmUninitializedValue` above.
287       (GprUninitializedValue (dst WritableGpr))
288
289       ;; A call to the `ElfTlsGetAddr` libcall. Returns address of TLS symbol
290       ;; `dst`, which is constrained to `rax`.
291       (ElfTlsGetAddr (symbol ExternalName)
292                      (dst WritableGpr))
293
294       ;; A Mach-O TLS symbol access. Returns address of the TLS symbol in
295       ;; `dst`, which is constrained to `rax`.
296       (MachOTlsGetAddr (symbol ExternalName)
297                        (dst WritableGpr))
298
299       ;; A Coff TLS symbol access. Returns address of the TLS symbol in
300       ;; `dst`, which is constrained to `rax`.
301       (CoffTlsGetAddr (symbol ExternalName)
302                       (dst WritableGpr)
303                       (tmp WritableGpr))
304
305       ;; An unwind pseudoinstruction describing the state of the machine at
306       ;; this program point.
307       (Unwind (inst UnwindInst))
308
309       ;; A pseudoinstruction that just keeps a value alive.
310       (DummyUse (reg Reg))
311
312       ;; A pseudoinstruction that loads the address of a label.
313       (LabelAddress (dst WritableGpr)
314                     (label MachLabel))
315
316       ;; A pseudoinstruction that serves as a sequence point.
317       (SequencePoint)
318
319       ;; An instruction assembled outside of cranelift-codegen.
320       (External (inst AssemblerInst))))
321
322(type AssemblerInst extern (enum))
323
324(type OperandSize extern
325      (enum Size8
326            Size16
327            Size32
328            Size64))
329
330(type BoxCallInfo extern (enum))
331(type BoxCallIndInfo extern (enum))
332(type BoxReturnCallInfo extern (enum))
333(type BoxReturnCallIndInfo extern (enum))
334(type BoxAtomic128RmwSeqArgs extern (enum))
335(type BoxAtomic128XchgSeqArgs extern (enum))
336
337;; Get the `OperandSize` for a given `Type`, rounding smaller types up to 32 bits.
338(decl operand_size_of_type_32_64 (Type) OperandSize)
339(extern constructor operand_size_of_type_32_64 operand_size_of_type_32_64)
340
341;; Get the true `OperandSize` for a given `Type`, with no rounding.
342(decl raw_operand_size_of_type (Type) OperandSize)
343(extern constructor raw_operand_size_of_type raw_operand_size_of_type)
344
345(decl atomic128_rmw_seq_args
346  (Atomic128RmwSeqOp SyntheticAmode SyntheticAmode Gpr Gpr WritableGpr WritableGpr WritableGpr WritableGpr)
347  BoxAtomic128RmwSeqArgs)
348(extern constructor atomic128_rmw_seq_args atomic128_rmw_seq_args)
349
350(decl atomic128_xchg_seq_args
351  (SyntheticAmode SyntheticAmode Gpr Gpr WritableGpr WritableGpr)
352  BoxAtomic128XchgSeqArgs)
353(extern constructor atomic128_xchg_seq_args atomic128_xchg_seq_args)
354
355;; Get the bit width of an `OperandSize`.
356(decl operand_size_bits (OperandSize) u16)
357(rule (operand_size_bits (OperandSize.Size8)) 8)
358(rule (operand_size_bits (OperandSize.Size16)) 16)
359(rule (operand_size_bits (OperandSize.Size32)) 32)
360(rule (operand_size_bits (OperandSize.Size64)) 64)
361
362(type RegMemImm extern
363      (enum
364       (Reg (reg Reg))
365       (Mem (addr SyntheticAmode))
366       (Imm (simm32 u32))))
367
368;; Put the given clif value into a `RegMemImm` operand.
369;;
370;; Asserts that the value fits into a single register, and doesn't require
371;; multiple registers for its representation (like `i128` for example).
372;;
373;; As a side effect, this marks the value as used.
374(decl put_in_reg_mem_imm (Value) RegMemImm)
375(extern constructor put_in_reg_mem_imm put_in_reg_mem_imm)
376
377(type RegMem extern
378      (enum
379       (Reg (reg Reg))
380       (Mem (addr SyntheticAmode))))
381
382;; Convert a RegMem to a RegMemImm.
383(decl reg_mem_to_reg_mem_imm (RegMem) RegMemImm)
384(rule (reg_mem_to_reg_mem_imm (RegMem.Reg reg))
385      (RegMemImm.Reg reg))
386(rule (reg_mem_to_reg_mem_imm (RegMem.Mem addr))
387      (RegMemImm.Mem addr))
388
389;; Put the given clif value into a `RegMem` operand.
390;;
391;; Asserts that the value fits into a single register, and doesn't require
392;; multiple registers for its representation (like `i128` for example).
393;;
394;; As a side effect, this marks the value as used.
395(decl put_in_reg_mem (Value) RegMem)
396(extern constructor put_in_reg_mem put_in_reg_mem)
397
398;; Addressing modes.
399
400(type SyntheticAmode extern (enum))
401
402(decl synthetic_amode_to_reg_mem (SyntheticAmode) RegMem)
403(extern constructor synthetic_amode_to_reg_mem synthetic_amode_to_reg_mem)
404
405(spec (amode_to_synthetic_amode amode) (provide (= result amode)))
406(decl amode_to_synthetic_amode (Amode) SyntheticAmode)
407(extern constructor amode_to_synthetic_amode amode_to_synthetic_amode)
408
409(decl synthetic_amode_slot (i32) SyntheticAmode)
410(extern constructor synthetic_amode_slot synthetic_amode_slot)
411
412;; Helper for loads/stores to/from stackslots.
413(decl stackslot_amode (StackSlot Offset32 Offset32) SyntheticAmode)
414(rule (stackslot_amode slot offset1 offset2)
415      (let ((slot_offset i32 (abi_stackslot_offset_into_slot_region slot offset1 offset2)))
416        (synthetic_amode_slot slot_offset)))
417
418;; An `Amode` represents a possible addressing mode that can be used
419;; in instructions. These denote a 64-bit value only.
420(type Amode (enum
421             ;; Immediate sign-extended and a register
422             (ImmReg (simm32 i32)
423                     (base Reg)
424                     (flags MemFlags))
425
426             ;; Sign-extend-32-to-64(simm32) + base + (index << shift)
427             (ImmRegRegShift (simm32 i32)
428                             (base Gpr)
429                             (index Gpr)
430                             (shift u8)
431                             (flags MemFlags))
432
433             ;; Sign-extend-32-to-64(immediate) + RIP (instruction
434             ;; pointer). The appropriate relocation is emitted so
435             ;; that the resulting immediate makes this Amode refer to
436             ;; the given MachLabel.
437             (RipRelative (target MachLabel))))
438
439;; Model an Amode as a combination of flags and the calculated 64-bit address.
440;;   16 bits     64 bits
441;; [ flags  |    address    ]
442(model Amode (type (bv 80)))
443
444(spec (Amode.ImmReg simm base flags)
445      (provide (= result (concat flags (bvadd base (sign_ext 64 simm)))))
446      (require
447        (= (widthof simm) 32)
448        (= (widthof base) 64)
449        (= (widthof flags) 16)))
450
451(spec (Amode.ImmRegRegShift simm base index shift flags)
452  (provide
453    (= result
454      (concat flags
455      (bvadd
456        (bvadd base (sign_ext 64 simm))
457        (bvshl index (zero_ext 64 shift))))))
458  (require
459       (= (widthof simm) 32)
460       (= (widthof base) 64)
461       (= (widthof index) 64)
462       (= (widthof shift) 8)
463       (= (widthof flags) 16)))
464
465;; A helper to both check that the `Imm64` and `Offset32` values sum to less
466;; than 32-bits AND return this summed `u32` value. Also, the `Imm64` will be
467;; zero-extended from `Type` up to 64 bits. This is useful for `to_amode`.
468(decl pure partial sum_extend_fits_in_32_bits (Type Imm64 Offset32) u32)
469(extern constructor sum_extend_fits_in_32_bits sum_extend_fits_in_32_bits)
470
471;;;; Amode lowering ;;;;
472
473;; Converts a `Value` and a static offset into an `Amode` for x64, attempting
474;; to be as fancy as possible with offsets/registers/shifts/etc to make maximal
475;; use of the x64 addressing modes.
476;;
477;; This is a bit subtle unfortunately due to a few constraints. This function
478;; was originally written recursively but that can lead to stack overflow
479;; for certain inputs due to the recursion being defined by user-controlled
480;; input. This means that nowadays this function is not recursive and has a
481;; specific structure to handle that.
482;;
483;; Additionally currently in CLIF all loads/stores have an `Offset32` immediate
484;; to go with them, but the wasm lowering to CLIF doesn't use this meaning that
485;; it's frequently 0. Additionally mid-end optimizations do not fold `iconst`
486;; values into this `Offset32`, meaning that it's left up to backends to hunt
487;; for constants for good codegen. That means that one important aspect of this
488;; function is that it searches for constants to fold into the `Offset32` to
489;; avoid unnecessary instructions.
490;;
491;; Note, though, that the "optimal addressing modes" are only guaranteed to be
492;; generated if egraph-based optimizations have run. For example this will only
493;; attempt to find one constant as opposed to many, and that'll only happen
494;; with constant folding from optimizations.
495;;
496;; Finally there's two primary entry points for this function. One is this
497;; function here, `to_amode,` and another is `to_amode_add`. The latter is used
498;; by the lowering of `iadd` in the x64 backend to use the `lea` instruction
499;; where the input is two `Value` operands instead of just one. Most of the
500;; logic here is then deferred through `to_amode_add`.
501;;
502;; In the future if mid-end optimizations fold constants into `Offset32` then
503;; this in theory can "simply" delegate to the `amode_imm_reg` helper, and
504;; below can delegate to `amode_imm_reg_reg_shift`, or something like that.
505(spec (to_amode flags val offset)
506      (provide (= result (concat flags (bvadd val (sign_ext 64 offset)))))
507      (require
508            (= (widthof val) 64)))
509(decl to_amode (MemFlags Value Offset32) SyntheticAmode)
510(rule 0 (to_amode flags base offset)
511        (amode_imm_reg flags base offset))
512(rule 1 (to_amode flags (iadd _ x y) offset)
513        (to_amode_add flags x y offset))
514
515(rule 2
516      (to_amode flags (stack_addr _ slot offset1) offset2)
517      (stackslot_amode slot offset1 offset2))
518
519;; Same as `to_amode`, except that the base address is computed via the addition
520;; of the two `Value` arguments provided.
521;;
522;; The primary purpose of this is to hunt for constants within the two `Value`
523;; operands provided. Failing that this will defer to `amode_imm_reg` or
524;; `amode_imm_reg_reg_shift` which is the final step in amode lowering and
525;; performs final pattern matches related to shifts to see if that can be
526;; peeled out into the amode.
527;;
528;; In other words this function's job is to find constants and then defer to
529;; `amode_imm_reg*`.
530;;
531(spec (to_amode_add flags x y offset)
532      (provide (= result (concat flags (bvadd (bvadd (sign_ext 64 x) (sign_ext 64 y)) (sign_ext 64 offset))))))
533(instantiate to_amode_add
534    ((args (bv 16) (bv 64) (bv 64) (bv 32)) (ret (bv 80)) (canon (bv 64))))
535(decl to_amode_add (MemFlags Value Value Offset32) Amode)
536
537(rule to_amode_add_base_case 0 (to_amode_add flags x y offset)
538        (amode_imm_reg_reg_shift flags x y offset))
539(rule to_amode_add_const_rhs 1 (to_amode_add flags x (i32_from_iconst c) offset)
540        (if-let sum (i32_checked_add offset c))
541        (amode_imm_reg flags x sum))
542(rule to_amode_add_const_lhs 2 (to_amode_add flags (i32_from_iconst c) x offset)
543        (if-let sum (i32_checked_add offset c))
544        (amode_imm_reg flags x sum))
545(rule to_amode_add_const_fold_iadd_lhs_rhs 3 (to_amode_add flags (iadd _ x (i32_from_iconst c)) y offset)
546        (if-let sum (i32_checked_add offset c))
547        (amode_imm_reg_reg_shift flags x y sum))
548(rule to_amode_add_const_fold_iadd_lhs_lhs 4 (to_amode_add flags (iadd _ (i32_from_iconst c) x) y offset)
549        (if-let sum (i32_checked_add offset c))
550        (amode_imm_reg_reg_shift flags x y sum))
551(rule to_amode_add_const_fold_iadd_rhs_rhs 5 (to_amode_add flags x (iadd _ y (i32_from_iconst c)) offset)
552        (if-let sum (i32_checked_add offset c))
553        (amode_imm_reg_reg_shift flags x y sum))
554(rule to_amode_add_const_fold_iadd_rhs_lhs 6 (to_amode_add flags x (iadd _ (i32_from_iconst c) y) offset)
555        (if-let sum (i32_checked_add offset c))
556        (amode_imm_reg_reg_shift flags x y sum))
557
558;; Final cases of amode lowering. Does not hunt for constants and only attempts
559;; to pattern match add-of-shifts to generate fancier `ImmRegRegShift` modes,
560;; otherwise falls back on `ImmReg`.
561(spec (amode_imm_reg flags x offset)
562      (provide (= result (concat flags (bvadd (sign_ext 64 x) (sign_ext 64 offset))))))
563(instantiate amode_imm_reg
564    ((args (bv 16) (bv 64) (bv 32)) (ret (bv 80)) (canon (bv 64))))
565(decl amode_imm_reg (MemFlags Value Offset32) Amode)
566(rule amode_imm_reg_base 0 (amode_imm_reg flags base offset)
567        (Amode.ImmReg offset base flags))
568(rule amode_imm_reg_iadd 1 (amode_imm_reg flags (iadd _ x y) offset)
569        (amode_imm_reg_reg_shift flags x y offset))
570
571(spec (amode_imm_reg_reg_shift flags x y offset)
572      (provide (= result (concat flags (bvadd (sign_ext 64 (bvadd x y)) (sign_ext 64 offset)))))
573      (require
574            (= (widthof flags) 16)
575            (= (widthof x) (widthof y))
576            (= (widthof offset) 32)))
577(instantiate amode_imm_reg_reg_shift
578    ((args (bv 16) (bv 64) (bv 64) (bv 32)) (ret (bv 80)) (canon (bv 64))))
579(decl amode_imm_reg_reg_shift (MemFlags Value Value Offset32) Amode)
580(rule amode_imm_reg_reg_shift_no_shift 0 (amode_imm_reg_reg_shift flags x y offset)
581        (Amode.ImmRegRegShift offset x y 0 flags)) ;; 0 == y<<0 == "no shift"
582(rule amode_imm_reg_reg_shift_shl_rhs 1 (amode_imm_reg_reg_shift flags x (ishl _ y (iconst _ (uimm8 shift))) offset)
583        (if-let true (u32_lt_eq shift 3))
584        (Amode.ImmRegRegShift offset x y shift flags))
585(rule amode_imm_reg_reg_shift_shl_lhs 2 (amode_imm_reg_reg_shift flags (ishl _ y (iconst _ (uimm8 shift))) x offset)
586        (if-let true (u32_lt_eq shift 3))
587        (Amode.ImmRegRegShift offset x y shift flags))
588
589;; Offsetting an Amode. Used when we need to do consecutive
590;; loads/stores to adjacent addresses.
591;;
592;; Note that the input `Amode` might have a static offset that's too large to
593;; add the specified `i32` to so in the case of overflow a new `Amode` is
594;; generated entirely using `lea`
595(decl amode_offset (SyntheticAmode MemFlags i32) SyntheticAmode)
596(rule 1 (amode_offset base _flags offset)
597  (if-let new (amode_try_offset base offset))
598  new)
599(rule 0 (amode_offset base flags offset)
600  (Amode.ImmReg offset (x64_lea $I64 base) flags))
601
602(decl pure partial amode_try_offset (SyntheticAmode i32) SyntheticAmode)
603(extern constructor amode_try_offset amode_try_offset)
604
605;; Return a zero offset as an `Offset32`.
606(spec (zero_offset) (provide (= result #x00000000)))
607(decl zero_offset () Offset32)
608(extern constructor zero_offset zero_offset)
609
610;; Shift kinds.
611
612(type ShiftKind extern
613      (enum ShiftLeft
614            ShiftRightLogical
615            ShiftRightArithmetic
616            RotateLeft
617            RotateRight))
618
619(type Imm8Gpr
620      (enum (Imm8 (imm u8))
621            (Gpr (reg Gpr))))
622
623;; Put the given clif value into a `Imm8Reg` operand, masked to the bit width of
624;; the given type.
625;;
626;; Asserts that the value fits into a single register, and doesn't require
627;; multiple registers for its representation (like `i128` for example).
628;;
629;; As a side effect, this marks the value as used.
630;;
631;; This is used when lowering various shifts and rotates.
632(decl put_masked_in_imm8_gpr (Value Type) Imm8Gpr)
633(rule 2 (put_masked_in_imm8_gpr (u64_from_iconst amt) ty)
634      (Imm8Gpr.Imm8 (u64_truncate_into_u8 (u64_and amt (shift_mask ty)))))
635(rule 1 (put_masked_in_imm8_gpr amt (fits_in_16 ty))
636      (x64_and $I64 (value_regs_get_gpr amt 0) (RegMemImm.Imm (shift_mask ty))))
637(rule (put_masked_in_imm8_gpr amt ty)
638      (value_regs_get_gpr amt 0))
639
640;; Condition codes
641(type CC extern
642      (enum O
643            NO
644            B
645            NB
646            Z
647            NZ
648            BE
649            NBE
650            S
651            NS
652            L
653            NL
654            LE
655            NLE
656            P
657            NP))
658
659(decl intcc_to_cc (IntCC) CC)
660(extern constructor intcc_to_cc intcc_to_cc)
661
662(decl cc_invert (CC) CC)
663(extern constructor cc_invert cc_invert)
664
665;; Fails if the argument is not either CC.NZ or CC.Z.
666(decl cc_nz_or_z (CC) CC)
667(extern extractor cc_nz_or_z cc_nz_or_z)
668
669(type FcmpImm extern
670      (enum Equal
671            LessThan
672            LessThanOrEqual
673            Unordered
674            NotEqual
675            UnorderedOrGreaterThanOrEqual
676            UnorderedOrGreaterThan
677            Ordered))
678
679(decl encode_fcmp_imm (FcmpImm) u8)
680(extern constructor encode_fcmp_imm encode_fcmp_imm)
681
682(type RoundImm extern
683      (enum RoundNearest
684            RoundDown
685            RoundUp
686            RoundZero))
687
688(decl encode_round_imm (RoundImm) u8)
689(extern constructor encode_round_imm encode_round_imm)
690
691;;;; Newtypes for Different Register Classes ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
692
693(type Gpr (primitive Gpr))
694(type WritableGpr (primitive WritableGpr))
695(type OptionWritableGpr (primitive OptionWritableGpr))
696(type GprMem extern (enum))
697(type GprMemImm extern (enum))
698
699(type Xmm (primitive Xmm))
700(type WritableXmm (primitive WritableXmm))
701(type OptionWritableXmm (primitive OptionWritableXmm))
702(type XmmMem extern (enum))
703(type XmmMemAligned extern (enum))
704(type XmmMemImm extern (enum))
705(type XmmMemAlignedImm extern (enum))
706
707;; Convert a `WritableGpr` to a `WritableReg`.
708(decl writable_gpr_to_reg (WritableGpr) WritableReg)
709(extern constructor writable_gpr_to_reg writable_gpr_to_reg)
710
711;; Convert a `WritableXmm` to a `WritableReg`.
712(decl writable_xmm_to_reg (WritableXmm) WritableReg)
713(extern constructor writable_xmm_to_reg writable_xmm_to_reg)
714
715;; Convert a `WritableReg` to a `WritableXmm`.
716(decl writable_reg_to_xmm (WritableReg) WritableXmm)
717(extern constructor writable_reg_to_xmm writable_reg_to_xmm)
718
719;; Convert a `WritableXmm` to an `Xmm`.
720(decl writable_xmm_to_xmm (WritableXmm) Xmm)
721(extern constructor writable_xmm_to_xmm writable_xmm_to_xmm)
722
723;; Convert a `WritableGpr` to an `Gpr`.
724(decl writable_gpr_to_gpr (WritableGpr) Gpr)
725(extern constructor writable_gpr_to_gpr writable_gpr_to_gpr)
726
727;; Convert an `Gpr` to a `Reg`.
728(decl gpr_to_reg (Gpr) Reg)
729(extern constructor gpr_to_reg gpr_to_reg)
730
731;; Convert an `Gpr` to a `GprMem`.
732(decl gpr_to_gpr_mem (Gpr) GprMem)
733(extern constructor gpr_to_gpr_mem gpr_to_gpr_mem)
734
735;; Convert an `Gpr` to a `GprMemImm`.
736(decl gpr_to_gpr_mem_imm (Gpr) GprMemImm)
737(extern constructor gpr_to_gpr_mem_imm gpr_to_gpr_mem_imm)
738
739;; Convert an `Xmm` to a `Reg`.
740(decl xmm_to_reg (Xmm) Reg)
741(extern constructor xmm_to_reg xmm_to_reg)
742
743;; Convert an `Xmm` into an `XmmMemImm`.
744(decl xmm_to_xmm_mem_imm (Xmm) XmmMemImm)
745(extern constructor xmm_to_xmm_mem_imm xmm_to_xmm_mem_imm)
746
747;; Convert an `XmmMem` into an `XmmMemImm`.
748(decl xmm_mem_to_xmm_mem_imm (XmmMem) XmmMemImm)
749(extern constructor xmm_mem_to_xmm_mem_imm xmm_mem_to_xmm_mem_imm)
750
751;; Convert an `XmmMem` into an `XmmMemAligned`.
752;;
753;; Note that this is an infallible conversion, not a fallible one. If the
754;; original `XmmMem` source is a register, then it's passed through directly.
755;; If it's `Mem` and refers to aligned memory, it's also passed through
756;; directly. Otherwise, though, it's a memory source which is not aligned to
757;; 16 bytes so a load is performed and the temporary register which is the
758;; result of the load is passed through. The end-result is that the return value
759;; here is guaranteed to be a register or an aligned memory location.
760(decl xmm_mem_to_xmm_mem_aligned (XmmMem) XmmMemAligned)
761(extern constructor xmm_mem_to_xmm_mem_aligned xmm_mem_to_xmm_mem_aligned)
762
763;; Convert an `XmmMemImm` into an `XmmMemImmAligned`.
764;;
765;; Note that this is the same as `xmm_mem_to_xmm_mem_aligned` except it handles
766;; an immediate case as well.
767(decl xmm_mem_imm_to_xmm_mem_aligned_imm (XmmMemImm) XmmMemAlignedImm)
768(extern constructor xmm_mem_imm_to_xmm_mem_aligned_imm xmm_mem_imm_to_xmm_mem_aligned_imm)
769
770;; Allocate a new temporary GPR register.
771(decl temp_writable_gpr () WritableGpr)
772(extern constructor temp_writable_gpr temp_writable_gpr)
773
774;; Allocate a new temporary XMM register.
775(decl temp_writable_xmm () WritableXmm)
776(extern constructor temp_writable_xmm temp_writable_xmm)
777
778;; Construct a new `XmmMem` from the given `RegMem`.
779;;
780;; Asserts that the `RegMem`'s register, if any, is an XMM register.
781(decl reg_mem_to_xmm_mem (RegMem) XmmMem)
782(extern constructor reg_mem_to_xmm_mem reg_mem_to_xmm_mem)
783
784;; Construct a new `RegMemImm` from the given `Reg`.
785(decl reg_to_reg_mem_imm (Reg) RegMemImm)
786(extern constructor reg_to_reg_mem_imm reg_to_reg_mem_imm)
787
788;; Construct a new `GprMemImm` from the given `RegMemImm`.
789;;
790;; Asserts that the `RegMemImm`'s register, if any, is an GPR register.
791(decl gpr_mem_imm_new (RegMemImm) GprMemImm)
792(extern constructor gpr_mem_imm_new gpr_mem_imm_new)
793
794;; Construct a new `XmmMemImm` from the given `RegMemImm`.
795;;
796;; Asserts that the `RegMemImm`'s register, if any, is an XMM register.
797(decl xmm_mem_imm_new (RegMemImm) XmmMemImm)
798(extern constructor xmm_mem_imm_new xmm_mem_imm_new)
799
800;; Construct a new `XmmMem` from an `Xmm`.
801(decl xmm_to_xmm_mem (Xmm) XmmMem)
802(extern constructor xmm_to_xmm_mem xmm_to_xmm_mem)
803
804;; Construct a new `XmmMem` from an `RegMem`.
805(decl pure xmm_mem_to_reg_mem (XmmMem) RegMem)
806(extern constructor xmm_mem_to_reg_mem xmm_mem_to_reg_mem)
807
808;; Convert a `GprMem` to a `RegMem`.
809(decl gpr_mem_to_reg_mem (GprMem) RegMem)
810(extern constructor gpr_mem_to_reg_mem gpr_mem_to_reg_mem)
811
812;; Construct a new `Xmm` from a `Reg`.
813;;
814;; Asserts that the register is a XMM.
815(decl xmm_new (Reg) Xmm)
816(extern constructor xmm_new xmm_new)
817
818;; Construct a new `Gpr` from a `Reg`.
819;;
820;; Asserts that the register is a GPR.
821(decl gpr_new (Reg) Gpr)
822(extern constructor gpr_new gpr_new)
823
824;; Construct a new `GprMem` from a `RegMem`.
825;;
826;; Asserts that the `RegMem`'s register, if any, is a GPR.
827(decl reg_mem_to_gpr_mem (RegMem) GprMem)
828(extern constructor reg_mem_to_gpr_mem reg_mem_to_gpr_mem)
829
830;; Construct a `GprMem` from a `Reg`.
831;;
832;; Asserts that the `Reg` is a GPR.
833(decl reg_to_gpr_mem (Reg) GprMem)
834(extern constructor reg_to_gpr_mem reg_to_gpr_mem)
835
836;; Construct a `GprMemImm` from a `Reg`.
837;;
838;; Asserts that the `Reg` is a GPR.
839(decl reg_to_gpr_mem_imm (Reg) GprMemImm)
840(rule (reg_to_gpr_mem_imm r)
841      (gpr_to_gpr_mem_imm (gpr_new r)))
842
843;; Put a value into a GPR.
844;;
845;; Moves the value into a GPR if it is a type that would naturally go into an
846;; XMM register.
847(spec (put_in_gpr arg) (provide (= result (conv_to 64 arg))))
848(decl put_in_gpr (Value) Gpr)
849
850;; Case for when the value naturally lives in a GPR.
851(rule (put_in_gpr val)
852      (if-let (value_type ty) val)
853      (if-let (type_register_class (RegisterClass.Gpr _)) ty)
854      (gpr_new (put_in_reg val)))
855
856;; Case for when the value naturally lives in an XMM register and we must
857;; bitcast it from an XMM into a GPR.
858(rule (put_in_gpr val)
859      (if-let (value_type ty) val)
860      (if-let (type_register_class (RegisterClass.Xmm)) ty)
861      (bitcast_xmm_to_gpr (ty_bits ty) (xmm_new (put_in_reg val))))
862
863;; Put a value into a `GprMem`.
864;;
865;; Asserts that the value goes into a GPR.
866(decl put_in_gpr_mem (Value) GprMem)
867(rule (put_in_gpr_mem val)
868      (reg_mem_to_gpr_mem (put_in_reg_mem val)))
869
870;; Put a value into a `GprMemImm`.
871;;
872;; Asserts that the value goes into a GPR.
873(decl put_in_gpr_mem_imm (Value) GprMemImm)
874(rule (put_in_gpr_mem_imm val)
875      (gpr_mem_imm_new (put_in_reg_mem_imm val)))
876
877;; Put a value into a XMM.
878;;
879;; Asserts that the value goes into a XMM.
880(decl put_in_xmm (Value) Xmm)
881(rule (put_in_xmm val)
882      (xmm_new (put_in_reg val)))
883
884;; Put a value into a `XmmMem`.
885;;
886;; Asserts that the value goes into a XMM.
887(decl put_in_xmm_mem (Value) XmmMem)
888(extern constructor put_in_xmm_mem put_in_xmm_mem)
889
890;; Put a value into a `XmmMemImm`.
891;;
892;; Asserts that the value goes into a XMM.
893(decl put_in_xmm_mem_imm (Value) XmmMemImm)
894(extern constructor put_in_xmm_mem_imm put_in_xmm_mem_imm)
895
896;; Construct an `InstOutput` out of a single GPR register.
897(spec (output_gpr x)
898      (provide (= result (conv_to (widthof result) x))))
899(decl output_gpr (Gpr) InstOutput)
900(rule (output_gpr x)
901      (output_reg (gpr_to_reg x)))
902
903;; Construct a `ValueRegs` out of two GPR registers.
904(decl value_gprs (Gpr Gpr) ValueRegs)
905(rule (value_gprs x y)
906      (value_regs (gpr_to_reg x) (gpr_to_reg y)))
907
908;; Construct an `InstOutput` out of a single XMM register.
909(decl output_xmm (Xmm) InstOutput)
910(rule (output_xmm x)
911      (output_reg (xmm_to_reg x)))
912
913;; Get the `n`th reg in a `ValueRegs` and construct a GPR from it.
914;;
915;; Asserts that the register is a GPR.
916(decl value_regs_get_gpr (ValueRegs usize) Gpr)
917(rule (value_regs_get_gpr regs n)
918      (gpr_new (value_regs_get regs n)))
919
920;; Convert a `Gpr` to an `Imm8Gpr`.
921(decl gpr_to_imm8_gpr (Gpr) Imm8Gpr)
922(rule (gpr_to_imm8_gpr gpr) (Imm8Gpr.Gpr gpr))
923
924;; Get the low half of the given `Value` as a GPR.
925(decl lo_gpr (Value) Gpr)
926(rule (lo_gpr regs) (gpr_new (lo_reg regs)))
927
928;; Construct a new `XmmMemImm` from a 32-bit immediate.
929(decl xmi_imm (u32) XmmMemImm)
930(extern constructor xmi_imm xmi_imm)
931
932;;;; Helpers for determining the register class of a value type ;;;;;;;;;;;;;;;;
933
934(type RegisterClass
935      (enum
936        (Gpr (single_register bool))
937        (Xmm)))
938
939(decl type_register_class (RegisterClass) Type)
940(extern extractor type_register_class type_register_class)
941
942(decl is_xmm_type (Type) Type)
943(extractor (is_xmm_type ty) (and (type_register_class (RegisterClass.Xmm)) ty))
944
945(spec (is_gpr_type arg)
946      (provide (= result arg))
947      (require (<= arg 64)))
948(decl is_gpr_type (Type) Type)
949(extractor (is_gpr_type ty) (and (type_register_class (RegisterClass.Gpr _)) ty))
950
951(decl is_single_register_gpr_type (Type) Type)
952(extractor (is_single_register_gpr_type ty)
953           (and (type_register_class (RegisterClass.Gpr true)) ty))
954
955(decl is_multi_register_gpr_type (Type) Type)
956(extractor (is_multi_register_gpr_type ty)
957           (and (type_register_class (RegisterClass.Gpr false)) ty))
958
959;;;; Helpers for matching operands ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
960;;
961;; These are mainly used for matching operands for the external assembler.
962
963(decl is_imm8 (u8) GprMemImm)
964(extern extractor is_imm8 is_imm8)
965(decl is_imm8_xmm (u8) XmmMemImm)
966(extern extractor is_imm8_xmm is_imm8_xmm)
967(decl is_simm8 (i8) GprMemImm)
968(extern extractor is_simm8 is_simm8)
969(decl is_imm16 (u16) GprMemImm)
970(extern extractor is_imm16 is_imm16)
971(decl is_simm16 (i16) GprMemImm)
972(extern extractor is_simm16 is_simm16)
973(decl is_imm32 (u32) GprMemImm)
974(extern extractor is_imm32 is_imm32)
975(decl is_simm32 (i32) GprMemImm)
976(extern extractor is_simm32 is_simm32)
977(decl is_gpr (Gpr) GprMemImm)
978(extern extractor is_gpr is_gpr)
979(decl is_gpr_mem (GprMem) GprMemImm)
980(extern extractor is_gpr_mem is_gpr_mem)
981(decl is_xmm_mem (XmmMem) XmmMemImm)
982(extern extractor is_xmm_mem is_xmm_mem)
983(decl is_xmm (Xmm) XmmMem)
984(extern extractor is_xmm is_xmm)
985(decl is_mem (SyntheticAmode) XmmMem)
986(extern extractor is_mem is_mem)
987
988;;;; Helpers for Querying Enabled ISA Extensions ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
989
990(decl pure has_avx512vl () bool)
991(extern constructor has_avx512vl has_avx512vl)
992
993(decl pure has_avx512dq () bool)
994(extern constructor has_avx512dq has_avx512dq)
995
996(decl pure has_avx512f () bool)
997(extern constructor has_avx512f has_avx512f)
998
999(decl pure has_avx512bitalg () bool)
1000(extern constructor has_avx512bitalg has_avx512bitalg)
1001
1002(decl pure has_avx512vbmi () bool)
1003(extern constructor has_avx512vbmi has_avx512vbmi)
1004
1005(decl pure has_lzcnt () bool)
1006(extern constructor has_lzcnt has_lzcnt)
1007
1008(decl pure has_bmi1 () bool)
1009(extern constructor has_bmi1 has_bmi1)
1010
1011(decl pure has_bmi2 () bool)
1012(extern constructor has_bmi2 has_bmi2)
1013
1014(decl pure use_popcnt () bool)
1015(extern constructor use_popcnt use_popcnt)
1016
1017(decl pure use_fma () bool)
1018(extern constructor use_fma use_fma)
1019
1020(decl pure has_sse3 () bool)
1021(extern constructor has_sse3 has_sse3)
1022
1023(decl pure has_ssse3 () bool)
1024(extern constructor has_ssse3 has_ssse3)
1025
1026(decl pure has_sse41 () bool)
1027(extern constructor has_sse41 has_sse41)
1028
1029(decl pure use_sse42 () bool)
1030(extern constructor use_sse42 use_sse42)
1031
1032(decl pure has_avx () bool)
1033(extern constructor has_avx has_avx)
1034
1035(decl pure use_avx2 () bool)
1036(extern constructor use_avx2 use_avx2)
1037
1038(decl pure has_cmpxchg16b () bool)
1039(extern constructor has_cmpxchg16b has_cmpxchg16b)
1040
1041;;;; Helpers for Merging and Sinking Immediates/Loads  ;;;;;;;;;;;;;;;;;;;;;;;;;
1042
1043;; Generate a mask for the bit-width of the given type
1044(decl shift_mask (Type) u8)
1045(extern constructor shift_mask shift_mask)
1046
1047;; Mask a constant with the type's shift mask
1048(decl shift_amount_masked (Type Imm64) u8)
1049(extern constructor shift_amount_masked shift_amount_masked)
1050
1051;; Extract a constant `GprMemImm.Imm` from a value operand.
1052(decl simm32_from_value (GprMemImm) Value)
1053(extern extractor simm32_from_value simm32_from_value)
1054
1055;; A load that can be sunk into another operation.
1056(type SinkableLoad extern (enum))
1057
1058;; Extract a `SinkableLoad` that works with `RegMemImm.Mem` from a value
1059;; operand.
1060;;
1061;; Note that this will only work for 32-bit-types-or-larger since this is
1062;; pervasively used with operations that load a minimum of 32-bits. For
1063;; instructions which load exactly the type width necessary use
1064;; `sinkable_load_exact`.
1065(decl sinkable_load (SinkableLoad) Value)
1066(spec (sinkable_load inst)
1067      (provide (= result inst)))
1068(extern extractor sinkable_load sinkable_load)
1069
1070;; Same as `sinkable_load` except that all type widths of loads are supported.
1071;;
1072;; Only use this when the instruction which performs the load is guaranteed to
1073;; load the precisely correct size.
1074(decl sinkable_load_exact (SinkableLoad) Value)
1075(extern extractor sinkable_load_exact sinkable_load_exact)
1076
1077;; Sink a `SinkableLoad` into a `SyntheticAmode`.
1078;;
1079;; This is a side-effectful operation that notifies the context that the
1080;; instruction that produced the `SinkableImm` has been sunk into another
1081;; instruction, and no longer needs to be lowered.
1082(decl sink_load (SinkableLoad) SyntheticAmode)
1083(extern constructor sink_load sink_load)
1084
1085(decl sink_load_to_gpr_mem_imm (SinkableLoad) GprMemImm)
1086(rule (sink_load_to_gpr_mem_imm load)
1087      (gpr_mem_imm_new load))
1088
1089(decl sink_load_to_xmm_mem (SinkableLoad) XmmMem)
1090(rule (sink_load_to_xmm_mem load)
1091      (reg_mem_to_xmm_mem load))
1092
1093(decl sink_load_to_reg_mem (SinkableLoad) RegMem)
1094(rule (sink_load_to_reg_mem load) (RegMem.Mem load))
1095
1096(decl sink_load_to_gpr_mem (SinkableLoad) GprMem)
1097(rule (sink_load_to_gpr_mem load) (RegMem.Mem load))
1098
1099(decl sink_load_to_reg_mem_imm (SinkableLoad) RegMemImm)
1100(spec (sink_load_to_reg_mem_imm load)
1101      (provide (= result load)))
1102(rule (sink_load_to_reg_mem_imm load) (RegMemImm.Mem load))
1103
1104;;;; Helpers for constructing and emitting an `MInst` ;;;;;;;;;;;;;;;;;;;;;;;;;;
1105;;
1106;; These helpers are intended to assist in emitting instructions by taking
1107;; source operands and automatically creating output operands which are then
1108;; returned. These are additionally designed to assist with SSA-like
1109;; construction where the writable version of a register is only present in
1110;; an `MInst` and every other reference to it is a readonly version.
1111
1112;; Helper for creating XmmUninitializedValue instructions.
1113(decl xmm_uninit_value () Xmm)
1114(rule (xmm_uninit_value)
1115      (let ((dst WritableXmm (temp_writable_xmm))
1116            (_ Unit (emit (MInst.XmmUninitializedValue dst))))
1117        dst))
1118
1119;; Helper for creating GprUninitializedValue instructions.
1120(decl gpr_uninit_value () Gpr)
1121(rule (gpr_uninit_value)
1122      (let ((dst WritableGpr (temp_writable_gpr))
1123            (_ Unit (emit (MInst.GprUninitializedValue dst))))
1124        dst))
1125
1126;; Helper for constructing a LoadExtName instruction.
1127(decl load_ext_name (ExternalName i64 RelocDistance) Gpr)
1128(rule (load_ext_name extname offset distance)
1129      (let ((dst WritableGpr (temp_writable_gpr))
1130            (_ Unit (emit (MInst.LoadExtName dst extname offset distance))))
1131        dst))
1132
1133;; Helper for creating `xmm_min_max_seq` pseudo-instructions.
1134(decl xmm_min_max_seq (Type bool Xmm Xmm) Xmm)
1135(rule (xmm_min_max_seq ty is_min lhs rhs)
1136      (let ((dst WritableXmm (temp_writable_xmm))
1137            (size OperandSize (operand_size_of_type_32_64 ty))
1138            (_ Unit (emit (MInst.XmmMinMaxSeq size is_min lhs rhs dst))))
1139        dst))
1140
1141(decl cvt_u64_to_float_seq (Type Gpr) Xmm)
1142(rule (cvt_u64_to_float_seq ty src)
1143      (let ((size OperandSize (raw_operand_size_of_type ty))
1144            (dst WritableXmm (temp_writable_xmm))
1145            (tmp_gpr1 WritableGpr (temp_writable_gpr))
1146            (tmp_gpr2 WritableGpr (temp_writable_gpr))
1147            (_ Unit (emit (MInst.CvtUint64ToFloatSeq size src dst tmp_gpr1 tmp_gpr2))))
1148        dst))
1149
1150(decl cvt_float_to_uint_seq (Type Value bool) Gpr)
1151(rule (cvt_float_to_uint_seq out_ty src @ (value_type src_ty) is_saturating)
1152      (let ((out_size OperandSize (raw_operand_size_of_type out_ty))
1153            (src_size OperandSize (raw_operand_size_of_type src_ty))
1154
1155            (dst WritableGpr (temp_writable_gpr))
1156            (tmp_xmm WritableXmm (temp_writable_xmm))
1157            (tmp_xmm2 WritableXmm (temp_writable_xmm))
1158            (tmp_gpr WritableGpr (temp_writable_gpr))
1159            (_ Unit (emit (MInst.CvtFloatToUintSeq out_size src_size is_saturating src dst tmp_gpr tmp_xmm tmp_xmm2))))
1160        dst))
1161
1162(decl cvt_float_to_sint_seq (Type Value bool) Gpr)
1163(rule (cvt_float_to_sint_seq out_ty src @ (value_type src_ty) is_saturating)
1164      (let ((out_size OperandSize (raw_operand_size_of_type out_ty))
1165            (src_size OperandSize (raw_operand_size_of_type src_ty))
1166
1167            (dst WritableGpr (temp_writable_gpr))
1168            (tmp_xmm WritableXmm (temp_writable_xmm))
1169            (tmp_gpr WritableGpr (temp_writable_gpr))
1170            (_ Unit (emit (MInst.CvtFloatToSintSeq out_size src_size is_saturating src dst tmp_gpr tmp_xmm))))
1171        dst))
1172
1173;; Helper for creating `MovFromPReg` instructions.
1174(decl mov_from_preg (PReg) Reg)
1175(rule (mov_from_preg preg)
1176      (let ((dst WritableGpr (temp_writable_gpr))
1177            (_ Unit (emit (MInst.MovFromPReg preg dst))))
1178        dst))
1179
1180;;;; Helpers for Sign/Zero Extending ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1181
1182(type ExtKind extern
1183      (enum None
1184            SignExtend
1185            ZeroExtend))
1186
1187(type ExtendKind (enum Sign Zero))
1188
1189(model ExtMode (enum
1190      (BL #b000)
1191      (BQ #b001)
1192      (WL #b010)
1193      (WQ #b011)
1194      (LQ #b100)
1195))
1196(type ExtMode extern (enum BL BQ WL WQ LQ))
1197
1198;; `ExtMode::new`
1199
1200(spec (ext_mode x y)
1201      (provide (= result (switch x
1202            (#x0008 (switch y
1203                  (#x0020 (ExtMode.BL))
1204                  (#x0040 (ExtMode.BQ))
1205                  ))
1206            (#x0010 (switch y
1207                  (#x0020 (ExtMode.WL))
1208                  (#x0040 (ExtMode.WQ))
1209                  ))
1210            (#x0020 (switch y
1211                  (#x0040 (ExtMode.LQ))
1212                  ))
1213            ))
1214      )
1215)
1216(decl ext_mode (u16 u16) ExtMode)
1217(extern constructor ext_mode ext_mode)
1218
1219;; Put the given value into a register, but extended as the given type.
1220(decl extend_to_gpr (Value Type ExtendKind) Gpr)
1221
1222;; If the value is already of the requested type, no extending is necessary.
1223(rule 3 (extend_to_gpr val @ (value_type ty) ty _kind)
1224      val)
1225
1226;; I32 -> I64 with op that produces a zero-extended value in a register.
1227;;
1228;; As a particular x64 extra-pattern matching opportunity, all the ALU
1229;; opcodes on 32-bits will zero-extend the upper 32-bits, so we can
1230;; even not generate a zero-extended move in this case.
1231(rule 2 (extend_to_gpr src @ (value_type $I32) $I64 (ExtendKind.Zero))
1232        (if-let true (value32_zeros_upper32 src))
1233        src)
1234
1235;; Both extend instructions are guaranteed to load exactly the source type's size.
1236;; So we can use `sinkable_load_exact` here to sink loads for small types (<= 16 bits).
1237(rule 1 (extend_to_gpr (and (sinkable_load_exact val) (value_type from_ty)) to_ty kind)
1238      (extend_to_gpr_types val from_ty to_ty kind))
1239
1240;; Otherwise emit the extend from a Gpr to a Gpr.
1241(rule (extend_to_gpr (and val (value_type from_ty)) to_ty kind)
1242      (extend_to_gpr_types val from_ty to_ty kind))
1243
1244;; Calculates the correct extension mode for an extend between `from_ty` and `to_ty`.
1245(decl extend_to_gpr_types (GprMem Type Type ExtendKind) Gpr)
1246(rule (extend_to_gpr_types val from_ty to_ty kind)
1247      (let ((from_bits u16 (ty_bits_u16 from_ty))
1248            ;; Use `operand_size_of_type` so that the we clamp the output to 32-
1249            ;; or 64-bit width types.
1250            (to_bits u16 (operand_size_bits (operand_size_of_type_32_64 to_ty))))
1251        (extend kind
1252                to_ty
1253                (ext_mode from_bits to_bits)
1254                val)))
1255
1256
1257;; Do a sign or zero extension of the given `GprMem`.
1258(decl extend (ExtendKind Type ExtMode GprMem) Gpr)
1259
1260;; Zero extending uses `movzx`.
1261(rule (extend (ExtendKind.Zero) ty mode src)
1262      (x64_movzx mode src))
1263
1264;; Sign extending uses `movsx`.
1265(rule (extend (ExtendKind.Sign) ty mode src)
1266      (x64_movsx mode src))
1267
1268;; Tests whether the operation used to produce the input `Value`, which must
1269;; be a 32-bit operation, will automatically zero the upper 32-bits of the
1270;; destination register that `Value` is placed in.
1271(decl pure value32_zeros_upper32 (Value) bool)
1272(rule (value32_zeros_upper32 (iadd _ _ _)) true)
1273(rule (value32_zeros_upper32 (isub _ _ _)) true)
1274(rule (value32_zeros_upper32 (imul _ _ _)) true)
1275(rule (value32_zeros_upper32 (band _ _ _)) true)
1276(rule (value32_zeros_upper32 (bor _ _ _)) true)
1277(rule (value32_zeros_upper32 (bxor _ _ _)) true)
1278(rule (value32_zeros_upper32 (ishl _ _ _)) true)
1279(rule (value32_zeros_upper32 (ushr _ _ _)) true)
1280(rule (value32_zeros_upper32 (uload32 _ _ _ _)) true)
1281(rule -1 (value32_zeros_upper32 _) false)
1282
1283;;;; Helpers for Working SSE tidbits ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1284
1285;; Turn a vector type into its integer-typed vector equivalent.
1286(decl vec_int_type (Type) Type)
1287(rule (vec_int_type (multi_lane 8 16)) $I8X16)
1288(rule (vec_int_type (multi_lane 16 8)) $I16X8)
1289(rule (vec_int_type (multi_lane 32 4)) $I32X4)
1290(rule (vec_int_type (multi_lane 64 2)) $I64X2)
1291
1292;; Performs an xor operation of the two operands specified.
1293(decl x64_xor_vector (Type Xmm XmmMem) Xmm)
1294(rule 1 (x64_xor_vector $F16 x y) (x64_xorps x y))
1295(rule 1 (x64_xor_vector $F32 x y) (x64_xorps x y))
1296(rule 1 (x64_xor_vector $F64 x y) (x64_xorpd x y))
1297(rule 1 (x64_xor_vector $F128 x y) (x64_xorps x y))
1298(rule 1 (x64_xor_vector $F32X4 x y) (x64_xorps x y))
1299(rule 1 (x64_xor_vector $F64X2 x y) (x64_xorpd x y))
1300(rule 0 (x64_xor_vector (multi_lane _ _) x y) (x64_pxor x y))
1301
1302;; Generates a register value which has an all-ones pattern.
1303;;
1304;; Note that this is accomplished by comparing a fresh register with itself,
1305;; which for integers is always true. Also note that the comparison is always
1306;; done for integers. This is because we're comparing a fresh register to itself
1307;; and we don't know the previous contents of the register. If a floating-point
1308;; comparison is used then it runs the risk of comparing NaN against NaN and not
1309;; actually producing an all-ones mask. By using integer comparison operations
1310;; we're guaranteeed that everything is equal to itself.
1311(decl vector_all_ones () Xmm)
1312(rule (vector_all_ones)
1313      (let ((tmp Xmm (xmm_uninit_value)))
1314        (x64_pcmpeqd tmp tmp)))
1315
1316;; Move a `RegMemImm.Reg` operand to an XMM register, if necessary.
1317(decl mov_rmi_to_xmm (RegMemImm) XmmMemImm)
1318(rule (mov_rmi_to_xmm rmi @ (RegMemImm.Mem _)) (xmm_mem_imm_new rmi))
1319(rule (mov_rmi_to_xmm rmi @ (RegMemImm.Imm _)) (xmm_mem_imm_new rmi))
1320(rule (mov_rmi_to_xmm (RegMemImm.Reg r)) (x64_movd_to_xmm r))
1321
1322;;;; Helpers for Emitting Calls ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1323
1324(decl gen_call_info (Sig ExternalName CallArgList CallRetList OptionTryCallInfo bool) BoxCallInfo)
1325(extern constructor gen_call_info gen_call_info)
1326
1327(decl gen_call_ind_info (Sig RegMem CallArgList CallRetList OptionTryCallInfo) BoxCallIndInfo)
1328(extern constructor gen_call_ind_info gen_call_ind_info)
1329
1330(decl gen_return_call_info (Sig ExternalName CallArgList) BoxReturnCallInfo)
1331(extern constructor gen_return_call_info gen_return_call_info)
1332
1333(decl gen_return_call_ind_info (Sig Reg CallArgList) BoxReturnCallIndInfo)
1334(extern constructor gen_return_call_ind_info gen_return_call_ind_info)
1335
1336;; Helper for creating `CallKnown` instructions.
1337(decl call_known (BoxCallInfo) SideEffectNoResult)
1338(rule (call_known info)
1339      (SideEffectNoResult.Inst (MInst.CallKnown info)))
1340
1341;; Helper for creating `CallUnknown` instructions.
1342(decl call_unknown (BoxCallIndInfo) SideEffectNoResult)
1343(rule (call_unknown info)
1344      (SideEffectNoResult.Inst (MInst.CallUnknown info)))
1345
1346;; Helper for creating `ReturnCallKnown` instructions.
1347(decl return_call_known (BoxReturnCallInfo) SideEffectNoResult)
1348(rule (return_call_known info)
1349      (SideEffectNoResult.Inst (MInst.ReturnCallKnown info)))
1350
1351;; Helper for creating `ReturnCallUnknown` instructions.
1352(decl return_call_unknown (BoxReturnCallIndInfo) SideEffectNoResult)
1353(rule (return_call_unknown info)
1354      (SideEffectNoResult.Inst (MInst.ReturnCallUnknown info)))
1355
1356;;;; Helpers for emitting stack switches ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1357
1358(decl x64_stack_switch_basic (Gpr Gpr Gpr) Gpr)
1359(rule (x64_stack_switch_basic store_context_ptr load_context_ptr in_payload0)
1360      (let ((out_payload0 WritableGpr (temp_writable_gpr))
1361            (_ Unit (emit (MInst.StackSwitchBasic store_context_ptr
1362                                                  load_context_ptr
1363                                                  in_payload0
1364                                                  out_payload0))))
1365        out_payload0))
1366
1367;;;; Helpers for Emitting Loads ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1368
1369;; Load a value into an XMM register.
1370(decl x64_load_xmm (Type SyntheticAmode) Xmm)
1371(rule 1 (x64_load_xmm $F32 addr) (x64_movss_load addr))
1372(rule 1 (x64_load_xmm $F64 addr) (x64_movsd_load addr))
1373(rule 1 (x64_load_xmm $F128 addr) (x64_movdqu_load addr))
1374(rule 1 (x64_load_xmm $F32X4 addr) (x64_movups_load addr))
1375(rule 1 (x64_load_xmm $F64X2 addr) (x64_movupd_load addr))
1376(rule 0 (x64_load_xmm (multi_lane _bits _lanes) addr) (x64_movdqu_load addr))
1377
1378(decl x64_mov (SyntheticAmode) Reg)
1379(spec (x64_mov addr)
1380      (provide (= result (conv_to 64 (load_effect (extract 79 64 addr) 64 (extract 63 0 addr))))))
1381(rule (x64_mov addr) (x64_movq_rm addr))
1382
1383(decl x64_movzx (ExtMode GprMem) Gpr)
1384(spec (x64_movzx mode src)
1385      (provide
1386            (= result
1387                  (conv_to
1388                        64
1389                        (zero_ext
1390                              32
1391                              (load_effect
1392                                    (extract 79 64 src)
1393                                    (switch mode
1394                                          ((ExtMode.BL) 8)
1395                                          ((ExtMode.BQ) 8)
1396                                          ((ExtMode.WL) 16)
1397                                          ((ExtMode.WQ) 16)
1398                                          ((ExtMode.LQ) 32))
1399                                    (extract 63 0 src))))))
1400      (require (or (= mode (ExtMode.BL))
1401                  (= mode (ExtMode.BQ))
1402                  (= mode (ExtMode.WL))
1403                  (= mode (ExtMode.WQ))
1404                  (= mode (ExtMode.LQ)))))
1405(rule (x64_movzx (ExtMode.BL) src) (x64_movzbl_rm src))
1406(rule (x64_movzx (ExtMode.BQ) src) (x64_movzbq_rm src))
1407(rule (x64_movzx (ExtMode.WL) src) (x64_movzwl_rm src))
1408(rule (x64_movzx (ExtMode.WQ) src) (x64_movzwq_rm src))
1409;; This instruction selection may seem strange but is correct in 64-bit mode:
1410;; section 3.4.1.1 of the Intel manual says that "32-bit operands generate a
1411;; 32-bit result, zero-extended to a 64-bit result in the destination
1412;; general-purpose register." This is applicable beyond `mov` but we use this
1413;; fact to zero-extend `src` into `dst`.
1414(rule (x64_movzx (ExtMode.LQ) src) (x64_movl_rm src))
1415
1416(decl x64_movsx (ExtMode GprMem) Gpr)
1417(rule (x64_movsx (ExtMode.BL) src) (x64_movsbl_rm src))
1418(rule (x64_movsx (ExtMode.BQ) src) (x64_movsbq_rm src))
1419(rule (x64_movsx (ExtMode.WL) src) (x64_movswl_rm src))
1420(rule (x64_movsx (ExtMode.WQ) src) (x64_movswq_rm src))
1421(rule (x64_movsx (ExtMode.LQ) src) (x64_movslq_rm src))
1422
1423(decl x64_movss_load (SyntheticAmode) Xmm)
1424(rule (x64_movss_load from) (x64_movss_a_m_or_avx from))
1425
1426(decl x64_movss_store (SyntheticAmode Xmm) SideEffectNoResult)
1427(rule (x64_movss_store addr data) (x64_movss_c_m_mem_or_avx addr data))
1428
1429(decl x64_movsd_load (SyntheticAmode) Xmm)
1430(rule (x64_movsd_load from) (x64_movsd_a_m_or_avx from))
1431
1432(decl x64_movsd_store (SyntheticAmode Xmm) SideEffectNoResult)
1433(rule (x64_movsd_store addr data) (x64_movsd_c_m_mem_or_avx addr data))
1434
1435(decl x64_movups_load (SyntheticAmode) Xmm)
1436(rule (x64_movups_load from) (x64_movups_a_or_avx from))
1437
1438(decl x64_movups_store (SyntheticAmode Xmm) SideEffectNoResult)
1439(rule (x64_movups_store addr data) (x64_movups_b_mem_or_avx addr data))
1440
1441(decl x64_movupd_load (SyntheticAmode) Xmm)
1442(rule (x64_movupd_load from) (x64_movupd_a_or_avx from))
1443
1444(decl x64_movupd_store (SyntheticAmode Xmm) SideEffectNoResult)
1445(rule (x64_movupd_store addr data) (x64_movupd_b_mem_or_avx addr data))
1446
1447;; Helper for creating `movd` instructions.
1448(decl x64_movd_to_gpr (Xmm) Gpr)
1449(rule (x64_movd_to_gpr from) (x64_movd_b from))
1450(rule 1 (x64_movd_to_gpr from)
1451        (if-let true (has_avx))
1452        (x64_vmovd_b from))
1453
1454;; Helper for creating `movd` instructions.
1455(decl x64_movd_to_xmm (GprMem) Xmm)
1456(rule (x64_movd_to_xmm from) (x64_movd_a from))
1457(rule 1 (x64_movd_to_xmm from)
1458        (if-let true (has_avx))
1459        (x64_vmovd_a from))
1460
1461;; Helper for creating `movq` instructions.
1462(decl x64_movq_to_xmm (GprMem) Xmm)
1463(rule (x64_movq_to_xmm src) (x64_movq_a src))
1464(rule 1 (x64_movq_to_xmm from)
1465        (if-let true (has_avx))
1466        (x64_vmovq_a from))
1467
1468;; Helper for creating `movq` instructions.
1469(decl x64_movq_to_gpr (Xmm) Gpr)
1470(rule (x64_movq_to_gpr src) (x64_movq_b src))
1471(rule 1 (x64_movq_to_gpr from)
1472        (if-let true (has_avx))
1473        (x64_vmovq_b from))
1474
1475(decl x64_movdqu_load (XmmMem) Xmm)
1476(rule (x64_movdqu_load from) (x64_movdqu_a_or_avx from))
1477
1478(decl x64_movdqu_store (SyntheticAmode Xmm) SideEffectNoResult)
1479(rule (x64_movdqu_store addr data) (x64_movdqu_b_mem_or_avx addr data))
1480
1481(decl x64_pmovsxbw (XmmMem) Xmm)
1482(rule (x64_pmovsxbw from) (x64_pmovsxbw_a_or_avx from))
1483
1484(decl x64_pmovzxbw (XmmMem) Xmm)
1485(rule (x64_pmovzxbw from) (x64_pmovzxbw_a_or_avx from))
1486
1487(decl x64_pmovsxwd (XmmMem) Xmm)
1488(rule (x64_pmovsxwd from) (x64_pmovsxwd_a_or_avx from))
1489
1490(decl x64_pmovzxwd (XmmMem) Xmm)
1491(rule (x64_pmovzxwd from) (x64_pmovzxwd_a_or_avx from))
1492
1493(decl x64_pmovsxdq (XmmMem) Xmm)
1494(rule (x64_pmovsxdq from) (x64_pmovsxdq_a_or_avx from))
1495
1496(decl x64_pmovzxdq (XmmMem) Xmm)
1497(rule (x64_pmovzxdq from) (x64_pmovzxdq_a_or_avx from))
1498
1499(decl x64_movrm (Type SyntheticAmode Gpr) SideEffectNoResult)
1500(spec (x64_movrm ty addr data)
1501       (provide (= result (store_effect (extract 79 64 addr) ty (conv_to ty data) (extract 63 0 addr)))))
1502(rule (x64_movrm $I8 addr data) (x64_movb_mr_mem addr data))
1503(rule (x64_movrm $I16 addr data) (x64_movw_mr_mem addr data))
1504(rule (x64_movrm $I32 addr data) (x64_movl_mr_mem addr data))
1505(rule (x64_movrm $I64 addr data) (x64_movq_mr_mem addr data))
1506
1507(decl x64_movimm_m (Type SyntheticAmode i32) SideEffectNoResult)
1508(rule (x64_movimm_m $I8 addr (i8_from_i32 imm)) (x64_movb_mi_mem addr (i8_cast_unsigned imm)))
1509(rule (x64_movimm_m $I16 addr (i16_from_i32 imm)) (x64_movw_mi_mem addr (i16_cast_unsigned imm)))
1510(rule (x64_movimm_m $I32 addr imm) (x64_movl_mi_mem addr (i32_cast_unsigned imm)))
1511(rule (x64_movimm_m $I64 addr imm) (x64_movq_mi_sxl_mem addr imm))
1512
1513;; Load a constant into an XMM register.
1514(decl x64_xmm_load_const (Type VCodeConstant) Xmm)
1515(rule (x64_xmm_load_const ty const)
1516      (x64_load_xmm ty (const_to_synthetic_amode const)))
1517
1518
1519;;;; Flag Helpers ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1520;;
1521;; These helpers are used to emit instructions that produce or consume flags.
1522;; The operations used here are by no means the only ones possible; they are
1523;; simply the ones currently used in Cranelift's lowerings.
1524
1525;; Some operations produce flags.
1526(type ProduceFlagsOp (enum (Add) (Sub)))
1527
1528(decl x64_produce_flags (ProduceFlagsOp Type Gpr GprMemImm) ProducesFlags)
1529(rule (x64_produce_flags (ProduceFlagsOp.Add) ty src1 src2)
1530      (x64_add_with_flags_paired ty src1 src2))
1531(rule (x64_produce_flags (ProduceFlagsOp.Sub) ty src1 src2)
1532      (x64_sub_with_flags_paired ty src1 src2))
1533
1534;; This should only be use for instructions that _do_ produce flags that can be
1535;; consumed later. It is semantically "unsafe" and must be used correctly.
1536(decl asm_produce_flags (AssemblerOutputs) ProducesFlags)
1537(rule (asm_produce_flags (AssemblerOutputs.RetGpr inst gpr))
1538      (ProducesFlags.ProducesFlagsReturnsResultWithConsumer inst gpr))
1539(rule (asm_produce_flags (AssemblerOutputs.RetValueRegs inst regs))
1540      (ProducesFlags.ProducesFlagsReturnsResultWithConsumer inst (value_regs_get_gpr regs 0)))
1541
1542;; Other operations consume _and_ produce flags--"chaining".
1543(type ChainFlagsOp (enum (Adc) (Sbb)))
1544
1545(decl x64_chain_flags (ChainFlagsOp Type Gpr Gpr) ConsumesAndProducesFlags)
1546(rule (x64_chain_flags (ChainFlagsOp.Adc) ty src1 src2)
1547      (x64_adc_chained ty src1 src2))
1548(rule (x64_chain_flags (ChainFlagsOp.Sbb) ty src1 src2)
1549      (x64_sbb_chained ty src1 src2))
1550
1551(decl asm_chain_flags (AssemblerOutputs) ConsumesAndProducesFlags)
1552(rule (asm_chain_flags (AssemblerOutputs.RetGpr inst gpr))
1553      (ConsumesAndProducesFlags.ReturnsReg inst gpr))
1554
1555;; Still others produce flags a part of a side-effect operation.
1556
1557(type ProduceFlagsSideEffectOp (enum (Or) (Sbb)))
1558
1559(decl x64_produce_flags_side_effect (ProduceFlagsSideEffectOp Type Gpr GprMemImm) ProducesFlags)
1560(rule (x64_produce_flags_side_effect (ProduceFlagsSideEffectOp.Or) (fits_in_64 ty) src1 src2)
1561      (x64_or_with_flags_paired_side_effect ty src1 src2))
1562(rule (x64_produce_flags_side_effect (ProduceFlagsSideEffectOp.Sbb) (fits_in_64 ty) src1 src2)
1563      (x64_sbb_paired_side_effect ty src1 src2))
1564
1565(decl asm_produce_flags_side_effect (AssemblerOutputs) ProducesFlags)
1566(rule (asm_produce_flags_side_effect (AssemblerOutputs.RetGpr inst gpr))
1567      (ProducesFlags.ProducesFlagsSideEffect inst))
1568(rule (asm_produce_flags_side_effect (AssemblerOutputs.SideEffect inst))
1569      (ProducesFlags.ProducesFlagsSideEffect inst))
1570
1571;; Other helpers for instruction emission.
1572
1573(decl asm_consume_flags (AssemblerOutputs) ConsumesFlags)
1574(rule (asm_consume_flags (AssemblerOutputs.RetGpr inst gpr))
1575      (ConsumesFlags.ConsumesFlagsReturnsResultWithProducer inst gpr))
1576
1577(decl asm_consumes_flags_returns_gpr (AssemblerOutputs) ConsumesFlags)
1578(rule (asm_consumes_flags_returns_gpr (AssemblerOutputs.RetGpr inst gpr))
1579      (ConsumesFlags.ConsumesFlagsReturnsReg inst gpr))
1580
1581
1582
1583;;;; Instruction Constructors ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
1584;;
1585;; These constructors create SSA-style `MInst`s. It is their responsibility to
1586;; maintain the invariant that each temporary register they allocate and define
1587;; only gets defined the once.
1588
1589;; Helper for creating raw `add` instructions.
1590(decl x64_add_raw (Type Gpr GprMemImm) AssemblerOutputs)
1591
1592;; Match 8-bit immediates first; allows a smaller instruction encoding.
1593(rule 2 (x64_add_raw $I32 src1 (is_simm8 src2))   (x64_addl_mi_sxb_raw src1 src2))
1594(rule 2 (x64_add_raw $I64 src1 (is_simm8 src2))   (x64_addq_mi_sxb_raw src1 src2))
1595
1596;; Match the remaining immediates.
1597(rule 1 (x64_add_raw $I8  src1 (is_imm8 src2))    (x64_addb_mi_raw src1 src2))
1598(rule 1 (x64_add_raw $I16 src1 (is_imm16 src2))   (x64_addw_mi_raw src1 src2))
1599(rule 1 (x64_add_raw $I32 src1 (is_imm32 src2))   (x64_addl_mi_raw src1 src2))
1600(rule 1 (x64_add_raw $I64 src1 (is_simm32 src2))  (x64_addq_mi_sxl_raw src1 src2))
1601
1602;; Match the operand size to the instruction width.
1603(rule 0 (x64_add_raw $I8  src1 (is_gpr_mem src2)) (x64_addb_rm_raw src1 src2))
1604(rule 0 (x64_add_raw $I16 src1 (is_gpr_mem src2)) (x64_addw_rm_raw src1 src2))
1605(rule 0 (x64_add_raw $I32 src1 (is_gpr_mem src2)) (x64_addl_rm_raw src1 src2))
1606(rule 0 (x64_add_raw $I64 src1 (is_gpr_mem src2)) (x64_addq_rm_raw src1 src2))
1607
1608;; When the overflow flag is not considered, we can use wider instructions than
1609;; necessary for 8/16-bit register-to-register operations to avoid CPU false
1610;; dependencies.
1611(decl x64_add_break_deps (Type Gpr GprMemImm) AssemblerOutputs)
1612(rule 1 (x64_add_break_deps $I8  src1 (is_gpr src2)) (x64_addl_rm_raw src1 src2))
1613(rule 1 (x64_add_break_deps $I16 src1 (is_gpr src2)) (x64_addl_rm_raw src1 src2))
1614(rule 0 (x64_add_break_deps ty   src1 src2)          (x64_add_raw ty src1 src2))
1615
1616;; Normal use of `add` returns a `Gpr` register.
1617(decl x64_add (Type Gpr GprMemImm) Gpr)
1618(rule (x64_add ty src1 src2)
1619      (emit_ret_gpr (x64_add_break_deps ty src1 src2)))
1620
1621;; When using `add` for its overflow flag (OF), we track that the flags are
1622;; changed (and avoid the "dependency-breaking" rules that short-circuit
1623;; overflow).
1624(decl x64_add_with_flags_paired (Type Gpr GprMemImm) ProducesFlags)
1625(rule (x64_add_with_flags_paired ty src1 src2)
1626      (asm_produce_flags (x64_add_raw ty src1 src2)))
1627
1628
1629
1630;; Helper for creating raw `adc` instructions; Cranelift only uses the 64-bit
1631;; variant of this instruction. As with `add`, we match 8-bit immediates first;
1632;; this allows a smaller instruction encoding.
1633(decl x64_adc_raw (Type Gpr GprMemImm) AssemblerOutputs)
1634(rule 2 (x64_adc_raw $I64 src1 (is_simm8 src2))   (x64_adcq_mi_sxb_raw src1 src2))
1635(rule 1 (x64_adc_raw $I64 src1 (is_simm32 src2))  (x64_adcq_mi_sxl_raw src1 src2))
1636(rule 0 (x64_adc_raw $I64 src1 (is_gpr_mem src2)) (x64_adcq_rm_raw src1 src2))
1637
1638;; Normal use of the `adc` instruction consumes a previously-produced flag.
1639(decl x64_adc_paired (Type Gpr GprMemImm) ConsumesFlags)
1640(rule (x64_adc_paired ty src1 src2)
1641      (asm_consume_flags (x64_adc_raw ty src1 src2)))
1642
1643;; We also use `adc` to modify flags that are used later.
1644(decl x64_adc_chained (Type Gpr GprMemImm) ConsumesAndProducesFlags)
1645(rule (x64_adc_chained ty src1 src2)
1646      (asm_chain_flags (x64_adc_raw ty src1 src2)))
1647
1648
1649
1650;; Helper for emitting raw `sub` instructions.
1651(decl x64_sub_raw (Type Gpr GprMemImm) AssemblerOutputs)
1652
1653;; Match 8-bit immediates first; allows a smaller instruction encoding.
1654(rule 3 (x64_sub_raw $I32 src1 (is_simm8 src2))   (x64_subl_mi_sxb_raw src1 src2))
1655(rule 3 (x64_sub_raw $I64 src1 (is_simm8 src2))   (x64_subq_mi_sxb_raw src1 src2))
1656
1657;; Match the remaining immediates.
1658(rule 2 (x64_sub_raw $I8  src1 (is_imm8 src2))    (x64_subb_mi_raw src1 src2))
1659(rule 2 (x64_sub_raw $I16 src1 (is_imm16 src2))   (x64_subw_mi_raw src1 src2))
1660(rule 2 (x64_sub_raw $I32 src1 (is_imm32 src2))   (x64_subl_mi_raw src1 src2))
1661(rule 2 (x64_sub_raw $I64 src1 (is_simm32 src2))  (x64_subq_mi_sxl_raw src1 src2))
1662
1663;; Match the operand size to the instruction width.
1664(rule 0 (x64_sub_raw $I8  src1 (is_gpr_mem src2)) (x64_subb_rm_raw src1 src2))
1665(rule 0 (x64_sub_raw $I16 src1 (is_gpr_mem src2)) (x64_subw_rm_raw src1 src2))
1666(rule 0 (x64_sub_raw $I32 src1 (is_gpr_mem src2)) (x64_subl_rm_raw src1 src2))
1667(rule 0 (x64_sub_raw $I64 src1 (is_gpr_mem src2)) (x64_subq_rm_raw src1 src2))
1668
1669;; When the overflow flag is not considered, we can use wider instructions than
1670;; necessary for 8/16-bit register-to-register operations to avoid CPU false
1671;; dependencies.
1672(decl x64_sub_break_deps (Type Gpr GprMemImm) AssemblerOutputs)
1673(rule 1 (x64_sub_break_deps $I8  src1 (is_gpr src2)) (x64_subl_rm_raw src1 src2))
1674(rule 1 (x64_sub_break_deps $I16 src1 (is_gpr src2)) (x64_subl_rm_raw src1 src2))
1675(rule 0 (x64_sub_break_deps ty   src1 src2)          (x64_sub_raw ty src1 src2))
1676
1677;; Normal use of `sub` returns a `Gpr` register.
1678(decl x64_sub (Type Gpr GprMemImm) Gpr)
1679(rule (x64_sub ty src1 src2)
1680      (emit_ret_gpr (x64_sub_break_deps ty src1 src2)))
1681
1682;; When using `sub` for its flags (OF, CF, SF), we track that the flags are
1683;; changed.
1684(decl x64_sub_with_flags_paired (Type Gpr GprMemImm) ProducesFlags)
1685(rule (x64_sub_with_flags_paired ty src1 src2)
1686      (asm_produce_flags (x64_sub_raw ty src1 src2)))
1687
1688
1689
1690;; Helper for creating raw `sbb` instructions; Cranelift only uses the 64-bit
1691;; variant of this instruction.
1692(decl x64_sbb_raw (Type Gpr GprMemImm) AssemblerOutputs)
1693
1694;; Match 8-bit immediates first; allows a smaller instruction encoding.
1695(rule 2 (x64_sbb_raw $I32 src1 (is_simm8 src2))   (x64_sbbl_mi_sxb_raw src1 src2))
1696(rule 2 (x64_sbb_raw $I64 src1 (is_simm8 src2))   (x64_sbbq_mi_sxb_raw src1 src2))
1697
1698;; Match the remaining immediates.
1699(rule 1 (x64_sbb_raw $I8  src1 (is_imm8 src2))    (x64_sbbb_mi_raw src1 src2))
1700(rule 1 (x64_sbb_raw $I16 src1 (is_imm16 src2))   (x64_sbbw_mi_raw src1 src2))
1701(rule 1 (x64_sbb_raw $I32 src1 (is_imm32 src2))   (x64_sbbl_mi_raw src1 src2))
1702(rule 1 (x64_sbb_raw $I64 src1 (is_simm32 src2))  (x64_sbbq_mi_sxl_raw src1 src2))
1703
1704;; Match the operand size to the instruction width.
1705(rule 0 (x64_sbb_raw $I8  src1 (is_gpr_mem src2)) (x64_sbbb_rm_raw src1 src2))
1706(rule 0 (x64_sbb_raw $I16 src1 (is_gpr_mem src2)) (x64_sbbw_rm_raw src1 src2))
1707(rule 0 (x64_sbb_raw $I32 src1 (is_gpr_mem src2)) (x64_sbbl_rm_raw src1 src2))
1708(rule 0 (x64_sbb_raw $I64 src1 (is_gpr_mem src2)) (x64_sbbq_rm_raw src1 src2))
1709
1710;; When the overflow flag is not considered, we can use wider instructions than
1711;; necessary for 8/16-bit register-to-register operations to avoid CPU false
1712;; dependencies.
1713(decl x64_sbb_break_deps (Type Gpr GprMemImm) AssemblerOutputs)
1714(rule 1 (x64_sbb_break_deps $I8  src1 (is_gpr src2)) (x64_sbbl_rm_raw src1 src2))
1715(rule 1 (x64_sbb_break_deps $I16 src1 (is_gpr src2)) (x64_sbbl_rm_raw src1 src2))
1716(rule 0 (x64_sbb_break_deps ty   src1 src2)          (x64_sbb_raw ty src1 src2))
1717
1718;; Normal use of the `sbb` instruction consumes previously-produced flags (OF,
1719;; CF, SF).
1720(decl x64_sbb_paired (Type Gpr GprMemImm) ConsumesFlags)
1721(rule (x64_sbb_paired ty src1 src2)
1722      (asm_consume_flags (x64_sbb_break_deps ty src1 src2)))
1723
1724;; We also use `sbb` to modify flags that all later used.
1725(decl x64_sbb_chained (Type Gpr GprMemImm) ConsumesAndProducesFlags)
1726(rule (x64_sbb_chained ty src1 src2)
1727      (asm_chain_flags (x64_sbb_raw ty src1 src2)))
1728
1729;; We also use `sbb` in side-effecting operations.
1730(decl x64_sbb_paired_side_effect (Type Gpr GprMemImm) ProducesFlags)
1731(rule (x64_sbb_paired_side_effect  ty src1 src2)
1732      (asm_produce_flags_side_effect (x64_sbb_raw ty src1 src2)))
1733
1734
1735
1736;; Helper for creating `mul` instructions or `imul` instructions (depending
1737;; on `signed`). For the 8-bit rules, see `x64_mul8`.
1738(decl x64_mul_raw (Type bool Gpr GprMem) AssemblerOutputs)
1739(rule (x64_mul_raw $I16 false src1 src2) (x64_mulw_m_raw src1 src2))
1740(rule (x64_mul_raw $I32 false src1 src2) (x64_mull_m_raw src1 src2))
1741(rule (x64_mul_raw $I64 false src1 src2) (x64_mulq_m_raw src1 src2))
1742(rule (x64_mul_raw $I16 true src1 src2)  (x64_imulw_m_raw src1 src2))
1743(rule (x64_mul_raw $I32 true src1 src2)  (x64_imull_m_raw src1 src2))
1744(rule (x64_mul_raw $I64 true src1 src2)  (x64_imulq_m_raw src1 src2))
1745
1746(decl x64_mul (Type bool Gpr GprMem) ValueRegs)
1747(rule 0 (x64_mul ty signed src1 src2)
1748      (emit_ret_value_regs (x64_mul_raw ty signed src1 src2)))
1749
1750;; Special case the `mulx` pattern with the BMI2 instruction set.
1751;;
1752;; Note that mulx returns the high bits in the first result and the low bits in
1753;; the second result, so here the result registers are swapped to match
1754;; `x64_mul` above.
1755(rule 1 (x64_mul $I32 false src1 src2)
1756  (if-let true (has_bmi2))
1757  (let ((regs ValueRegs (x64_mulxl_rvm src2 src1)))
1758    (value_regs (value_regs_get regs 1) (value_regs_get regs 0))))
1759(rule 1 (x64_mul $I64 false src1 src2)
1760  (if-let true (has_bmi2))
1761  (let ((regs ValueRegs (x64_mulxq_rvm src2 src1)))
1762    (value_regs (value_regs_get regs 1) (value_regs_get regs 0))))
1763
1764(decl x64_mulx_hi (Type Gpr GprMem) Gpr)
1765(rule (x64_mulx_hi $I32 src1 src2) (x64_mulxl_rvm_hi src2 src1))
1766(rule (x64_mulx_hi $I64 src1 src2) (x64_mulxq_rvm_hi src2 src1))
1767
1768(decl x64_mulxl_rvm_hi (GprMem Gpr) Gpr)
1769(extern constructor x64_mulxl_rvm_hi x64_mulxl_rvm_hi)
1770(decl x64_mulxq_rvm_hi (GprMem Gpr) Gpr)
1771(extern constructor x64_mulxq_rvm_hi x64_mulxq_rvm_hi)
1772
1773(decl x64_mul_lo_with_flags_paired (Type bool Gpr GprMem) ProducesFlags)
1774(rule (x64_mul_lo_with_flags_paired ty signed src1 src2)
1775      (asm_produce_flags (x64_mul_raw ty signed src1 src2)))
1776
1777;; Get the invalid register as writable
1778(decl writable_invalid_gpr () WritableGpr)
1779(extern constructor writable_invalid_gpr writable_invalid_gpr)
1780
1781;; Helper for creating `imul` instructions.
1782(decl x64_imul (Type Gpr GprMem) Gpr)
1783(rule (x64_imul $I16 src1 src2) (x64_imulw_rm src1 src2))
1784(rule (x64_imul $I32 src1 src2) (x64_imull_rm src1 src2))
1785(rule (x64_imul $I64 src1 src2) (x64_imulq_rm src1 src2))
1786
1787;; Helper for creating `imul` instructions with an immediate operand. Match
1788;; 8-bit immediates first to allow a smaller instruction encoding.
1789(decl x64_imul_imm (Type GprMem i32) Gpr)
1790(rule 2 (x64_imul_imm $I16 src1 (i8_from_i32 src2))  (x64_imulw_rmi_sxb src1 src2))
1791(rule 2 (x64_imul_imm $I32 src1 (i8_from_i32 src2))  (x64_imull_rmi_sxb src1 src2))
1792(rule 2 (x64_imul_imm $I64 src1 (i8_from_i32 src2))  (x64_imulq_rmi_sxb src1 src2))
1793(rule 1 (x64_imul_imm $I16 src1 (i16_from_i32 src2)) (x64_imulw_rmi src1 (i16_cast_unsigned src2)))
1794(rule 1 (x64_imul_imm $I32 src1 src2) (x64_imull_rmi src1 (i32_cast_unsigned src2)))
1795(rule 1 (x64_imul_imm $I64 src1 src2) (x64_imulq_rmi_sxl src1 src2))
1796
1797;; Helper for creating `mul` instructions or `imul` instructions (depending
1798;; on `signed`) for 8-bit operands.
1799(decl x64_mul8_raw (bool Gpr GprMem) AssemblerOutputs)
1800(rule (x64_mul8_raw false src1 src2) (x64_mulb_m_raw src1 src2))
1801(rule (x64_mul8_raw true src1 src2)  (x64_imulb_m_raw src1 src2))
1802
1803(decl x64_mul8 (bool Gpr GprMem) Gpr)
1804(rule (x64_mul8 signed src1 src2)
1805      (emit_ret_gpr (x64_mul8_raw signed src1 src2)))
1806
1807(decl x64_mul8_with_flags_paired (bool Gpr GprMem) ProducesFlags)
1808(rule (x64_mul8_with_flags_paired signed src1 src2)
1809      (asm_produce_flags (x64_mul8_raw signed src1 src2)))
1810
1811
1812
1813;; Helper for emitting `and` instructions.
1814(decl x64_and (Type Gpr GprMemImm) Gpr)
1815
1816;; Match 8-bit immediates first; allows a smaller instruction encoding.
1817(rule 3 (x64_and $I32 src1 (is_simm8 src2))   (x64_andl_mi_sxb src1 src2))
1818(rule 3 (x64_and $I64 src1 (is_simm8 src2))   (x64_andq_mi_sxb src1 src2))
1819
1820;; Match the remaining immediates.
1821(rule 2 (x64_and $I8  src1 (is_imm8 src2))    (x64_andb_mi src1 src2))
1822(rule 2 (x64_and $I16 src1 (is_imm16 src2))   (x64_andw_mi src1 src2))
1823(rule 2 (x64_and $I32 src1 (is_imm32 src2))   (x64_andl_mi src1 src2))
1824(rule 2 (x64_and $I64 src1 (is_simm32 src2))  (x64_andq_mi_sxl src1 src2))
1825
1826;; Use wider instructions than necessary for 8/16-bit register-to-register
1827;; operations to avoid CPU false dependencies.
1828(rule 1 (x64_and $I8  src1 (is_gpr src2))     (x64_andl_rm src1 src2))
1829(rule 1 (x64_and $I16 src1 (is_gpr src2))     (x64_andl_rm src1 src2))
1830
1831;; Match the operand size to the instruction width.
1832(rule 0 (x64_and $I8  src1 (is_gpr_mem src2)) (x64_andb_rm src1 src2))
1833(rule 0 (x64_and $I16 src1 (is_gpr_mem src2)) (x64_andw_rm src1 src2))
1834(rule 0 (x64_and $I32 src1 (is_gpr_mem src2)) (x64_andl_rm src1 src2))
1835(rule 0 (x64_and $I64 src1 (is_gpr_mem src2)) (x64_andq_rm src1 src2))
1836
1837
1838
1839;; Helper for emitting raw `or` instructions.
1840(decl x64_or_raw (Type Gpr GprMemImm) AssemblerOutputs)
1841
1842;; Match 8-bit immediates first; allows a smaller instruction encoding.
1843(rule 2 (x64_or_raw $I32 src1 (is_simm8 src2))   (x64_orl_mi_sxb_raw src1 src2))
1844(rule 2 (x64_or_raw $I64 src1 (is_simm8 src2))   (x64_orq_mi_sxb_raw src1 src2))
1845
1846;; Match the remaining immediates.
1847(rule 1 (x64_or_raw $I8  src1 (is_imm8 src2))    (x64_orb_mi_raw src1 src2))
1848(rule 1 (x64_or_raw $I16 src1 (is_imm16 src2))   (x64_orw_mi_raw src1 src2))
1849(rule 1 (x64_or_raw $I32 src1 (is_imm32 src2))   (x64_orl_mi_raw src1 src2))
1850(rule 1 (x64_or_raw $I64 src1 (is_simm32 src2))  (x64_orq_mi_sxl_raw src1 src2))
1851
1852;; Match the operand size to the instruction width.
1853(rule 0 (x64_or_raw $I8  src1 (is_gpr_mem src2)) (x64_orb_rm_raw src1 src2))
1854(rule 0 (x64_or_raw $I16 src1 (is_gpr_mem src2)) (x64_orw_rm_raw src1 src2))
1855(rule 0 (x64_or_raw $I32 src1 (is_gpr_mem src2)) (x64_orl_rm_raw src1 src2))
1856(rule 0 (x64_or_raw $I64 src1 (is_gpr_mem src2)) (x64_orq_rm_raw src1 src2))
1857
1858;; When flags are not considered, we can use wider instructions than necessary
1859;; for 8/16-bit register-to-register operations to avoid CPU false dependencies.
1860(decl x64_or_break_deps (Type Gpr GprMemImm) AssemblerOutputs)
1861(rule 1 (x64_or_break_deps $I8  src1 (is_gpr src2)) (x64_orl_rm_raw src1 src2))
1862(rule 1 (x64_or_break_deps $I16 src1 (is_gpr src2)) (x64_orl_rm_raw src1 src2))
1863(rule 0 (x64_or_break_deps ty   src1 src2)          (x64_or_raw ty src1 src2))
1864
1865;; Normal use of `or` returns a `Gpr` register.
1866(decl x64_or (Type Gpr GprMemImm) Gpr)
1867(rule (x64_or ty src1 src2)
1868      (emit_ret_gpr (x64_or_break_deps ty src1 src2)))
1869
1870;; When using `or` for its flags (SF, ZF, PF), we track that the flags are
1871;; changed. Note t
1872(decl x64_or_with_flags_paired_side_effect (Type Gpr GprMemImm) ProducesFlags)
1873(rule (x64_or_with_flags_paired_side_effect ty src1 src2)
1874      (asm_produce_flags_side_effect (x64_or_raw ty src1 src2)))
1875
1876
1877
1878;; Helper for emitting `xor` instructions.
1879(decl x64_xor (Type Gpr GprMemImm) Gpr)
1880
1881;; Match 8-bit immediates first; allows a smaller instruction encoding.
1882(rule 3 (x64_xor $I32 src1 (is_simm8 src2))   (x64_xorl_mi_sxb src1 src2))
1883(rule 3 (x64_xor $I64 src1 (is_simm8 src2))   (x64_xorq_mi_sxb src1 src2))
1884
1885;; Match the remaining immediates.
1886(rule 2 (x64_xor $I8  src1 (is_imm8 src2))    (x64_xorb_mi src1 src2))
1887(rule 2 (x64_xor $I16 src1 (is_imm16 src2))   (x64_xorw_mi src1 src2))
1888(rule 2 (x64_xor $I32 src1 (is_imm32 src2))   (x64_xorl_mi src1 src2))
1889(rule 2 (x64_xor $I64 src1 (is_simm32 src2))  (x64_xorq_mi_sxl src1 src2))
1890
1891;; Use wider instructions than necessary for 8/16-bit register-to-register
1892;; operations to avoid CPU false dependencies.
1893(rule 1 (x64_xor $I8  src1 (is_gpr src2))     (x64_xorl_rm src1 src2))
1894(rule 1 (x64_xor $I16 src1 (is_gpr src2))     (x64_xorl_rm src1 src2))
1895
1896;; Match the operand size to the instruction width.
1897(rule 0 (x64_xor $I8  src1 (is_gpr_mem src2)) (x64_xorb_rm src1 src2))
1898(rule 0 (x64_xor $I16 src1 (is_gpr_mem src2)) (x64_xorw_rm src1 src2))
1899(rule 0 (x64_xor $I32 src1 (is_gpr_mem src2)) (x64_xorl_rm src1 src2))
1900(rule 0 (x64_xor $I64 src1 (is_gpr_mem src2)) (x64_xorq_rm src1 src2))
1901
1902;; Helper for `andn` instructions
1903;;
1904;; Note that 8/16-bit versions of these instructions do not exist, so for
1905;; those bit-widths the 32-bit version of the instruction is used which has the
1906;; desired semantics for the lower bits of the register.
1907(decl x64_andn (Type Gpr GprMem) Gpr)
1908(rule (x64_andn $I8 src1 src2) (x64_andnl_rvm src1 src2))
1909(rule (x64_andn $I16 src1 src2) (x64_andnl_rvm src1 src2))
1910(rule (x64_andn $I32 src1 src2) (x64_andnl_rvm src1 src2))
1911(rule (x64_andn $I64 src1 src2) (x64_andnq_rvm src1 src2))
1912
1913;; Helper for emitting immediates with an `i64` value. Note that
1914;; integer constants in ISLE are always parsed as `i128`s; this enables
1915;; negative numbers to be used as immediates.
1916(decl imm_i64 (Type i64) Reg)
1917(rule (imm_i64 ty value)
1918      (imm ty (i64_cast_unsigned value)))
1919
1920;; Helper for emitting immediates.
1921;;
1922;; Note that if `Type` is less than 64-bits then the upper bits of the `imm`
1923;; argument will be set to zero and lost.
1924;;
1925;; Recursion: at most once to implement floats with integer bit patterns.
1926(decl rec imm (Type u64) Reg)
1927
1928;; Base case: integers of up to at most 32-bits.
1929;;
1930;; FIXME: the immediate argument to this constructor is `u64` but it's logically
1931;; interpreted as the bit pattern for a signed 32-bit immediate. That means
1932;; that ideally this would convert the immediate to a 64-bit signed immediate,
1933;; fallibly convert that to a signed 32-bit integer, and then convert that to
1934;; unsigned to pass to the raw instruction. In doing so there would be a
1935;; guarantee that the value in the register is the same logical value as the
1936;; immediate passed to this constructor. This is not possible today though
1937;; because literals like `0x8000_0000_u64` don't convert to `i32`.
1938(rule 0 (imm (fits_in_32 (ty_int ty)) (u32_from_u64 imm)) (x64_movl_oi imm))
1939
1940;; Base cases for other types
1941(rule 1 (imm $I64 imm) (x64_movabsq_oi imm))
1942(rule 1 (imm $F16 (u64_extract_non_zero bits)) (bitcast_gpr_to_xmm 16 (imm $I16 bits)))
1943(rule 1 (imm $F32 (u64_extract_non_zero bits)) (x64_movd_to_xmm (imm $I32 bits)))
1944(rule 1 (imm $F64 (u64_extract_non_zero bits)) (x64_movq_to_xmm (imm $I64 bits)))
1945
1946;; Special case: a 64-bit immediate which sign extends from a 32-bit immediate.
1947(rule 2 (imm $I64 imm)
1948  (if-let imm32 (i64_try_into_i32 (u64_cast_signed imm)))
1949  (x64_movq_mi_sxl imm32))
1950
1951;; Special case: a 64-bit immediate which zero extends from a 32-bit immediate.
1952;;
1953;; Note that `movl` here will zero-extend the destination register in 64-bit
1954;; mode which is the zero-extension we want.
1955(rule 3 (imm $I64 (u32_from_u64 imm32))
1956  (x64_movl_oi imm32))
1957
1958;; Special case the 0 immediate:
1959(rule 4 (imm (fits_in_64 (ty_int ty)) 0)
1960      (let ((tmp Gpr (gpr_uninit_value)))
1961        (x64_xor ty tmp tmp)))
1962(rule 5 (imm ty @ (multi_lane _bits _lanes) 0) (xmm_to_reg (xmm_zero ty)))
1963(rule 6 (imm ty @ $F16 0) (xmm_zero ty))
1964(rule 6 (imm ty @ $F32 0) (xmm_zero ty))
1965(rule 6 (imm ty @ $F64 0) (xmm_zero ty))
1966
1967;; TODO: use cmpeqp{s,d} for all 1s float immediates
1968
1969(decl xmm_zero (Type) Xmm)
1970(rule (xmm_zero ty)
1971      (let ((tmp Xmm (xmm_uninit_value)))
1972        (x64_xor_vector ty tmp tmp)))
1973
1974;; Helper for creating `rotl` instructions.
1975(decl x64_rotl (Type Gpr Imm8Gpr) Gpr)
1976(rule (x64_rotl $I8 src1 (Imm8Gpr.Gpr src2)) (x64_rolb_mc src1 src2))
1977(rule (x64_rotl $I8 src1 (Imm8Gpr.Imm8 src2)) (x64_rolb_mi src1 src2))
1978(rule (x64_rotl $I16 src1 (Imm8Gpr.Gpr src2)) (x64_rolw_mc src1 src2))
1979(rule (x64_rotl $I16 src1 (Imm8Gpr.Imm8 src2)) (x64_rolw_mi src1 src2))
1980(rule (x64_rotl $I32 src1 (Imm8Gpr.Gpr src2)) (x64_roll_mc src1 src2))
1981(rule (x64_rotl $I32 src1 (Imm8Gpr.Imm8 src2)) (x64_roll_mi src1 src2))
1982(rule (x64_rotl $I64 src1 (Imm8Gpr.Gpr src2)) (x64_rolq_mc src1 src2))
1983(rule (x64_rotl $I64 src1 (Imm8Gpr.Imm8 src2)) (x64_rolq_mi src1 src2))
1984(rule 1 (x64_rotl $I8 src1 (Imm8Gpr.Imm8 1)) (x64_rolb_m1 src1))
1985(rule 1 (x64_rotl $I16 src1 (Imm8Gpr.Imm8 1)) (x64_rolw_m1 src1))
1986(rule 1 (x64_rotl $I32 src1 (Imm8Gpr.Imm8 1)) (x64_roll_m1 src1))
1987(rule 1 (x64_rotl $I64 src1 (Imm8Gpr.Imm8 1)) (x64_rolq_m1 src1))
1988(rule 2 (x64_rotl (ty_32_or_64 ty) src (Imm8Gpr.Imm8 imm))
1989        (if-let true (has_bmi2))
1990        (x64_rorx ty src (u8_wrapping_sub (ty_bits ty) imm)))
1991
1992;; Helper for creating `rotr` instructions.
1993(decl x64_rotr (Type Gpr Imm8Gpr) Gpr)
1994(rule (x64_rotr $I8 src1 (Imm8Gpr.Gpr src2)) (x64_rorb_mc src1 src2))
1995(rule (x64_rotr $I8 src1 (Imm8Gpr.Imm8 src2)) (x64_rorb_mi src1 src2))
1996(rule (x64_rotr $I16 src1 (Imm8Gpr.Gpr src2)) (x64_rorw_mc src1 src2))
1997(rule (x64_rotr $I16 src1 (Imm8Gpr.Imm8 src2)) (x64_rorw_mi src1 src2))
1998(rule (x64_rotr $I32 src1 (Imm8Gpr.Gpr src2)) (x64_rorl_mc src1 src2))
1999(rule (x64_rotr $I32 src1 (Imm8Gpr.Imm8 src2)) (x64_rorl_mi src1 src2))
2000(rule (x64_rotr $I64 src1 (Imm8Gpr.Gpr src2)) (x64_rorq_mc src1 src2))
2001(rule (x64_rotr $I64 src1 (Imm8Gpr.Imm8 src2)) (x64_rorq_mi src1 src2))
2002(rule 1 (x64_rotr $I8 src1 (Imm8Gpr.Imm8 1)) (x64_rorb_m1 src1))
2003(rule 1 (x64_rotr $I16 src1 (Imm8Gpr.Imm8 1)) (x64_rorw_m1 src1))
2004(rule 1 (x64_rotr $I32 src1 (Imm8Gpr.Imm8 1)) (x64_rorl_m1 src1))
2005(rule 1 (x64_rotr $I64 src1 (Imm8Gpr.Imm8 1)) (x64_rorq_m1 src1))
2006(rule 2 (x64_rotr (ty_32_or_64 ty) src (Imm8Gpr.Imm8 imm))
2007        (if-let true (has_bmi2))
2008        (x64_rorx ty src imm))
2009
2010;; Helper for creating `shl` instructions.
2011(decl x64_shl (Type Gpr Imm8Gpr) Gpr)
2012(rule (x64_shl $I8 src1 (Imm8Gpr.Gpr src2)) (x64_shlb_mc src1 src2))
2013(rule (x64_shl $I8 src1 (Imm8Gpr.Imm8 src2)) (x64_shlb_mi src1 src2))
2014(rule (x64_shl $I16 src1 (Imm8Gpr.Gpr src2)) (x64_shlw_mc src1 src2))
2015(rule (x64_shl $I16 src1 (Imm8Gpr.Imm8 src2)) (x64_shlw_mi src1 src2))
2016(rule (x64_shl $I32 src1 (Imm8Gpr.Gpr src2)) (x64_shll_mc src1 src2))
2017(rule (x64_shl $I32 src1 (Imm8Gpr.Imm8 src2)) (x64_shll_mi src1 src2))
2018(rule (x64_shl $I64 src1 (Imm8Gpr.Gpr src2)) (x64_shlq_mc src1 src2))
2019(rule (x64_shl $I64 src1 (Imm8Gpr.Imm8 src2)) (x64_shlq_mi src1 src2))
2020(rule 1 (x64_shl $I8 src1 (Imm8Gpr.Imm8 1)) (x64_shlb_m1 src1))
2021(rule 1 (x64_shl $I16 src1 (Imm8Gpr.Imm8 1)) (x64_shlw_m1 src1))
2022(rule 1 (x64_shl $I32 src1 (Imm8Gpr.Imm8 1)) (x64_shll_m1 src1))
2023(rule 1 (x64_shl $I64 src1 (Imm8Gpr.Imm8 1)) (x64_shlq_m1 src1))
2024;; With BMI2 the `shlx` instruction is also available, and it's unconditionally
2025;; used for registers shifted by registers since it provides more freedom
2026;; in regalloc since nothing is constrained. Note that the `shlx` instruction
2027;; doesn't encode an immediate so any immediate-based shift still uses `shl`.
2028(rule 1 (x64_shl (ty_32_or_64 ty) src1 (Imm8Gpr.Gpr src2))
2029        (if-let true (has_bmi2))
2030        (x64_shlx ty src1 src2))
2031
2032;; Helper for creating logical shift-right instructions.
2033(decl x64_shr (Type Gpr Imm8Gpr) Gpr)
2034(rule (x64_shr $I8 src1 (Imm8Gpr.Gpr src2)) (x64_shrb_mc src1 src2))
2035(rule (x64_shr $I8 src1 (Imm8Gpr.Imm8 src2)) (x64_shrb_mi src1 src2))
2036(rule (x64_shr $I16 src1 (Imm8Gpr.Gpr src2)) (x64_shrw_mc src1 src2))
2037(rule (x64_shr $I16 src1 (Imm8Gpr.Imm8 src2)) (x64_shrw_mi src1 src2))
2038(rule (x64_shr $I32 src1 (Imm8Gpr.Gpr src2)) (x64_shrl_mc src1 src2))
2039(rule (x64_shr $I32 src1 (Imm8Gpr.Imm8 src2)) (x64_shrl_mi src1 src2))
2040(rule (x64_shr $I64 src1 (Imm8Gpr.Gpr src2)) (x64_shrq_mc src1 src2))
2041(rule (x64_shr $I64 src1 (Imm8Gpr.Imm8 src2)) (x64_shrq_mi src1 src2))
2042(rule 1 (x64_shr $I8 src1 (Imm8Gpr.Imm8 1)) (x64_shrb_m1 src1))
2043(rule 1 (x64_shr $I16 src1 (Imm8Gpr.Imm8 1)) (x64_shrw_m1 src1))
2044(rule 1 (x64_shr $I32 src1 (Imm8Gpr.Imm8 1)) (x64_shrl_m1 src1))
2045(rule 1 (x64_shr $I64 src1 (Imm8Gpr.Imm8 1)) (x64_shrq_m1 src1))
2046;; see `x64_shl` for more info about this rule
2047(rule 1 (x64_shr (ty_32_or_64 ty) src1 (Imm8Gpr.Gpr src2))
2048        (if-let true (has_bmi2))
2049        (x64_shrx ty src1 src2))
2050
2051;; Helper for creating arithmetic shift-right instructions.
2052(decl x64_sar (Type Gpr Imm8Gpr) Gpr)
2053(rule (x64_sar $I8 src1 (Imm8Gpr.Gpr src2)) (x64_sarb_mc src1 src2))
2054(rule (x64_sar $I8 src1 (Imm8Gpr.Imm8 src2)) (x64_sarb_mi src1 src2))
2055(rule (x64_sar $I16 src1 (Imm8Gpr.Gpr src2)) (x64_sarw_mc src1 src2))
2056(rule (x64_sar $I16 src1 (Imm8Gpr.Imm8 src2)) (x64_sarw_mi src1 src2))
2057(rule (x64_sar $I32 src1 (Imm8Gpr.Gpr src2)) (x64_sarl_mc src1 src2))
2058(rule (x64_sar $I32 src1 (Imm8Gpr.Imm8 src2)) (x64_sarl_mi src1 src2))
2059(rule (x64_sar $I64 src1 (Imm8Gpr.Gpr src2)) (x64_sarq_mc src1 src2))
2060(rule (x64_sar $I64 src1 (Imm8Gpr.Imm8 src2)) (x64_sarq_mi src1 src2))
2061(rule 1 (x64_sar $I8 src1 (Imm8Gpr.Imm8 1)) (x64_sarb_m1 src1))
2062(rule 1 (x64_sar $I16 src1 (Imm8Gpr.Imm8 1)) (x64_sarw_m1 src1))
2063(rule 1 (x64_sar $I32 src1 (Imm8Gpr.Imm8 1)) (x64_sarl_m1 src1))
2064(rule 1 (x64_sar $I64 src1 (Imm8Gpr.Imm8 1)) (x64_sarq_m1 src1))
2065;; see `x64_shl` for more info about this rule
2066(rule 1 (x64_sar (ty_32_or_64 ty) src1 (Imm8Gpr.Gpr src2))
2067        (if-let true (has_bmi2))
2068        (x64_sarx ty src1 src2))
2069
2070;; Helper for creating `shld` instructions.
2071(decl x64_shld (Type Gpr Gpr u8) Gpr)
2072;; NB: i8 is intentionally missing here as x64 doesn't have such an instruction
2073(rule (x64_shld $I16 src1 src2 amt) (x64_shldw_mri src1 src2 amt))
2074(rule (x64_shld $I32 src1 src2 amt) (x64_shldl_mri src1 src2 amt))
2075(rule (x64_shld $I64 src1 src2 amt) (x64_shldq_mri src1 src2 amt))
2076
2077;; Helper for creating zeroing-of-high-bits instructions bzhi
2078(decl x64_bzhi (Type GprMem Gpr) Gpr)
2079(rule (x64_bzhi $I32 src1 src2) (x64_bzhil_rmv src1 src2))
2080(rule (x64_bzhi $I64 src1 src2) (x64_bzhiq_rmv src1 src2))
2081
2082;; Helper for creating byteswap instructions.
2083;; In x64, 32- and 64-bit registers use BSWAP instruction, and
2084;; for 16-bit registers one must instead use xchg or rol/ror
2085(decl x64_bswap (Type Gpr) Gpr)
2086(rule (x64_bswap $I32 src) (x64_bswapl_o src))
2087(rule (x64_bswap $I64 src) (x64_bswapq_o src))
2088
2089;; Helper for creating `cmp` instructions.
2090(decl x64_cmp (Type Gpr GprMemImm) ProducesFlags)
2091
2092;; If the rhs is an immediate try to use the 8-bit form if the immediate fits.
2093(rule 2 (x64_cmp $I16 src1 (is_simm8 src2)) (x64_cmpw_mi_sxb src1 src2))
2094(rule 2 (x64_cmp $I32 src1 (is_simm8 src2)) (x64_cmpl_mi_sxb src1 src2))
2095(rule 2 (x64_cmp $I64 src1 (is_simm8 src2)) (x64_cmpq_mi_sxb src1 src2))
2096
2097;; Base case: rhs is an immediate
2098(rule 1 (x64_cmp $I8 src1 (is_imm8 src2)) (x64_cmpb_mi src1 src2))
2099(rule 1 (x64_cmp $I16 src1 (is_imm16 src2)) (x64_cmpw_mi src1 src2))
2100(rule 1 (x64_cmp $I32 src1 (is_imm32 src2)) (x64_cmpl_mi src1 src2))
2101(rule 1 (x64_cmp $I64 src1 (is_simm32 src2)) (x64_cmpq_mi src1 src2))
2102
2103;; Base case: rhs is a GprMem operand.
2104(rule 0 (x64_cmp $I8 src1 (is_gpr_mem src2)) (x64_cmpb_rm src1 src2))
2105(rule 0 (x64_cmp $I16 src1 (is_gpr_mem src2)) (x64_cmpw_rm src1 src2))
2106(rule 0 (x64_cmp $I32 src1 (is_gpr_mem src2)) (x64_cmpl_rm src1 src2))
2107(rule 0 (x64_cmp $I64 src1 (is_gpr_mem src2)) (x64_cmpq_rm src1 src2))
2108
2109;; Helper for creating floating-point comparison instructions (`UCOMIS[S|D]`).
2110(decl x64_ucomis (Type Xmm XmmMem) ProducesFlags)
2111(rule (x64_ucomis $F32 src1 src2) (x64_ucomiss_a_or_avx src1 src2))
2112(rule (x64_ucomis $F64 src1 src2) (x64_ucomisd_a_or_avx src1 src2))
2113
2114;; Helper for creating `test` instructions.
2115(decl x64_test (Type Gpr GprMemImm) ProducesFlags)
2116
2117(rule 1 (x64_test $I8 src1 (is_imm8 src2)) (x64_testb_mi src1 src2))
2118(rule 1 (x64_test $I16 src1 (is_imm16 src2)) (x64_testw_mi src1 src2))
2119(rule 1 (x64_test $I32 src1 (is_imm32 src2)) (x64_testl_mi src1 src2))
2120(rule 1 (x64_test $I64 src1 (is_simm32 src2)) (x64_testq_mi src1 src2))
2121
2122(rule 0 (x64_test $I8 src1 (is_gpr_mem src2)) (x64_testb_mr src2 src1))
2123(rule 0 (x64_test $I16 src1 (is_gpr_mem src2)) (x64_testw_mr src2 src1))
2124(rule 0 (x64_test $I32 src1 (is_gpr_mem src2)) (x64_testl_mr src2 src1))
2125(rule 0 (x64_test $I64 src1 (is_gpr_mem src2)) (x64_testq_mr src2 src1))
2126
2127;; Helper for creating `ptest` instructions.
2128(decl x64_ptest (Xmm XmmMem) ProducesFlags)
2129(rule (x64_ptest src1 src2) (x64_ptest_rm_or_avx src1 src2))
2130
2131;; Helper for creating `cmove` instructions. Note that these instructions do not
2132;; always result in a single emitted x86 instruction; e.g., XmmCmove uses jumps
2133;; to conditionally move the selected value into an XMM register.
2134;;
2135;; Also note that 8/16-bit conditional moves use the 32-bit instruction variant
2136;; since that is semantically equivalent and helps break data dependencies by
2137;; defining the entire register.
2138;;
2139;; Also note that the mnemonics used in `CC` don't always match those used in
2140;; the instruction variants and that is intentiona. This is due to the fact
2141;; that the Intel manual (and assemblers) support multiple mnemonics for the
2142;; same instruction but disassemblers only print one mnemonic and that's the
2143;; name used here.
2144(decl cmove (Type CC GprMem Gpr) ConsumesFlags)
2145(rule 0 (cmove (fits_in_32 _) (CC.O) c a) (x64_cmovol_rm a c))
2146(rule 0 (cmove (fits_in_32 _) (CC.NO) c a) (x64_cmovnol_rm a c))
2147(rule 0 (cmove (fits_in_32 _) (CC.B) c a) (x64_cmovbl_rm a c))
2148(rule 0 (cmove (fits_in_32 _) (CC.NB) c a) (x64_cmovael_rm a c)) ;;  nb == ae
2149(rule 0 (cmove (fits_in_32 _) (CC.Z) c a) (x64_cmovel_rm a c))   ;;   z ==  e
2150(rule 0 (cmove (fits_in_32 _) (CC.NZ) c a) (x64_cmovnel_rm a c)) ;;  nz == ne
2151(rule 0 (cmove (fits_in_32 _) (CC.BE) c a) (x64_cmovbel_rm a c))
2152(rule 0 (cmove (fits_in_32 _) (CC.NBE) c a) (x64_cmoval_rm a c)) ;; nbe ==  a
2153(rule 0 (cmove (fits_in_32 _) (CC.S) c a) (x64_cmovsl_rm a c))
2154(rule 0 (cmove (fits_in_32 _) (CC.NS) c a) (x64_cmovnsl_rm a c))
2155(rule 0 (cmove (fits_in_32 _) (CC.L) c a) (x64_cmovll_rm a c))
2156(rule 0 (cmove (fits_in_32 _) (CC.NL) c a) (x64_cmovgel_rm a c)) ;;  nl == ge
2157(rule 0 (cmove (fits_in_32 _) (CC.LE) c a) (x64_cmovlel_rm a c))
2158(rule 0 (cmove (fits_in_32 _) (CC.NLE) c a) (x64_cmovgl_rm a c)) ;; nle ==  g
2159(rule 0 (cmove (fits_in_32 _) (CC.P) c a) (x64_cmovpl_rm a c))
2160(rule 0 (cmove (fits_in_32 _) (CC.NP) c a) (x64_cmovnpl_rm a c))
2161(rule 1 (cmove $I64 (CC.O) c a) (x64_cmovoq_rm a c))
2162(rule 1 (cmove $I64 (CC.NO) c a) (x64_cmovnoq_rm a c))
2163(rule 1 (cmove $I64 (CC.B) c a) (x64_cmovbq_rm a c))
2164(rule 1 (cmove $I64 (CC.NB) c a) (x64_cmovaeq_rm a c)) ;;  nb == ae
2165(rule 1 (cmove $I64 (CC.Z) c a) (x64_cmoveq_rm a c))   ;;   z ==  e
2166(rule 1 (cmove $I64 (CC.NZ) c a) (x64_cmovneq_rm a c)) ;;  nz == ne
2167(rule 1 (cmove $I64 (CC.BE) c a) (x64_cmovbeq_rm a c))
2168(rule 1 (cmove $I64 (CC.NBE) c a) (x64_cmovaq_rm a c)) ;; nbe ==  a
2169(rule 1 (cmove $I64 (CC.S) c a) (x64_cmovsq_rm a c))
2170(rule 1 (cmove $I64 (CC.NS) c a) (x64_cmovnsq_rm a c))
2171(rule 1 (cmove $I64 (CC.L) c a) (x64_cmovlq_rm a c))
2172(rule 1 (cmove $I64 (CC.NL) c a) (x64_cmovgeq_rm a c)) ;;  nl == ge
2173(rule 1 (cmove $I64 (CC.LE) c a) (x64_cmovleq_rm a c))
2174(rule 1 (cmove $I64 (CC.NLE) c a) (x64_cmovgq_rm a c)) ;; nle ==  g
2175(rule 1 (cmove $I64 (CC.P) c a) (x64_cmovpq_rm a c))
2176(rule 1 (cmove $I64 (CC.NP) c a) (x64_cmovnpq_rm a c))
2177
2178(decl cmove_xmm (Type CC Xmm Xmm) ConsumesFlags)
2179(rule (cmove_xmm ty cc consequent alternative)
2180      (let ((dst WritableXmm (temp_writable_xmm)))
2181        (ConsumesFlags.ConsumesFlagsReturnsReg
2182         (MInst.XmmCmove ty cc consequent alternative dst)
2183         dst)))
2184
2185;; Helper for creating `setcc` instructions.
2186;;
2187;; Note that the mnemonics here don't always match exactly with the raw
2188;; instruction and that's intentional. The Intel manual documents multiple
2189;; mnemonics for the same opcode and the ones in Cranelift (CC.*) don't
2190;; match the ones that Capstone disassembles to (which the assembler matches).
2191(decl x64_setcc (CC) ConsumesFlags)
2192(rule (x64_setcc (CC.O)) (x64_seto_m))
2193(rule (x64_setcc (CC.NO)) (x64_setno_m))
2194(rule (x64_setcc (CC.B)) (x64_setb_m))
2195(rule (x64_setcc (CC.NB)) (x64_setae_m)) ;;  nb == ae
2196(rule (x64_setcc (CC.Z)) (x64_sete_m))   ;;   z ==  e
2197(rule (x64_setcc (CC.NZ)) (x64_setne_m)) ;;  nz == ne
2198(rule (x64_setcc (CC.BE)) (x64_setbe_m))
2199(rule (x64_setcc (CC.NBE)) (x64_seta_m)) ;; nbe ==  a
2200(rule (x64_setcc (CC.S)) (x64_sets_m))
2201(rule (x64_setcc (CC.NS)) (x64_setns_m))
2202(rule (x64_setcc (CC.L)) (x64_setl_m))
2203(rule (x64_setcc (CC.NL)) (x64_setge_m)) ;;  nl == ge
2204(rule (x64_setcc (CC.LE)) (x64_setle_m))
2205(rule (x64_setcc (CC.NLE)) (x64_setg_m)) ;; nle ==  g
2206(rule (x64_setcc (CC.P)) (x64_setp_m))
2207(rule (x64_setcc (CC.NP)) (x64_setnp_m))
2208
2209;; Helper for creating `setcc` instructions, when the flags producer will
2210;; also return a value.
2211(decl x64_setcc_paired (CC) ConsumesFlags)
2212(rule (x64_setcc_paired cc) (consumes_flags_with_producer (x64_setcc cc)))
2213
2214(decl consumes_flags_with_producer (ConsumesFlags) ConsumesFlags)
2215(rule (consumes_flags_with_producer (ConsumesFlags.ConsumesFlagsReturnsReg flags reg))
2216  (ConsumesFlags.ConsumesFlagsReturnsResultWithProducer flags reg))
2217
2218
2219;; Helpers for creating vector `add` instructions.
2220(decl x64_addss (Xmm XmmMem) Xmm)
2221(rule (x64_addss src1 src2) (x64_addss_a_or_avx src1 src2))
2222
2223(decl x64_addsd (Xmm XmmMem) Xmm)
2224(rule (x64_addsd src1 src2) (x64_addsd_a_or_avx src1 src2))
2225
2226(decl x64_addps (Xmm XmmMem) Xmm)
2227(rule (x64_addps src1 src2) (x64_addps_a_or_avx src1 src2))
2228
2229(decl x64_addpd (Xmm XmmMem) Xmm)
2230(rule (x64_addpd src1 src2) (x64_addpd_a_or_avx src1 src2))
2231
2232(decl x64_paddb (Xmm XmmMem) Xmm)
2233(rule (x64_paddb src1 src2) (x64_paddb_a_or_avx src1 src2))
2234
2235(decl x64_paddw (Xmm XmmMem) Xmm)
2236(rule (x64_paddw src1 src2) (x64_paddw_a_or_avx src1 src2))
2237
2238(decl x64_paddd (Xmm XmmMem) Xmm)
2239(rule (x64_paddd src1 src2) (x64_paddd_a_or_avx src1 src2))
2240
2241(decl x64_paddq (Xmm XmmMem) Xmm)
2242(rule (x64_paddq src1 src2) (x64_paddq_a_or_avx src1 src2))
2243
2244(decl x64_paddsb (Xmm XmmMem) Xmm)
2245(rule (x64_paddsb src1 src2) (x64_paddsb_a_or_avx src1 src2))
2246
2247(decl x64_paddsw (Xmm XmmMem) Xmm)
2248(rule (x64_paddsw src1 src2) (x64_paddsw_a_or_avx src1 src2))
2249
2250(decl x64_phaddw (Xmm XmmMem) Xmm)
2251(rule (x64_phaddw src1 src2) (x64_phaddw_a_or_avx src1 src2))
2252
2253(decl x64_phaddd (Xmm XmmMem) Xmm)
2254(rule (x64_phaddd src1 src2) (x64_phaddd_a_or_avx src1 src2))
2255
2256(decl x64_paddusb (Xmm XmmMem) Xmm)
2257(rule (x64_paddusb src1 src2) (x64_paddusb_a_or_avx src1 src2))
2258
2259(decl x64_paddusw (Xmm XmmMem) Xmm)
2260(rule (x64_paddusw src1 src2) (x64_paddusw_a_or_avx src1 src2))
2261
2262;; Helpers for creating vector `sub` instructions.
2263(decl x64_subss (Xmm XmmMem) Xmm)
2264(rule (x64_subss src1 src2) (x64_subss_a_or_avx src1 src2))
2265
2266(decl x64_subsd (Xmm XmmMem) Xmm)
2267(rule (x64_subsd src1 src2) (x64_subsd_a_or_avx src1 src2))
2268
2269(decl x64_subps (Xmm XmmMem) Xmm)
2270(rule (x64_subps src1 src2) (x64_subps_a_or_avx src1 src2))
2271
2272(decl x64_subpd (Xmm XmmMem) Xmm)
2273(rule (x64_subpd src1 src2) (x64_subpd_a_or_avx src1 src2))
2274
2275(decl x64_psubb (Xmm XmmMem) Xmm)
2276(rule (x64_psubb src1 src2) (x64_psubb_a_or_avx src1 src2))
2277
2278(decl x64_psubw (Xmm XmmMem) Xmm)
2279(rule (x64_psubw src1 src2) (x64_psubw_a_or_avx src1 src2))
2280
2281(decl x64_psubd (Xmm XmmMem) Xmm)
2282(rule (x64_psubd src1 src2) (x64_psubd_a_or_avx src1 src2))
2283
2284(decl x64_psubq (Xmm XmmMem) Xmm)
2285(rule (x64_psubq src1 src2) (x64_psubq_a_or_avx src1 src2))
2286
2287(decl x64_psubsb (Xmm XmmMem) Xmm)
2288(rule (x64_psubsb src1 src2) (x64_psubsb_a_or_avx src1 src2))
2289
2290(decl x64_psubsw (Xmm XmmMem) Xmm)
2291(rule (x64_psubsw src1 src2) (x64_psubsw_a_or_avx src1 src2))
2292
2293(decl x64_psubusb (Xmm XmmMem) Xmm)
2294(rule (x64_psubusb src1 src2) (x64_psubusb_a_or_avx src1 src2))
2295
2296(decl x64_psubusw (Xmm XmmMem) Xmm)
2297(rule (x64_psubusw src1 src2) (x64_psubusw_a_or_avx src1 src2))
2298
2299;; Helpers for creating `pavg*` instructions.
2300(decl x64_pavgb (Xmm XmmMem) Xmm)
2301(rule (x64_pavgb src1 src2) (x64_pavgb_a_or_avx src1 src2))
2302
2303(decl x64_pavgw (Xmm XmmMem) Xmm)
2304(rule (x64_pavgw src1 src2) (x64_pavgw_a_or_avx src1 src2))
2305
2306;; Helpers for creating vector `and` instructions.
2307(decl x64_pand (Xmm XmmMem) Xmm)
2308(rule (x64_pand src1 src2) (x64_pand_a_or_avx src1 src2))
2309
2310(decl x64_andps (Xmm XmmMem) Xmm)
2311(rule (x64_andps src1 src2) (x64_andps_a_or_avx src1 src2))
2312
2313(decl x64_andpd (Xmm XmmMem) Xmm)
2314(rule (x64_andpd src1 src2) (x64_andpd_a_or_avx src1 src2))
2315
2316;; Helpers for creating vector `or` instructions.
2317(decl x64_por (Xmm XmmMem) Xmm)
2318(rule (x64_por src1 src2) (x64_por_a_or_avx src1 src2))
2319
2320(decl x64_orps (Xmm XmmMem) Xmm)
2321(rule (x64_orps src1 src2) (x64_orps_a_or_avx src1 src2))
2322
2323(decl x64_orpd (Xmm XmmMem) Xmm)
2324(rule (x64_orpd src1 src2) (x64_orpd_a_or_avx src1 src2))
2325
2326;; Helpers for creating vector `xor` instructions.
2327(decl x64_pxor (Xmm XmmMem) Xmm)
2328(rule (x64_pxor src1 src2) (x64_pxor_a_or_avx src1 src2))
2329
2330(decl x64_xorps (Xmm XmmMem) Xmm)
2331(rule (x64_xorps src1 src2) (x64_xorps_a_or_avx src1 src2))
2332
2333(decl x64_xorpd (Xmm XmmMem) Xmm)
2334(rule (x64_xorpd src1 src2) (x64_xorpd_a_or_avx src1 src2))
2335
2336;; Helpers for creating vector `andn` instructions.
2337(decl x64_andnps (Xmm XmmMem) Xmm)
2338(rule (x64_andnps src1 src2) (x64_andnps_a_or_avx src1 src2))
2339
2340(decl x64_andnpd (Xmm XmmMem) Xmm)
2341(rule (x64_andnpd src1 src2) (x64_andnpd_a_or_avx src1 src2))
2342
2343(decl x64_pandn (Xmm XmmMem) Xmm)
2344(rule (x64_pandn src1 src2) (x64_pandn_a_or_avx src1 src2))
2345
2346;; Helper for creating vector `mul` instructions.
2347(decl x64_mulss (Xmm XmmMem) Xmm)
2348(rule (x64_mulss src1 src2) (x64_mulss_a_or_avx src1 src2))
2349
2350(decl x64_mulsd (Xmm XmmMem) Xmm)
2351(rule (x64_mulsd src1 src2) (x64_mulsd_a_or_avx src1 src2))
2352
2353(decl x64_mulps (Xmm XmmMem) Xmm)
2354(rule (x64_mulps src1 src2) (x64_mulps_a_or_avx src1 src2))
2355
2356(decl x64_mulpd (Xmm XmmMem) Xmm)
2357(rule (x64_mulpd src1 src2) (x64_mulpd_a_or_avx src1 src2))
2358
2359(decl x64_pmullw (Xmm XmmMem) Xmm)
2360(rule (x64_pmullw src1 src2) (x64_pmullw_a_or_avx src1 src2))
2361
2362(decl x64_pmulld (Xmm XmmMem) Xmm)
2363(rule (x64_pmulld src1 src2) (x64_pmulld_a_or_avx src1 src2))
2364
2365(decl x64_pmulhw (Xmm XmmMem) Xmm)
2366(rule (x64_pmulhw src1 src2) (x64_pmulhw_a_or_avx src1 src2))
2367
2368(decl x64_pmulhrsw (Xmm XmmMem) Xmm)
2369(rule (x64_pmulhrsw src1 src2) (x64_pmulhrsw_a_or_avx src1 src2))
2370
2371(decl x64_pmulhuw (Xmm XmmMem) Xmm)
2372(rule (x64_pmulhuw src1 src2) (x64_pmulhuw_a_or_avx src1 src2))
2373
2374(decl x64_pmuldq (Xmm XmmMem) Xmm)
2375(rule (x64_pmuldq src1 src2) (x64_pmuldq_a_or_avx src1 src2))
2376
2377(decl x64_pmuludq (Xmm XmmMem) Xmm)
2378(rule (x64_pmuludq src1 src2) (x64_pmuludq_a_or_avx src1 src2))
2379
2380;; Helpers for creating vector `div` instructions.
2381(decl x64_divss (Xmm XmmMem) Xmm)
2382(rule (x64_divss src1 src2) (x64_divss_a_or_avx src1 src2))
2383
2384(decl x64_divsd (Xmm XmmMem) Xmm)
2385(rule (x64_divsd src1 src2) (x64_divsd_a_or_avx src1 src2))
2386
2387(decl x64_divps (Xmm XmmMem) Xmm)
2388(rule (x64_divps src1 src2) (x64_divps_a_or_avx src1 src2))
2389
2390(decl x64_divpd (Xmm XmmMem) Xmm)
2391(rule (x64_divpd src1 src2) (x64_divpd_a_or_avx src1 src2))
2392
2393;; Helpers for creating `unpack` instructions.
2394(decl x64_punpckhwd (Xmm XmmMem) Xmm)
2395(rule (x64_punpckhwd src1 src2) (x64_punpckhwd_a_or_avx src1 src2))
2396
2397(decl x64_punpcklwd (Xmm XmmMem) Xmm)
2398(rule (x64_punpcklwd src1 src2) (x64_punpcklwd_a_or_avx src1 src2))
2399
2400(decl x64_punpckldq (Xmm XmmMem) Xmm)
2401(rule (x64_punpckldq src1 src2) (x64_punpckldq_a_or_avx src1 src2))
2402
2403(decl x64_punpckhdq (Xmm XmmMem) Xmm)
2404(rule (x64_punpckhdq src1 src2) (x64_punpckhdq_a_or_avx src1 src2))
2405
2406(decl x64_punpcklqdq (Xmm XmmMem) Xmm)
2407(rule (x64_punpcklqdq src1 src2) (x64_punpcklqdq_a_or_avx src1 src2))
2408
2409(decl x64_punpckhqdq (Xmm XmmMem) Xmm)
2410(rule (x64_punpckhqdq src1 src2) (x64_punpckhqdq_a_or_avx src1 src2))
2411
2412(decl x64_unpcklps (Xmm XmmMem) Xmm)
2413(rule (x64_unpcklps src1 src2) (x64_unpcklps_a_or_avx src1 src2))
2414
2415(decl x64_unpcklpd (Xmm XmmMem) Xmm)
2416(rule (x64_unpcklpd src1 src2) (x64_unpcklpd_a_or_avx src1 src2))
2417
2418(decl x64_unpckhps (Xmm XmmMem) Xmm)
2419(rule (x64_unpckhps src1 src2) (x64_unpckhps_a_or_avx src1 src2))
2420
2421(decl x64_punpcklbw (Xmm XmmMem) Xmm)
2422(rule (x64_punpcklbw src1 src2) (x64_punpcklbw_a_or_avx src1 src2))
2423
2424(decl x64_punpckhbw (Xmm XmmMem) Xmm)
2425(rule  (x64_punpckhbw src1 src2) (x64_punpckhbw_a_or_avx src1 src2))
2426
2427;; Helper for creating `blendvpd` instructions.
2428(decl x64_blendvpd (Xmm XmmMem Xmm) Xmm)
2429(rule 0 (x64_blendvpd src1 src2 mask) (x64_blendvpd_rm0 src1 src2 mask))
2430(rule 1 (x64_blendvpd src1 src2 mask)
2431      (if-let true (has_avx))
2432      (x64_vblendvpd_rvmr src1 src2 mask))
2433
2434;; Helper for creating `blendvps` instructions.
2435(decl x64_blendvps (Xmm XmmMem Xmm) Xmm)
2436(rule 0 (x64_blendvps src1 src2 mask) (x64_blendvps_rm0 src1 src2 mask))
2437(rule 1 (x64_blendvps src1 src2 mask)
2438      (if-let true (has_avx))
2439      (x64_vblendvps_rvmr src1 src2 mask))
2440
2441;; Helper for creating `pblendvb` instructions.
2442(decl x64_pblendvb (Xmm XmmMem Xmm) Xmm)
2443(rule 0 (x64_pblendvb src1 src2 mask) (x64_pblendvb_rm src1 src2 mask))
2444(rule 1 (x64_pblendvb src1 src2 mask)
2445      (if-let true (has_avx))
2446      (x64_vpblendvb_rvmr src1 src2 mask))
2447
2448;; Helper for creating `pblendw` instructions.
2449(decl x64_pblendw (Xmm XmmMem u8) Xmm)
2450(rule (x64_pblendw src1 src2 imm) (x64_pblendw_rmi_or_avx src1 src2 imm))
2451
2452;; Helper for creating `movsd`/`movss` instructions which create a new vector
2453;; register where the upper bits are from the first operand and the low
2454;; bits are from the second operand.
2455;;
2456;; Note that the second argument here is specifically `Xmm` instead of `XmmMem`
2457;; because there is no encoding of a 3-operand form of `movsd` and otherwise
2458;; when used as a load instruction it wipes out the entire destination register
2459;; which defeats the purpose of this being a 2-operand instruction.
2460(decl x64_movsd_regmove (Xmm Xmm) Xmm)
2461(rule (x64_movsd_regmove src1 src2) (x64_movsd_a_r_or_avx src1 src2))
2462
2463(decl x64_movss_regmove (Xmm Xmm) Xmm)
2464(rule (x64_movss_regmove src1 src2) (x64_movss_a_r_or_avx src1 src2))
2465
2466;; Helper for creating `movlhps` instructions.
2467(decl x64_movlhps (Xmm Xmm) Xmm)
2468(rule (x64_movlhps src1 src2) (x64_movlhps_rm_or_avx src1 src2))
2469
2470;; Helpers for creating `pmaxs*` instructions.
2471(decl x64_pmaxs (Type Xmm XmmMem) Xmm)
2472(rule (x64_pmaxs $I8X16 x y) (x64_pmaxsb_a_or_avx x y))
2473(rule (x64_pmaxs $I16X8 x y) (x64_pmaxsw_a_or_avx x y))
2474(rule (x64_pmaxs $I32X4 x y) (x64_pmaxsd_a_or_avx x y))
2475;; No $I64X2 version (PMAXSQ) in SSE4.1.
2476
2477;; Helpers for creating `pmins*` instructions.
2478(decl x64_pmins (Type Xmm XmmMem) Xmm)
2479(rule (x64_pmins $I8X16 x y) (x64_pminsb_a_or_avx x y))
2480(rule (x64_pmins $I16X8 x y) (x64_pminsw_a_or_avx x y))
2481(rule (x64_pmins $I32X4 x y) (x64_pminsd_a_or_avx x y))
2482;; No $I64X2 version (PMINSQ) in SSE4.1.
2483
2484;; Helpers for creating `pmaxu*` instructions.
2485(decl x64_pmaxu (Type Xmm XmmMem) Xmm)
2486(rule (x64_pmaxu $I8X16 x y) (x64_pmaxub_a_or_avx x y))
2487(rule (x64_pmaxu $I16X8 x y) (x64_pmaxuw_a_or_avx x y))
2488(rule (x64_pmaxu $I32X4 x y) (x64_pmaxud_a_or_avx x y))
2489;; No $I64X2 version (PMAXUQ) in SSE4.1.
2490
2491;; Helper for creating `pminu*` instructions.
2492(decl x64_pminu (Type Xmm XmmMem) Xmm)
2493(rule (x64_pminu $I8X16 x y) (x64_pminub_a_or_avx x y))
2494(rule (x64_pminu $I16X8 x y) (x64_pminuw_a_or_avx x y))
2495(rule (x64_pminu $I32X4 x y) (x64_pminud_a_or_avx x y))
2496;; No $I64X2 version (PMINUQ) in SSE4.1.
2497
2498;; Helper for creating `packsswb` instructions.
2499(decl x64_packsswb (Xmm XmmMem) Xmm)
2500(rule (x64_packsswb src1 src2) (x64_packsswb_a_or_avx src1 src2))
2501
2502;; Helper for creating `packssdw` instructions.
2503(decl x64_packssdw (Xmm XmmMem) Xmm)
2504(rule (x64_packssdw src1 src2) (x64_packssdw_a_or_avx src1 src2))
2505
2506;; Helper for creating `packuswb` instructions.
2507(decl x64_packuswb (Xmm XmmMem) Xmm)
2508(rule (x64_packuswb src1 src2) (x64_packuswb_a_or_avx src1 src2))
2509
2510;; Helper for creating `packusdw` instructions.
2511(decl x64_packusdw (Xmm XmmMem) Xmm)
2512(rule (x64_packusdw src1 src2) (x64_packusdw_a_or_avx src1 src2))
2513
2514;; Helper for creating `palignr` instructions.
2515(decl x64_palignr (Xmm XmmMem u8) Xmm)
2516(rule (x64_palignr src1 src2 imm) (x64_palignr_a_or_avx src1 src2 imm))
2517
2518;; Helpers for creating `cmpp*` instructions.
2519(decl x64_cmpp (Type Xmm XmmMem FcmpImm) Xmm)
2520(rule (x64_cmpp $F32X4 x y imm) (x64_cmpps x y imm))
2521(rule (x64_cmpp $F64X2 x y imm) (x64_cmppd x y imm))
2522
2523(decl x64_cmpps (Xmm XmmMem FcmpImm) Xmm)
2524(rule 1 (x64_cmpps src1 src2 imm)
2525      (if-let true (has_avx))
2526      (x64_vcmpps_b src1 src2 (encode_fcmp_imm imm)))
2527(rule 0 (x64_cmpps src1 src2 imm) (x64_cmpps_a src1 src2 (encode_fcmp_imm imm)))
2528
2529;; Note that `Size32` is intentional despite this being used for 64-bit
2530;; operations, since this presumably induces the correct encoding of the
2531;; instruction.
2532(decl x64_cmppd (Xmm XmmMem FcmpImm) Xmm)
2533(rule 1 (x64_cmppd src1 src2 imm)
2534      (if-let true (has_avx))
2535      (x64_vcmppd_b src1 src2 (encode_fcmp_imm imm)))
2536(rule 0 (x64_cmppd src1 src2 imm) (x64_cmppd_a src1 src2 (encode_fcmp_imm imm)))
2537
2538;; Helper for creating `pinsrb` instructions.
2539(decl x64_pinsrb (Xmm GprMem u8) Xmm)
2540(rule 1 (x64_pinsrb src1 src2 lane)
2541      (if-let true (has_avx))
2542      (x64_vpinsrb_b src1 src2 lane))
2543(rule 0 (x64_pinsrb src1 src2 lane) (x64_pinsrb_a src1 src2 lane))
2544
2545;; Helper for creating `pinsrw` instructions.
2546(decl x64_pinsrw (Xmm GprMem u8) Xmm)
2547(rule 1 (x64_pinsrw src1 src2 lane)
2548      (if-let true (has_avx))
2549      (x64_vpinsrw_b src1 src2 lane))
2550(rule 0 (x64_pinsrw src1 src2 lane) (x64_pinsrw_a src1 src2 lane))
2551
2552;; Helper for creating `pinsrd` instructions.
2553(decl x64_pinsrd (Xmm GprMem u8) Xmm)
2554(rule 1 (x64_pinsrd src1 src2 lane)
2555      (if-let true (has_avx))
2556      (x64_vpinsrd_b src1 src2 lane))
2557(rule 0 (x64_pinsrd src1 src2 lane) (x64_pinsrd_a src1 src2 lane))
2558
2559;; Helper for creating `pinsrq` instructions.
2560(decl x64_pinsrq (Xmm GprMem u8) Xmm)
2561(rule 1 (x64_pinsrq src1 src2 lane)
2562      (if-let true (has_avx))
2563      (x64_vpinsrq_b src1 src2 lane))
2564(rule 0 (x64_pinsrq src1 src2 lane) (x64_pinsrq_a src1 src2 lane))
2565
2566;; Helper for creating `roundss` instructions.
2567(decl x64_roundss (XmmMem RoundImm) Xmm)
2568(rule 1 (x64_roundss src1 round)
2569        (if-let true (has_avx))
2570        (x64_vroundss_rvmi (xmm_zero $F32X4) src1 (encode_round_imm round)))
2571(rule 0 (x64_roundss src1 round)
2572        (x64_roundss_rmi src1 (encode_round_imm round)))
2573
2574;; Helper for creating `roundsd` instructions.
2575(decl x64_roundsd (XmmMem RoundImm) Xmm)
2576(rule 1 (x64_roundsd src1 round)
2577        (if-let true (has_avx))
2578        (x64_vroundsd_rvmi (xmm_zero $F64X2) src1 (encode_round_imm round)))
2579(rule 0 (x64_roundsd src1 round)
2580        (x64_roundsd_rmi src1 (encode_round_imm round)))
2581
2582;; Helper for creating `roundps` instructions.
2583(decl x64_roundps (XmmMem RoundImm) Xmm)
2584(rule 1 (x64_roundps src1 round)
2585      (if-let true (has_avx))
2586      (x64_vroundps_rmi src1 (encode_round_imm round)))
2587(rule (x64_roundps src1 round)
2588      (x64_roundps_rmi src1 (encode_round_imm round)))
2589
2590;; Helper for creating `roundpd` instructions.
2591(decl x64_roundpd (XmmMem RoundImm) Xmm)
2592(rule 1 (x64_roundpd src1 round)
2593      (if-let true (has_avx))
2594      (x64_vroundpd_rmi src1 (encode_round_imm round)))
2595(rule 0 (x64_roundpd src1 round)
2596      (x64_roundpd_rmi src1 (encode_round_imm round)))
2597
2598;; Helper for creating `pmaddwd` instructions.
2599(decl x64_pmaddwd (Xmm XmmMem) Xmm)
2600(rule 0 (x64_pmaddwd src1 src2) (x64_pmaddwd_a_or_avx src1 src2))
2601
2602(decl x64_pmaddubsw (Xmm XmmMem) Xmm)
2603(rule (x64_pmaddubsw src1 src2) (x64_pmaddubsw_a_or_avx src1 src2))
2604
2605;; Helper for creating `insertps` instructions.
2606(decl x64_insertps (Xmm XmmMem u8) Xmm)
2607(rule 0 (x64_insertps src1 src2 lane) (x64_insertps_a_or_avx src1 src2 lane))
2608
2609;; Helper for creating `pshufd` instructions.
2610(decl x64_pshufd (XmmMem u8) Xmm)
2611(rule (x64_pshufd src imm) (x64_pshufd_a src imm))
2612(rule 1 (x64_pshufd src imm)
2613      (if-let true (has_avx))
2614      (x64_vpshufd_a src imm))
2615
2616;; Helper for creating `pshufb` instructions.
2617(decl x64_pshufb (Xmm XmmMem) Xmm)
2618(rule (x64_pshufb src1 src2) (x64_pshufb_a_or_avx src1 src2))
2619
2620;; Helper for creating `shufpd` instructions.
2621(decl x64_shufpd (Xmm XmmMem u8) Xmm)
2622(rule (x64_shufpd src1 src2 byte) (x64_shufpd_a_or_avx src1 src2 byte))
2623
2624;; Helper for creating `shufps` instructions.
2625(decl x64_shufps (Xmm XmmMem u8) Xmm)
2626(rule (x64_shufps src1 src2 byte) (x64_shufps_a_or_avx src1 src2 byte))
2627
2628;; Helper for creating `pshuflw` instructions.
2629(decl x64_pshuflw (XmmMem u8) Xmm)
2630(rule (x64_pshuflw src imm) (x64_pshuflw_a src imm))
2631(rule 1 (x64_pshuflw src imm)
2632      (if-let true (has_avx))
2633      (x64_vpshuflw_a src imm))
2634
2635;; Helper for creating `pshufhw` instructions.
2636(decl x64_pshufhw (XmmMem u8) Xmm)
2637(rule (x64_pshufhw src imm) (x64_pshufhw_a src imm))
2638(rule 1 (x64_pshufhw src imm)
2639      (if-let true (has_avx))
2640      (x64_vpshufhw_a src imm))
2641
2642
2643
2644;; Helper for creating `vcvtudq2ps` instructions.
2645(decl x64_vcvtudq2ps (XmmMem) Xmm)
2646(rule (x64_vcvtudq2ps src) (x64_vcvtudq2ps_a src))
2647
2648;; Helper for creating `vpabsq` instructions.
2649(decl x64_vpabsq (XmmMem) Xmm)
2650(rule (x64_vpabsq src) (x64_vpabsq_c src))
2651
2652;; Helper for creating `vpopcntb` instructions.
2653(decl x64_vpopcntb (XmmMem) Xmm)
2654(rule (x64_vpopcntb src) (x64_vpopcntb_a src))
2655
2656;; Helper for creating `vpmullq` instructions.
2657;;
2658;; Requires AVX-512 vl and dq.
2659(decl x64_vpmullq (Xmm XmmMem) Xmm)
2660(rule (x64_vpmullq src1 src2) (x64_vpmullq_c src1 src2))
2661
2662;; Helper for creating `vpermi2b` instructions.
2663;;
2664;; Requires AVX-512 vl and vbmi extensions.
2665(decl x64_vpermi2b (Xmm Xmm XmmMem) Xmm)
2666(rule (x64_vpermi2b src1 src2 src3) (x64_vpermi2b_a src1 src2 src3))
2667
2668;; Helpers for creating vector `shift` instructions.
2669(decl x64_psllw (Xmm XmmMemImm) Xmm)
2670(rule 1 (x64_psllw src1 (is_xmm_mem src2))  (x64_psllw_a_or_avx src1 src2))
2671(rule 0 (x64_psllw src1 (is_imm8_xmm src2)) (x64_psllw_b_or_avx src1 src2))
2672
2673(decl x64_pslld (Xmm XmmMemImm) Xmm)
2674(rule 1 (x64_pslld src1 (is_xmm_mem src2))  (x64_pslld_a_or_avx src1 src2))
2675(rule 0 (x64_pslld src1 (is_imm8_xmm src2)) (x64_pslld_b_or_avx src1 src2))
2676
2677(decl x64_psllq (Xmm XmmMemImm) Xmm)
2678(rule 1 (x64_psllq src1 (is_xmm_mem src2))  (x64_psllq_a_or_avx src1 src2))
2679(rule 0 (x64_psllq src1 (is_imm8_xmm src2)) (x64_psllq_b_or_avx src1 src2))
2680
2681(decl x64_psrlw (Xmm XmmMemImm) Xmm)
2682(rule 1 (x64_psrlw src1 (is_xmm_mem src2))  (x64_psrlw_a_or_avx src1 src2))
2683(rule 0 (x64_psrlw src1 (is_imm8_xmm src2)) (x64_psrlw_b_or_avx src1 src2))
2684
2685(decl x64_psrld (Xmm XmmMemImm) Xmm)
2686(rule 1 (x64_psrld src1 (is_xmm_mem src2))  (x64_psrld_a_or_avx src1 src2))
2687(rule 0 (x64_psrld src1 (is_imm8_xmm src2)) (x64_psrld_b_or_avx src1 src2))
2688
2689(decl x64_psrlq (Xmm XmmMemImm) Xmm)
2690(rule 1 (x64_psrlq src1 (is_xmm_mem src2))  (x64_psrlq_a_or_avx src1 src2))
2691(rule 0 (x64_psrlq src1 (is_imm8_xmm src2)) (x64_psrlq_b_or_avx src1 src2))
2692
2693(decl x64_psraw (Xmm XmmMemImm) Xmm)
2694(rule 1 (x64_psraw src1 (is_xmm_mem src2))  (x64_psraw_a_or_avx src1 src2))
2695(rule 0 (x64_psraw src1 (is_imm8_xmm src2)) (x64_psraw_b_or_avx src1 src2))
2696
2697(decl x64_psrad (Xmm XmmMemImm) Xmm)
2698(rule 1 (x64_psrad src1 (is_xmm_mem src2))  (x64_psrad_a_or_avx src1 src2))
2699(rule 0 (x64_psrad src1 (is_imm8_xmm src2)) (x64_psrad_b_or_avx src1 src2))
2700
2701;; Helper for creating `vpsraq` instructions.
2702(decl x64_vpsraq (Xmm XmmMem) Xmm)
2703(rule (x64_vpsraq src1 src2) (x64_vpsraq_g src1 src2))
2704
2705;; Helper for creating `vpsraq` instructions.
2706(decl x64_vpsraq_imm (XmmMem u8) Xmm)
2707(rule (x64_vpsraq_imm src imm) (x64_vpsraq_f src imm))
2708
2709;; Helper for creating `pextr*` instructions.
2710(decl x64_pextrb (Xmm u8) Gpr)
2711(rule (x64_pextrb src lane) (x64_pextrb_a_or_avx src lane))
2712
2713(decl x64_pextrb_store (SyntheticAmode Xmm u8) SideEffectNoResult)
2714(rule (x64_pextrb_store addr src lane) (x64_pextrb_a_mem_or_avx addr src lane))
2715
2716(decl x64_pextrw (Xmm u8) Gpr)
2717(rule (x64_pextrw src lane) (x64_pextrw_a_or_avx src lane))
2718
2719(decl x64_pextrw_store (SyntheticAmode Xmm u8) SideEffectNoResult)
2720(rule (x64_pextrw_store addr src lane) (x64_pextrw_b_mem_or_avx addr src lane))
2721
2722(decl x64_pextrd (Xmm u8) Gpr)
2723(rule (x64_pextrd src lane) (x64_pextrd_a_or_avx src lane))
2724
2725(decl x64_pextrd_store (SyntheticAmode Xmm u8) SideEffectNoResult)
2726(rule (x64_pextrd_store addr src lane) (x64_pextrd_a_mem_or_avx addr src lane))
2727
2728(decl x64_pextrq (Xmm u8) Gpr)
2729(rule (x64_pextrq src lane) (x64_pextrq_a_or_avx src lane))
2730
2731(decl x64_pextrq_store (SyntheticAmode Xmm u8) SideEffectNoResult)
2732(rule (x64_pextrq_store addr src lane) (x64_pextrq_a_mem_or_avx addr src lane))
2733
2734;; Helper for creating `pmovmskb` instructions.
2735(decl x64_pmovmskb (Xmm) Gpr)
2736(rule (x64_pmovmskb src) (x64_pmovmskb_rm src))
2737(rule 1 (x64_pmovmskb src)
2738        (if-let true (has_avx))
2739        (x64_vpmovmskb_rm src))
2740
2741;; Helper for creating `movmskps` instructions.
2742(decl x64_movmskps (Xmm) Gpr)
2743(rule (x64_movmskps src) (x64_movmskps_rm src))
2744(rule 1 (x64_movmskps src)
2745        (if-let true (has_avx))
2746        (x64_vmovmskps_rm src))
2747
2748;; Helper for creating `movmskpd` instructions.
2749(decl x64_movmskpd (Xmm) Gpr)
2750(rule (x64_movmskpd src) (x64_movmskpd_rm src))
2751(rule 1 (x64_movmskpd src)
2752        (if-let true (has_avx))
2753        (x64_vmovmskpd_rm src))
2754
2755;; Helper for creating `not` instructions.
2756(decl x64_not (Type Gpr) Gpr)
2757(rule (x64_not $I8 src)  (x64_notb_m src))
2758(rule (x64_not $I16 src) (x64_notw_m src))
2759(rule (x64_not $I32 src) (x64_notl_m src))
2760(rule (x64_not $I64 src) (x64_notq_m src))
2761
2762;; Helpers for creating `neg` instructions.
2763(decl x64_neg_raw (Type Gpr) AssemblerOutputs)
2764(rule (x64_neg_raw $I8 src)  (x64_negb_m_raw src))
2765(rule (x64_neg_raw $I16 src) (x64_negw_m_raw src))
2766(rule (x64_neg_raw $I32 src) (x64_negl_m_raw src))
2767(rule (x64_neg_raw $I64 src) (x64_negq_m_raw src))
2768
2769(decl x64_neg (Type Gpr) Gpr)
2770(rule (x64_neg ty src)
2771      (emit_ret_gpr (x64_neg_raw ty src)))
2772
2773(decl x64_neg_paired (Type Gpr) ProducesFlags)
2774(rule (x64_neg_paired ty src)
2775      (asm_produce_flags (x64_neg_raw ty src)))
2776
2777(spec (x64_lea ty amode)
2778      (provide (= result amode))
2779      (require (or (= ty 32) (= ty 64))))
2780(decl x64_lea (Type SyntheticAmode) Gpr)
2781(rule (x64_lea $I16 addr) (x64_leaw_rm addr))
2782(rule (x64_lea $I32 addr) (x64_leal_rm addr))
2783(rule (x64_lea $I64 addr) (x64_leaq_rm addr))
2784
2785;; Helper for creating `lzcnt` instructions.
2786(decl x64_lzcnt (Type GprMem) Gpr)
2787(rule (x64_lzcnt $I16 src) (x64_lzcntw_rm src))
2788(rule (x64_lzcnt $I32 src) (x64_lzcntl_rm src))
2789(rule (x64_lzcnt $I64 src) (x64_lzcntq_rm src))
2790
2791;; Helper for creating `tzcnt` instructions.
2792(decl x64_tzcnt (Type GprMem) Gpr)
2793(rule (x64_tzcnt $I16 src) (x64_tzcntw_a src))
2794(rule (x64_tzcnt $I32 src) (x64_tzcntl_a src))
2795(rule (x64_tzcnt $I64 src) (x64_tzcntq_a src))
2796
2797;; Helper for creating `bsr` instructions.
2798(decl x64_bsr (Type GprMem) ProducesFlags)
2799(rule (x64_bsr $I16 src) (asm_produce_flags (x64_bsrw_rm_raw src)))
2800(rule (x64_bsr $I32 src) (asm_produce_flags (x64_bsrl_rm_raw src)))
2801(rule (x64_bsr $I64 src) (asm_produce_flags (x64_bsrq_rm_raw src)))
2802
2803;; Helper for creating `bsr + cmov` instruction pairs that produce the
2804;; result of the `bsr`, or `alt` if the input was zero.
2805(decl bsr_or_else (Type Gpr Gpr) Gpr)
2806(rule (bsr_or_else ty src alt)
2807      (let ((bsr ProducesFlags (x64_bsr ty src))
2808            ;; Manually extract the result from the bsr, then ignore
2809            ;; it below, since we need to thread it into the cmove
2810            ;; before we pass the cmove to with_flags_reg.
2811            (bsr_result Gpr (produces_flags_get_reg bsr))
2812            (cmove ConsumesFlags (cmove ty (CC.Z) alt bsr_result)))
2813        (with_flags_reg (produces_flags_ignore bsr) cmove)))
2814
2815;; Helper for creating `bsf` instructions.
2816(decl x64_bsf (Type GprMem) ProducesFlags)
2817(rule (x64_bsf $I16 src) (asm_produce_flags (x64_bsfw_rm_raw src)))
2818(rule (x64_bsf $I32 src) (asm_produce_flags (x64_bsfl_rm_raw src)))
2819(rule (x64_bsf $I64 src) (asm_produce_flags (x64_bsfq_rm_raw src)))
2820
2821;; Helper for creating `bsf + cmov` instruction pairs that produce the
2822;; result of the `bsf`, or `alt` if the input was zero.
2823(decl bsf_or_else (Type Gpr Gpr) Gpr)
2824(rule (bsf_or_else ty src alt)
2825      (let ((bsf ProducesFlags (x64_bsf ty src))
2826            ;; Manually extract the result from the bsf, then ignore
2827            ;; it below, since we need to thread it into the cmove
2828            ;; before we pass the cmove to with_flags_reg.
2829            (bsf_result Gpr (produces_flags_get_reg bsf))
2830            (cmove ConsumesFlags (cmove ty (CC.Z) alt bsf_result)))
2831        (with_flags_reg (produces_flags_ignore bsf) cmove)))
2832
2833;; Helper for creating `blsi` instructions.
2834(decl x64_blsi (Type GprMem) Gpr)
2835(rule (x64_blsi $I32 src) (x64_blsil_vm src))
2836(rule (x64_blsi $I64 src) (x64_blsiq_vm src))
2837
2838;; Helper for creating `blsmsk` instructions.
2839(decl x64_blsmsk (Type GprMem) Gpr)
2840(rule (x64_blsmsk $I32 src) (x64_blsmskl_vm src))
2841(rule (x64_blsmsk $I64 src) (x64_blsmskq_vm src))
2842
2843;; Helper for creating `blsr` instructions.
2844(decl x64_blsr (Type GprMem) Gpr)
2845(rule (x64_blsr $I32 src) (x64_blsrl_vm src))
2846(rule (x64_blsr $I64 src) (x64_blsrq_vm src))
2847
2848;; Helper for creating `bt` instructions.
2849(decl x64_bt (Type GprMem Gpr) ProducesFlags)
2850(rule (x64_bt $I16 src1 src2) (x64_btw_mr src1 src2))
2851(rule (x64_bt $I32 src1 src2) (x64_btl_mr src1 src2))
2852(rule (x64_bt $I64 src1 src2) (x64_btq_mr src1 src2))
2853
2854;; Helper for creating `bt` instructions.
2855(decl x64_bt_imm (Type GprMem u8) ProducesFlags)
2856(rule (x64_bt_imm $I16 src imm) (x64_btw_mi src imm))
2857(rule (x64_bt_imm $I32 src imm) (x64_btl_mi src imm))
2858(rule (x64_bt_imm $I64 src imm) (x64_btq_mi src imm))
2859
2860;; Helper for creating `sarx` instructions.
2861(decl x64_sarx (Type GprMem Gpr) Gpr)
2862(rule (x64_sarx $I32 val amt) (x64_sarxl_rmv val amt))
2863(rule (x64_sarx $I64 val amt) (x64_sarxq_rmv val amt))
2864
2865;; Helper for creating `shrx` instructions.
2866(decl x64_shrx (Type GprMem Gpr) Gpr)
2867(rule (x64_shrx $I32 val amt) (x64_shrxl_rmv val amt))
2868(rule (x64_shrx $I64 val amt) (x64_shrxq_rmv val amt))
2869
2870;; Helper for creating `shlx` instructions.
2871(decl x64_shlx (Type GprMem Gpr) Gpr)
2872(rule (x64_shlx $I32 val amt) (x64_shlxl_rmv val amt))
2873(rule (x64_shlx $I64 val amt) (x64_shlxq_rmv val amt))
2874
2875;; Helper for creating `rorx` instructions.
2876(decl x64_rorx (Type GprMem u8) Gpr)
2877(rule (x64_rorx $I32 src imm) (x64_rorxl_rmi src imm))
2878(rule (x64_rorx $I64 src imm) (x64_rorxq_rmi src imm))
2879
2880;; Helper for creating `popcnt` instructions.
2881(decl x64_popcnt (Type GprMem) Gpr)
2882(rule (x64_popcnt $I16 src) (x64_popcntw_rm src))
2883(rule (x64_popcnt $I32 src) (x64_popcntl_rm src))
2884(rule (x64_popcnt $I64 src) (x64_popcntq_rm src))
2885
2886;; Helpers for creating `min*` instructions.
2887(decl x64_minss (Xmm XmmMem) Xmm)
2888(rule (x64_minss src1 src2) (x64_minss_a_or_avx src1 src2))
2889
2890(decl x64_minsd (Xmm XmmMem) Xmm)
2891(rule (x64_minsd src1 src2) (x64_minsd_a_or_avx src1 src2))
2892
2893(decl x64_minps (Xmm XmmMem) Xmm)
2894(rule (x64_minps src1 src2) (x64_minps_a_or_avx src1 src2))
2895
2896(decl x64_minpd (Xmm XmmMem) Xmm)
2897(rule (x64_minpd src1 src2) (x64_minpd_a_or_avx src1 src2))
2898
2899(decl x64_maxss (Xmm XmmMem) Xmm)
2900(rule (x64_maxss src1 src2) (x64_maxss_a_or_avx src1 src2))
2901
2902(decl x64_maxsd (Xmm XmmMem) Xmm)
2903(rule (x64_maxsd src1 src2) (x64_maxsd_a_or_avx src1 src2))
2904
2905(decl x64_maxps (Xmm XmmMem) Xmm)
2906(rule (x64_maxps src1 src2) (x64_maxps_a_or_avx src1 src2))
2907
2908(decl x64_maxpd (Xmm XmmMem) Xmm)
2909(rule (x64_maxpd src1 src2) (x64_maxpd_a_or_avx src1 src2))
2910
2911;; Helper for creating `vfmadd213*` instructions
2912(decl x64_vfmadd213 (Type Xmm Xmm XmmMem) Xmm)
2913(rule (x64_vfmadd213 $F32 a b c)   (x64_vfmadd213ss_a a b c))
2914(rule (x64_vfmadd213 $F64 a b c)   (x64_vfmadd213sd_a a b c))
2915(rule (x64_vfmadd213 $F32X4 a b c) (x64_vfmadd213ps_a a b c))
2916(rule (x64_vfmadd213 $F64X2 a b c) (x64_vfmadd213pd_a a b c))
2917
2918;; Helper for creating `vfmadd132*` instructions
2919(decl x64_vfmadd132 (Type Xmm Xmm XmmMem) Xmm)
2920(rule (x64_vfmadd132 $F32 a b c)   (x64_vfmadd132ss_a a b c))
2921(rule (x64_vfmadd132 $F64 a b c)   (x64_vfmadd132sd_a a b c))
2922(rule (x64_vfmadd132 $F32X4 a b c) (x64_vfmadd132ps_a a b c))
2923(rule (x64_vfmadd132 $F64X2 a b c) (x64_vfmadd132pd_a a b c))
2924
2925;; Helper for creating `vfnmadd213*` instructions
2926(decl x64_vfnmadd213 (Type Xmm Xmm XmmMem) Xmm)
2927(rule (x64_vfnmadd213 $F32 a b c)   (x64_vfnmadd213ss_a a b c))
2928(rule (x64_vfnmadd213 $F64 a b c)   (x64_vfnmadd213sd_a a b c))
2929(rule (x64_vfnmadd213 $F32X4 a b c) (x64_vfnmadd213ps_a a b c))
2930(rule (x64_vfnmadd213 $F64X2 a b c) (x64_vfnmadd213pd_a a b c))
2931
2932;; Helper for creating `vfnmadd132*` instructions
2933(decl x64_vfnmadd132 (Type Xmm Xmm XmmMem) Xmm)
2934(rule (x64_vfnmadd132 $F32 a b c)   (x64_vfnmadd132ss_a a b c))
2935(rule (x64_vfnmadd132 $F64 a b c)   (x64_vfnmadd132sd_a a b c))
2936(rule (x64_vfnmadd132 $F32X4 a b c) (x64_vfnmadd132ps_a a b c))
2937(rule (x64_vfnmadd132 $F64X2 a b c) (x64_vfnmadd132pd_a a b c))
2938
2939;; Helper for creating `vfmsub213*` instructions
2940(decl x64_vfmsub213 (Type Xmm Xmm XmmMem) Xmm)
2941(rule (x64_vfmsub213 $F32 a b c)   (x64_vfmsub213ss_a a b c))
2942(rule (x64_vfmsub213 $F64 a b c)   (x64_vfmsub213sd_a a b c))
2943(rule (x64_vfmsub213 $F32X4 a b c) (x64_vfmsub213ps_a a b c))
2944(rule (x64_vfmsub213 $F64X2 a b c) (x64_vfmsub213pd_a a b c))
2945
2946;; Helper for creating `vfmsub132*` instructions
2947(decl x64_vfmsub132 (Type Xmm Xmm XmmMem) Xmm)
2948(rule (x64_vfmsub132 $F32 a b c)   (x64_vfmsub132ss_a a b c))
2949(rule (x64_vfmsub132 $F64 a b c)   (x64_vfmsub132sd_a a b c))
2950(rule (x64_vfmsub132 $F32X4 a b c) (x64_vfmsub132ps_a a b c))
2951(rule (x64_vfmsub132 $F64X2 a b c) (x64_vfmsub132pd_a a b c))
2952
2953;; Helper for creating `vfnmsub213*` instructions
2954(decl x64_vfnmsub213 (Type Xmm Xmm XmmMem) Xmm)
2955(rule (x64_vfnmsub213 $F32 a b c)   (x64_vfnmsub213ss_a a b c))
2956(rule (x64_vfnmsub213 $F64 a b c)   (x64_vfnmsub213sd_a a b c))
2957(rule (x64_vfnmsub213 $F32X4 a b c) (x64_vfnmsub213ps_a a b c))
2958(rule (x64_vfnmsub213 $F64X2 a b c) (x64_vfnmsub213pd_a a b c))
2959
2960;; Helper for creating `vfnmsub132*` instructions
2961(decl x64_vfnmsub132 (Type Xmm Xmm XmmMem) Xmm)
2962(rule (x64_vfnmsub132 $F32 a b c)   (x64_vfnmsub132ss_a a b c))
2963(rule (x64_vfnmsub132 $F64 a b c)   (x64_vfnmsub132sd_a a b c))
2964(rule (x64_vfnmsub132 $F32X4 a b c) (x64_vfnmsub132ps_a a b c))
2965(rule (x64_vfnmsub132 $F64X2 a b c) (x64_vfnmsub132pd_a a b c))
2966
2967;; Note, the `vfmsub231` and `vfnmsub231*` instructions are omitted, because
2968;; instruction selection happens before register allocation and therefore there
2969;; is no benefit to a a third permutation
2970
2971;; Helper for creating `sqrtss` instructions.
2972;;
2973;; NB: the square-root operation technically only has one operand but this
2974;; instruction has two. This is to reflect how the square root operation copies
2975;; the upper bits of the first register and only performs the square root
2976;; operation on the low bits of the second register. This introduces
2977;; a data-dependency on the contents of the first register which is modeled
2978;; here.
2979(decl x64_sqrtss (Xmm XmmMem) Xmm)
2980(rule (x64_sqrtss x y) (x64_sqrtss_a_or_avx x y))
2981
2982;; Helper for creating `sqrtsd` instructions.
2983;;
2984;; NB: see `x64_sqrtss` for explanation of why this has two args.
2985(decl x64_sqrtsd (Xmm XmmMem) Xmm)
2986(rule 0 (x64_sqrtsd x y) (x64_sqrtsd_a_or_avx x y))
2987
2988;; Helper for creating `sqrtps` instructions.
2989(decl x64_sqrtps (XmmMem) Xmm)
2990(rule (x64_sqrtps x) (x64_sqrtps_a_or_avx x))
2991
2992;; Helper for creating `sqrtpd` instructions.
2993(decl x64_sqrtpd (XmmMem) Xmm)
2994(rule (x64_sqrtpd x) (x64_sqrtpd_a_or_avx x))
2995
2996;; Helper for creating `reciprocal` instructions.
2997;;
2998;; Helper for creating `rcpps` instructions.
2999(decl x64_rcpps (XmmMem) Xmm)
3000(rule (x64_rcpps x) (x64_rcpps_rm_or_avx x))
3001
3002;; Helper for creating `rcpss` instructions.
3003(decl x64_rcpss (XmmMem) Xmm)
3004(rule (x64_rcpss x) (x64_rcpss_rm x))
3005
3006;; Helper for creating `vrsqrtss` instructions.
3007(decl x64_vrcpss (Xmm XmmMem) Xmm)
3008(rule (x64_vrcpss x y) (x64_vrcpss_rvm x y))
3009
3010;; Helper for creating `rsqrtps` instructions.
3011(decl x64_rsqrtps (XmmMem) Xmm)
3012(rule (x64_rsqrtps x) (x64_rsqrtps_rm_or_avx x))
3013
3014;; Helper for creating `rsqrtss` instructions.
3015(decl x64_rsqrtss (XmmMem) Xmm)
3016(rule (x64_rsqrtss x) (x64_rsqrtss_rm x))
3017
3018;; Helper for creating `vrsqrtss` instructions.
3019(decl x64_vrsqrtss (Xmm XmmMem) Xmm)
3020(rule (x64_vrsqrtss x y) (x64_vrsqrtss_rvm x y))
3021
3022
3023;; Helper for creating `cvtss2sd` instructions.
3024;;
3025;; NB: see `x64_sqrtss` for why this has two args (same reasoning, different op)
3026(decl x64_cvtss2sd (Xmm XmmMem) Xmm)
3027(rule 1 (x64_cvtss2sd x y)
3028        (if-let true (has_avx))
3029        (x64_vcvtss2sd_b x y))
3030(rule 0 (x64_cvtss2sd x y) (x64_cvtss2sd_a x y))
3031
3032;; Helper for creating `cvtsd2ss` instructions.
3033;;
3034;; NB: see `x64_sqrtss` for why this has two args (same reasoning, different op)
3035(decl x64_cvtsd2ss (Xmm XmmMem) Xmm)
3036(rule 1 (x64_cvtsd2ss x y)
3037        (if-let true (has_avx))
3038        (x64_vcvtsd2ss_b x y))
3039(rule 0 (x64_cvtsd2ss x y) (x64_cvtsd2ss_a x y))
3040
3041;; Helper for creating `cvtdq2ps` instructions.
3042(decl x64_cvtdq2ps (XmmMem) Xmm)
3043(rule 1 (x64_cvtdq2ps x)
3044        (if-let true (has_avx))
3045        (x64_vcvtdq2ps_a x))
3046(rule (x64_cvtdq2ps x) (x64_cvtdq2ps_a x))
3047
3048;; Helper for creating `cvtps2pd` instructions.
3049(decl x64_cvtps2pd (XmmMem) Xmm)
3050(rule 1 (x64_cvtps2pd x)
3051        (if-let true (has_avx))
3052        (x64_vcvtps2pd_a x))
3053(rule 0 (x64_cvtps2pd x) (x64_cvtps2pd_a x))
3054
3055;; Helper for creating `cvtpd2ps` instructions.
3056(decl x64_cvtpd2ps (XmmMem) Xmm)
3057(rule 1 (x64_cvtpd2ps x)
3058        (if-let true (has_avx))
3059        (x64_vcvtpd2ps_a x))
3060(rule 0 (x64_cvtpd2ps x) (x64_cvtpd2ps_a x))
3061
3062;; Helper for creating `cvtdq2pd` instructions.
3063(decl x64_cvtdq2pd (XmmMem) Xmm)
3064(rule 1 (x64_cvtdq2pd x)
3065        (if-let true (has_avx))
3066        (x64_vcvtdq2pd_a x))
3067(rule 0 (x64_cvtdq2pd x) (x64_cvtdq2pd_a x))
3068
3069;; Helper for creating `cvtsi2ss` instructions.
3070(decl x64_cvtsi2ss (Type Xmm GprMem) Xmm)
3071(rule 1 (x64_cvtsi2ss $I32 x y)
3072        (if-let true (has_avx))
3073        (x64_vcvtsi2ssl_b x y))
3074(rule 1 (x64_cvtsi2ss $I64 x y)
3075        (if-let true (has_avx))
3076        (x64_vcvtsi2ssq_b x y))
3077(rule 0 (x64_cvtsi2ss $I32 x y) (x64_cvtsi2ssl_a x y))
3078(rule 0 (x64_cvtsi2ss $I64 x y) (x64_cvtsi2ssq_a x y))
3079
3080;; Helper for creating `cvtsi2sd` instructions.
3081(decl x64_cvtsi2sd (Type Xmm GprMem) Xmm)
3082(rule 1 (x64_cvtsi2sd $I32 x y)
3083        (if-let true (has_avx))
3084        (x64_vcvtsi2sdl_b x y))
3085(rule 1 (x64_cvtsi2sd $I64 x y)
3086        (if-let true (has_avx))
3087        (x64_vcvtsi2sdq_b x y))
3088(rule 0 (x64_cvtsi2sd $I32 x y) (x64_cvtsi2sdl_a x y))
3089(rule 0 (x64_cvtsi2sd $I64 x y) (x64_cvtsi2sdq_a x y))
3090
3091;; Helper for creating `cvttps2dq` instructions.
3092(decl x64_cvttps2dq (XmmMem) Xmm)
3093(rule 1 (x64_cvttps2dq x)
3094        (if-let true (has_avx))
3095        (x64_vcvttps2dq_a x))
3096(rule 0 (x64_cvttps2dq x) (x64_cvttps2dq_a x))
3097
3098;; Helper for creating `cvttpd2dq` instructions.
3099(decl x64_cvttpd2dq (XmmMem) Xmm)
3100(rule 1 (x64_cvttpd2dq x)
3101        (if-let true (has_avx))
3102        (x64_vcvttpd2dq_a x))
3103(rule 0 (x64_cvttpd2dq x) (x64_cvttpd2dq_a x))
3104
3105;; Helpers for creating `pcmpeq*` instructions.
3106(decl x64_pcmpeq (Type Xmm XmmMem) Xmm)
3107(rule (x64_pcmpeq $I8X16 x y) (x64_pcmpeqb x y))
3108(rule (x64_pcmpeq $I16X8 x y) (x64_pcmpeqw x y))
3109(rule (x64_pcmpeq $I32X4 x y) (x64_pcmpeqd x y))
3110(rule (x64_pcmpeq $I64X2 x y)
3111      (if-let true (has_sse41))
3112      (x64_pcmpeqq x y))
3113
3114;; Without SSE 4.1 there's no access to `pcmpeqq`, so it's emulated by comparing
3115;; 32-bit lanes instead. The upper and lower halves of the 32-bit comparison are
3116;; swapped and then these two results are and'd together. This way only if both
3117;; 32-bit values were equal is the result all ones, otherwise the result is
3118;; all zeros if either 32-bit comparison was zero.
3119(rule -1 (x64_pcmpeq $I64X2 x y)
3120         (let ((cmp32         Xmm (x64_pcmpeqd x y))
3121               (cmp32_swapped Xmm (x64_pshufd cmp32 0b10_11_00_01)))
3122            (x64_pand cmp32 cmp32_swapped)))
3123
3124;; Helpers for creating `pcmpeq*` instructions.
3125(decl x64_pcmpeqb (Xmm XmmMem) Xmm)
3126(rule (x64_pcmpeqb x y) (x64_pcmpeqb_a_or_avx x y))
3127
3128(decl x64_pcmpeqw (Xmm XmmMem) Xmm)
3129(rule (x64_pcmpeqw x y) (x64_pcmpeqw_a_or_avx x y))
3130
3131(decl x64_pcmpeqd (Xmm XmmMem) Xmm)
3132(rule (x64_pcmpeqd x y) (x64_pcmpeqd_a_or_avx x y))
3133
3134(decl x64_pcmpeqq (Xmm XmmMem) Xmm)
3135(rule (x64_pcmpeqq x y) (x64_pcmpeqq_a_or_avx x y))
3136
3137;; Helpers for creating `pcmpgt*` instructions.
3138(decl x64_pcmpgt (Type Xmm XmmMem) Xmm)
3139(rule (x64_pcmpgt $I8X16 x y) (x64_pcmpgtb_a_or_avx x y))
3140(rule (x64_pcmpgt $I16X8 x y) (x64_pcmpgtw_a_or_avx x y))
3141(rule (x64_pcmpgt $I32X4 x y) (x64_pcmpgtd_a_or_avx x y))
3142
3143;; AVX has a single-instruction lowering; we do not use the `or_avx` suffix so
3144;; we can match a non-AVX/SSE4.2 below.
3145(rule 2 (x64_pcmpgt $I64X2 x y)
3146        (if-let true (has_avx))
3147        (x64_vpcmpgtq_b x y))
3148;; SSE4.2 also gives a single-instruction for this lowering, but prior to that
3149;; it's a bit more complicated.
3150(rule 1 (x64_pcmpgt $I64X2 x y)
3151        (if-let true (use_sse42))
3152        (x64_pcmpgtq_a x y))
3153
3154;; Without SSE4.2 a 64-bit comparison is expanded to a number of instructions.
3155;; The basic idea is to delegate to a 32-bit comparison and work with the
3156;; results from there. The comparison to execute is:
3157;;
3158;;    [ xhi ][ xlo ] > [ yhi ][ ylo ]
3159;;
3160;; If xhi != yhi, then the result is whatever the result of that comparison is.
3161;; If xhi == yhi, then the result is the unsigned comparison of xlo/ylo since
3162;; the 64-bit value is positive. To achieve this as part of the same comparison
3163;; the upper bit of `xlo` and `ylo` is flipped to change the sign when compared
3164;; as a 32-bit signed number. The result here is then:
3165;;
3166;; * if xlo and yhi had the same upper bit, then the unsigned comparison should
3167;;   be the same as comparing the flipped versions as signed.
3168;; * if xlo had an upper bit of 0 and ylo had an upper bit of 1, then xlo > ylo
3169;;   is false. When flipping the bits xlo becomes negative and ylo becomes
3170;;   positive when compared as 32-bits, so the result is the same.
3171;; * if xlo had an upper bit of 1 and ylo had an upper bit of 0, then xlo > ylo
3172;;   is true. When flipping the bits xlo becomes positive and ylo becomes
3173;;   negative when compared as 32-bits, so the result is the same.
3174;;
3175;; Given all that the sequence here is to flip the upper bits of xlo and ylo,
3176;; then compare the masked results for equality and for gt. If the upper 32-bits
3177;; are not equal then the gt result for the upper bits is used. If the upper
3178;; 32-bits are equal then the lower 32-bits comparison is used instead.
3179(rule 0 (x64_pcmpgt $I64X2 x y)
3180        (let (
3181            (mask Xmm (x64_movdqu_load (emit_u128_le_const 0x00000000_80000000_00000000_80000000)))
3182            (x_masked           Xmm (x64_pxor mask x))
3183            (y_masked           Xmm (x64_pxor mask y))
3184            (cmp32              Xmm (x64_pcmpgtd_a x_masked y_masked))
3185            (low_halves_gt      Xmm (x64_pshufd cmp32 0xa0))
3186            (high_halves_gt     Xmm (x64_pshufd cmp32 0xf5))
3187            (cmp_eq             Xmm (x64_pcmpeqd x_masked y_masked))
3188            (high_halves_eq     Xmm (x64_pshufd cmp_eq 0xf5))
3189            (low_gt_and_high_eq Xmm (x64_pand low_halves_gt high_halves_eq))
3190          )
3191          (x64_por low_gt_and_high_eq high_halves_gt)))
3192
3193(decl x64_add_mem (Type SyntheticAmode Value) SideEffectNoResult)
3194(spec (x64_add_mem ty addr val)
3195      (provide (= result (store_effect
3196                        (extract 79 64 addr)
3197                        ty
3198                        (conv_to ty (bvadd (load_effect (extract 79 64 addr) ty (extract 63 0 addr)) (conv_to ty val)))
3199                        (extract 63 0 addr))
3200            )
3201      )
3202       (require (or (= ty 32) (= ty 64)))
3203)
3204
3205;; `add mem, reg`
3206(rule 0 (x64_add_mem $I8 addr val) (x64_addb_mr_mem addr val))
3207(rule 0 (x64_add_mem $I16 addr val) (x64_addw_mr_mem addr val))
3208(rule 0 (x64_add_mem $I32 addr val) (x64_addl_mr_mem addr val))
3209(rule 0 (x64_add_mem $I64 addr val) (x64_addq_mr_mem addr val))
3210
3211;; `add mem, imm`
3212(rule 1 (x64_add_mem $I8 addr (u8_from_iconst val)) (x64_addb_mi_mem addr val))
3213(rule 1 (x64_add_mem $I16 addr (u16_from_iconst val)) (x64_addw_mi_mem addr val))
3214(rule 1 (x64_add_mem $I32 addr (u32_from_iconst val)) (x64_addl_mi_mem addr val))
3215(rule 1 (x64_add_mem $I64 addr (i32_from_iconst val)) (x64_addq_mi_sxl_mem addr val))
3216(rule 2 (x64_add_mem $I32 addr (i8_from_iconst val)) (x64_addl_mi_sxb_mem addr val))
3217(rule 2 (x64_add_mem $I64 addr (i8_from_iconst val)) (x64_addq_mi_sxb_mem addr val))
3218
3219(decl x64_sub_mem (Type SyntheticAmode Value) SideEffectNoResult)
3220
3221;; `sub mem, reg`
3222(rule 0 (x64_sub_mem $I8 addr val) (x64_subb_mr_mem addr val))
3223(rule 0 (x64_sub_mem $I16 addr val) (x64_subw_mr_mem addr val))
3224(rule 0 (x64_sub_mem $I32 addr val) (x64_subl_mr_mem addr val))
3225(rule 0 (x64_sub_mem $I64 addr val) (x64_subq_mr_mem addr val))
3226
3227;; `sub mem, imm`
3228(rule 1 (x64_sub_mem $I8 addr (u8_from_iconst val)) (x64_subb_mi_mem addr val))
3229(rule 1 (x64_sub_mem $I16 addr (u16_from_iconst val)) (x64_subw_mi_mem addr val))
3230(rule 1 (x64_sub_mem $I32 addr (u32_from_iconst val)) (x64_subl_mi_mem addr val))
3231(rule 1 (x64_sub_mem $I64 addr (i32_from_iconst val)) (x64_subq_mi_sxl_mem addr val))
3232(rule 2 (x64_sub_mem $I32 addr (i8_from_iconst val)) (x64_subl_mi_sxb_mem addr val))
3233(rule 2 (x64_sub_mem $I64 addr (i8_from_iconst val)) (x64_subq_mi_sxb_mem addr val))
3234
3235(decl x64_and_mem (Type SyntheticAmode Value) SideEffectNoResult)
3236
3237;; `and mem, imm`
3238(rule (x64_and_mem $I8 addr val) (x64_andb_mr_mem addr val))
3239(rule (x64_and_mem $I16 addr val) (x64_andw_mr_mem addr val))
3240(rule (x64_and_mem $I32 addr val) (x64_andl_mr_mem addr val))
3241(rule (x64_and_mem $F32 addr val) (x64_andl_mr_mem addr val))
3242(rule (x64_and_mem $I64 addr val) (x64_andq_mr_mem addr val))
3243(rule (x64_and_mem $F64 addr val) (x64_andq_mr_mem addr val))
3244
3245;; `and mem, imm`
3246(rule 1 (x64_and_mem $I8 addr (u8_from_iconst val)) (x64_andb_mi_mem addr val))
3247(rule 1 (x64_and_mem $I16 addr (u16_from_iconst val)) (x64_andw_mi_mem addr val))
3248(rule 1 (x64_and_mem $I32 addr (u32_from_iconst val)) (x64_andl_mi_mem addr val))
3249(rule 1 (x64_and_mem $I64 addr (i32_from_iconst val)) (x64_andq_mi_sxl_mem addr val))
3250(rule 2 (x64_and_mem $I32 addr (i8_from_iconst val)) (x64_andl_mi_sxb_mem addr val))
3251(rule 2 (x64_and_mem $I64 addr (i8_from_iconst val)) (x64_andq_mi_sxb_mem addr val))
3252
3253(decl x64_or_mem (Type SyntheticAmode Value) SideEffectNoResult)
3254
3255;; `or mem, reg`
3256(rule 0 (x64_or_mem $I8 addr val) (x64_orb_mr_mem addr val))
3257(rule 0 (x64_or_mem $I16 addr val) (x64_orw_mr_mem addr val))
3258(rule 0 (x64_or_mem $I32 addr val) (x64_orl_mr_mem addr val))
3259(rule 0 (x64_or_mem $F32 addr val) (x64_orl_mr_mem addr val))
3260(rule 0 (x64_or_mem $I64 addr val) (x64_orq_mr_mem addr val))
3261(rule 0 (x64_or_mem $F64 addr val) (x64_orq_mr_mem addr val))
3262
3263;; `or mem, imm`
3264(rule 1 (x64_or_mem $I8 addr (u8_from_iconst val)) (x64_orb_mi_mem addr val))
3265(rule 1 (x64_or_mem $I16 addr (u16_from_iconst val)) (x64_orw_mi_mem addr val))
3266(rule 1 (x64_or_mem $I32 addr (u32_from_iconst val)) (x64_orl_mi_mem addr val))
3267(rule 1 (x64_or_mem $I64 addr (i32_from_iconst val)) (x64_orq_mi_sxl_mem addr val))
3268(rule 2 (x64_or_mem $I32 addr (i8_from_iconst val)) (x64_orl_mi_sxb_mem addr val))
3269(rule 2 (x64_or_mem $I64 addr (i8_from_iconst val)) (x64_orq_mi_sxb_mem addr val))
3270
3271(decl x64_xor_mem (Type SyntheticAmode Value) SideEffectNoResult)
3272
3273;; `xor mem, reg`
3274(rule 0 (x64_xor_mem $I8 addr val) (x64_xorb_mr_mem addr val))
3275(rule 0 (x64_xor_mem $I16 addr val) (x64_xorw_mr_mem addr val))
3276(rule 0 (x64_xor_mem $I32 addr val) (x64_xorl_mr_mem addr val))
3277(rule 0 (x64_xor_mem $F32 addr val) (x64_xorl_mr_mem addr val))
3278(rule 0 (x64_xor_mem $I64 addr val) (x64_xorq_mr_mem addr val))
3279(rule 0 (x64_xor_mem $F64 addr val) (x64_xorq_mr_mem addr val))
3280
3281;; `xor mem, imm`
3282(rule 1 (x64_xor_mem $I8 addr (u8_from_iconst val)) (x64_xorb_mi_mem addr val))
3283(rule 1 (x64_xor_mem $I16 addr (u16_from_iconst val)) (x64_xorw_mi_mem addr val))
3284(rule 1 (x64_xor_mem $I32 addr (u32_from_iconst val)) (x64_xorl_mi_mem addr val))
3285(rule 1 (x64_xor_mem $I64 addr (i32_from_iconst val)) (x64_xorq_mi_sxl_mem addr val))
3286(rule 2 (x64_xor_mem $I32 addr (i8_from_iconst val)) (x64_xorl_mi_sxb_mem addr val))
3287(rule 2 (x64_xor_mem $I64 addr (i8_from_iconst val)) (x64_xorq_mi_sxb_mem addr val))
3288
3289;; Trap if the condition code supplied is set.
3290(decl trap_if (CC TrapCode) ConsumesFlags)
3291(rule (trap_if cc tc)
3292      (ConsumesFlags.ConsumesFlagsSideEffect (MInst.TrapIf cc tc)))
3293
3294;; Trap if both of the condition codes supplied are set.
3295(decl trap_if_and (CC CC TrapCode) ConsumesFlags)
3296(rule (trap_if_and cc1 cc2 tc)
3297      (ConsumesFlags.ConsumesFlagsSideEffect (MInst.TrapIfAnd cc1 cc2 tc)))
3298
3299;; Trap if either of the condition codes supplied are set.
3300(decl trap_if_or (CC CC TrapCode) ConsumesFlags)
3301(rule (trap_if_or cc1 cc2 tc)
3302      (ConsumesFlags.ConsumesFlagsSideEffect (MInst.TrapIfOr cc1 cc2 tc)))
3303
3304;; Helper for creating `movddup` instructions
3305(decl x64_movddup (XmmMem) Xmm)
3306(rule (x64_movddup src) (x64_movddup_a src))
3307(rule 1 (x64_movddup src)
3308        (if-let true (has_avx))
3309        (x64_vmovddup_a src))
3310
3311;; Helpers for creating `vpbroadcast*` instructions.
3312(decl x64_vpbroadcastb (XmmMem) Xmm)
3313(rule (x64_vpbroadcastb src) (x64_vpbroadcastb_a src))
3314
3315(decl x64_vpbroadcastw (XmmMem) Xmm)
3316(rule (x64_vpbroadcastw src) (x64_vpbroadcastw_a src))
3317
3318(decl x64_vpbroadcastd (XmmMem) Xmm)
3319(rule (x64_vpbroadcastd src) (x64_vpbroadcastd_a src))
3320
3321(decl x64_vbroadcastss (XmmMem) Xmm)
3322(rule 1 (x64_vbroadcastss (is_xmm src)) (x64_vbroadcastss_a_r src))
3323(rule 0 (x64_vbroadcastss (is_mem src)) (x64_vbroadcastss_a_m src))
3324
3325;;;; Jumps ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3326
3327;; Unconditional jump.
3328(decl jmp_known (MachLabel) SideEffectNoResult)
3329(rule (jmp_known target)
3330      (SideEffectNoResult.Inst (MInst.JmpKnown target)))
3331
3332;; Conditional jump based on the condition code.
3333(decl jmp_cond (CC MachLabel MachLabel) ConsumesFlags)
3334(rule (jmp_cond cc taken not_taken)
3335      (ConsumesFlags.ConsumesFlagsSideEffect (MInst.JmpCond cc taken not_taken)))
3336
3337;; Conditional jump based on the OR of two condition codes.
3338(decl jmp_cond_or (CC CC MachLabel MachLabel) ConsumesFlags)
3339(rule (jmp_cond_or cc1 cc2 taken not_taken)
3340      (ConsumesFlags.ConsumesFlagsSideEffect (MInst.JmpCondOr cc1 cc2 taken not_taken)))
3341
3342;; Conditional jump based on a `CondResult`
3343;;
3344;; Recursion: at most to convert `And` into `Or`.
3345(decl rec jmp_cond_result (CondResult MachLabel MachLabel) SideEffectNoResult)
3346(rule (jmp_cond_result (CondResult.CC producer cc) taken not_taken)
3347      (with_flags_side_effect producer (jmp_cond cc taken not_taken)))
3348(rule (jmp_cond_result cond @ (CondResult.And _ _ _) taken not_taken)
3349      (jmp_cond_result (cond_invert cond) not_taken taken))
3350(rule (jmp_cond_result (CondResult.Or producer cc1 cc2) taken not_taken)
3351      (with_flags_side_effect producer
3352                              (jmp_cond_or cc1 cc2 taken not_taken)))
3353
3354;; Emit the compound instruction that does:
3355;;
3356;; lea $jt, %rA
3357;; movsbl [%rA, %rIndex, 2], %rB
3358;; add %rB, %rA
3359;; j *%rA
3360;; [jt entries]
3361;;
3362;; This must be *one* instruction in the vcode because we cannot allow regalloc
3363;; to insert any spills/fills in the middle of the sequence; otherwise, the
3364;; lea PC-rel offset to the jumptable would be incorrect.  (The alternative
3365;; is to introduce a relocation pass for inlined jumptables, which is much
3366;; worse.)
3367(decl jmp_table_seq (Type Gpr MachLabel BoxVecMachLabel) SideEffectNoResult)
3368(rule (jmp_table_seq ty idx default_target jt_targets)
3369      (let (
3370            ;; This temporary is used as a signed integer of 64-bits (to hold
3371            ;; addresses).
3372            (tmp1 WritableGpr (temp_writable_gpr))
3373
3374            ;; This temporary is used as a signed integer of 32-bits (for the
3375            ;; wasm-table index) and then 64-bits (address addend). The small
3376            ;; lie about the I64 type is benign, since the temporary is dead
3377            ;; after this instruction (and its Cranelift type is thus unused).
3378            (tmp2 WritableGpr (temp_writable_gpr)))
3379
3380          (SideEffectNoResult.Inst
3381            (MInst.JmpTableSeq idx tmp1 tmp2 default_target jt_targets))))
3382
3383;;;; Comparisons ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3384
3385;; Representation of the result of a conditional instruction.
3386;;
3387;; Each variant here has what produces some condition flags in addition to
3388;; what condition code is being tested as a result of whatever produced the
3389;; flags.
3390;;
3391;; This type is intended to be a "narrow waist" for anything producing a
3392;; conditional which `icmp` might flow into for example. The `is_nonzero_cmp`
3393;; constructor is the main constructor of this type which takes any arbitrary
3394;; value used in a conditional-like location. There are further refined
3395;; constructors such as `emit_{cmp,fcmp}` which work specifically on the shapes
3396;; of `icmp` and `fcmp` CLIF instructions. Everything produces this type, and
3397;; then decisions about what instructions to emit flow from this type.
3398(type CondResult
3399      (enum
3400        ;; The given condition code must be set.
3401        (CC (producer ProducesFlags) (cc CC))
3402
3403        ;; Both condition codes must be set.
3404        (And (producer ProducesFlags) (cc1 CC) (cc2 CC))
3405
3406        ;; Either of the conditions codes must be set.
3407        (Or (producer ProducesFlags) (cc1 CC) (cc2 CC))))
3408
3409;; Inverts a `CondResult` to have the opposite meaning.
3410(decl cond_invert (CondResult) CondResult)
3411(rule (cond_invert (CondResult.CC flags cc)) (CondResult.CC flags (cc_invert cc)))
3412(rule (cond_invert (CondResult.Or flags cc1 cc2)) (CondResult.And flags (cc_invert cc1) (cc_invert cc2)))
3413(rule (cond_invert (CondResult.And flags cc1 cc2)) (CondResult.Or flags (cc_invert cc1) (cc_invert cc2)))
3414
3415;; Converts a `Value` to a `CondResult` with the condition being tested if
3416;; `Value` is nonzero.
3417;;
3418;; Note that this is used as the base entry case for instruction lowering such
3419;; as `select` and `brif`. The `Value` here is expected to, via CLIF validation,
3420;; have an integer type (and it can be I128)
3421(decl is_nonzero (Value) CondResult)
3422
3423;; Base case: fits in one GPR, use `x64_test`
3424(rule (is_nonzero val @ (value_type (is_single_register_gpr_type ty)))
3425  (let ((gpr Gpr val)) (CondResult.CC (x64_test ty gpr gpr) (CC.NZ))))
3426
3427;; Base case: i128
3428(rule 1 (is_nonzero val @ (value_type $I128))
3429      (let ((lo Gpr (value_regs_get_gpr val 0))
3430            (hi Gpr (value_regs_get_gpr val 1)))
3431          (CondResult.CC
3432            (x64_produce_flags_side_effect (ProduceFlagsSideEffectOp.Or) $I64 lo hi)
3433            (CC.NZ))))
3434
3435;; Special case some instructions where lowerings directly produce condition
3436;; codes.
3437(rule 2 (is_nonzero (vall_true _ vec)) (is_vall_true vec))
3438(rule 2 (is_nonzero (vany_true _ vec)) (is_vany_true vec))
3439(rule 2 (is_nonzero (uextend _ (vall_true _ vec))) (is_vall_true vec))
3440(rule 2 (is_nonzero (uextend _ (vany_true _ vec))) (is_vany_true vec))
3441(rule 2 (is_nonzero (band _ a @ (value_type (ty_int (fits_in_64 ty))) b))
3442      (is_nonzero_band ty a b))
3443
3444
3445;; Like `is_nonzero` but with additional specializations for compare
3446;; operators. We break this out from `is_nonzero` because we want to
3447;; avoid unbounded recursion.
3448(decl is_nonzero_cmp (Value) CondResult)
3449
3450(rule 1 (is_nonzero_cmp (fcmp _ cc a b)) (emit_fcmp cc a b))
3451(rule 1 (is_nonzero_cmp (icmp _ cc a b)) (emit_cmp cc a b))
3452(rule 1 (is_nonzero_cmp (uextend _ (fcmp _ cc a b))) (emit_fcmp cc a b))
3453(rule 1 (is_nonzero_cmp (uextend _ (icmp _ cc a b))) (emit_cmp cc a b))
3454(rule 0 (is_nonzero_cmp val) (is_nonzero val))
3455
3456(decl is_nonzero_band (Type Value Value) CondResult)
3457(rule 0 (is_nonzero_band ty a b) (CondResult.CC (x64_test ty a b) (CC.NZ)))
3458
3459;; If a value is and'd with an immediate that has exactly one bit set then this
3460;; can pattern-match to the native `bt` instruction. Note that to have the
3461;; same semantics this requires that `a` is in a register which forces `bt` to
3462;; use modulo semantics for the second operand `b`, thus `put_in_gpr` is
3463;; manually used.
3464(rule 1 (is_nonzero_band (ty_32_or_64 ty) a (ishl _ (u64_from_iconst 1) b))
3465  (CondResult.CC (x64_bt ty (put_in_gpr a) b) (CC.B)))
3466
3467;; If a value is and'd one shifted by a variable value that matches `bt` as
3468;; well.
3469(rule 1 (is_nonzero_band $I64 a (u64_from_iconst (bt_imm n)))
3470  (CondResult.CC (x64_bt_imm $I64 a n) (CC.B)))
3471
3472;; If what we're testing against is a 32-bit integer then this is a candidate
3473;; for both the `test` and `bt` instructions (only `bt` if the integer as one
3474;; bit set). According to [1] the `test` instruction has a higher throughput
3475;; at least historically than the `bt` instruction so here `test` is explicitly
3476;; favored over `bt`, even if `bt` were applicable. Note that LLVM also looks to
3477;; favor `bt` as well.
3478;;
3479;; [1]: https://github.com/bytecodealliance/wasmtime/pull/11128#discussion_r2164888415
3480(rule 2 (is_nonzero_band ty a b @ (i32_from_iconst _))
3481  (CondResult.CC (x64_test ty a b) (CC.NZ)))
3482
3483;; Helper to test whether the `u64` input has a single bit set, and if so
3484;; yields the bit position of where that bit is set. Used in the lowering of
3485;; `x64_bt_imm` above.
3486(decl bt_imm (u8) u64)
3487(extern extractor bt_imm bt_imm)
3488
3489;; Lower a CondResult to a boolean value in a register.
3490(decl lower_cond_bool (CondResult) Gpr)
3491(rule (lower_cond_bool (CondResult.CC producer cc))
3492  (value_regs_get_gpr (with_flags producer (x64_setcc cc)) 0))
3493(rule (lower_cond_bool (CondResult.And producer cc1 cc2))
3494      (let ((maybe ValueRegs (with_flags producer
3495                                         (consumes_flags_concat
3496                                           (x64_setcc cc1)
3497                                           (x64_setcc cc2))))
3498            (maybe0 Gpr (value_regs_get_gpr maybe 0))
3499            (maybe1 Gpr (value_regs_get_gpr maybe 1)))
3500        (x64_and $I8 maybe0 maybe1)))
3501(rule (lower_cond_bool (CondResult.Or producer cc1 cc2))
3502      (let ((maybe ValueRegs (with_flags producer
3503                                         (consumes_flags_concat
3504                                           (x64_setcc cc1)
3505                                           (x64_setcc cc2))))
3506            (maybe0 Gpr (value_regs_get_gpr maybe 0))
3507            (maybe1 Gpr (value_regs_get_gpr maybe 1)))
3508        (x64_or $I8 maybe0 maybe1)))
3509
3510;; Helper to transform an `icmp` node into a `CondResult`.
3511;;
3512;; Note that via CLIF validation the two values here should have the same type.
3513(decl emit_cmp (IntCC Value Value) CondResult)
3514
3515;; For GPR-held values we only need to emit `CMP`.
3516(rule 0 (emit_cmp cc a @ (value_type ty) b) (CondResult.CC (x64_cmp ty a b) cc))
3517
3518;; As a special case, swap the arguments to the comparison when the LHS is a
3519;; constant. This ensures that we avoid moving the constant into a register when
3520;; performing the comparison.
3521(rule 1 (emit_cmp cc (and (simm32_from_value a) (value_type ty)) b)
3522        (CondResult.CC (x64_cmp ty b a) (intcc_swap_args cc)))
3523
3524;; Special case: use the test instruction for comparisons with 0.
3525(rule 2 (emit_cmp cc a @ (value_type ty) (u64_from_iconst 0))
3526      (let ((a Gpr (put_in_reg a)))
3527        (CondResult.CC (x64_test ty a a) cc)))
3528(rule 3 (emit_cmp cc (u64_from_iconst 0) b @ (value_type ty))
3529      (let ((b Gpr (put_in_reg b)))
3530        (CondResult.CC (x64_test ty b b) (intcc_swap_args cc))))
3531
3532;; For I128 values (held in two GPRs), the instruction sequences depend on what
3533;; kind of condition is tested.
3534(rule 4 (emit_cmp cc a @ (value_type $I128) b)
3535      (let ((a_lo Gpr (value_regs_get_gpr a 0))
3536            (a_hi Gpr (value_regs_get_gpr a 1))
3537            (b_lo Gpr (value_regs_get_gpr b 0))
3538            (b_hi Gpr (value_regs_get_gpr b 1)))
3539        (emit_cmp_i128 cc a_hi a_lo b_hi b_lo)))
3540
3541;; For direct equality comparisons to zero transform the other operand into a
3542;; nonzero comparison and then invert the whole conditional to test for zero.
3543(rule 5 (emit_cmp (IntCC.Equal) a (u64_from_iconst 0)) (cond_invert (is_nonzero a)))
3544(rule 6 (emit_cmp (IntCC.Equal) (u64_from_iconst 0) a) (cond_invert (is_nonzero a)))
3545(rule 5 (emit_cmp (IntCC.NotEqual) a (u64_from_iconst 0)) (is_nonzero a))
3546(rule 6 (emit_cmp (IntCC.NotEqual) (u64_from_iconst 0) a) (is_nonzero a))
3547
3548; Recursion: at most one to eliminate "or equal" cases.
3549(decl rec emit_cmp_i128 (CC Gpr Gpr Gpr Gpr) CondResult)
3550;; Eliminate cases which compare something "or equal" by swapping arguments.
3551(rule 2 (emit_cmp_i128 (CC.NLE) a_hi a_lo b_hi b_lo)
3552        (emit_cmp_i128 (CC.L)   b_hi b_lo a_hi a_lo))
3553(rule 2 (emit_cmp_i128 (CC.LE)  a_hi a_lo b_hi b_lo)
3554        (emit_cmp_i128 (CC.NL)  b_hi b_lo a_hi a_lo))
3555(rule 2 (emit_cmp_i128 (CC.NBE) a_hi a_lo b_hi b_lo)
3556        (emit_cmp_i128 (CC.B)   b_hi b_lo a_hi a_lo))
3557(rule 2 (emit_cmp_i128 (CC.BE)  a_hi a_lo b_hi b_lo)
3558        (emit_cmp_i128 (CC.NB)  b_hi b_lo a_hi a_lo))
3559
3560;; 128-bit strict equality/inequality can't be easily tested using subtraction
3561;; but we can quickly determine whether any bits are different instead.
3562(rule 1 (emit_cmp_i128 (cc_nz_or_z cc) a_hi a_lo b_hi b_lo)
3563        (let ((same_lo Reg (x64_xor $I64 a_lo b_lo))
3564              (same_hi Reg (x64_xor $I64 a_hi b_hi)))
3565          (CondResult.CC
3566            (x64_produce_flags_side_effect (ProduceFlagsSideEffectOp.Or) $I64 same_lo same_hi)
3567            cc)))
3568
3569;; The only cases left are L/NL/B/NB which we can implement with a sub/sbb
3570;; sequence. But since we don't care about anything but the flags we can
3571;; replace the sub with cmp, which avoids clobbering one of the registers.
3572(rule 0 (emit_cmp_i128 cc a_hi a_lo b_hi b_lo)
3573        (CondResult.CC
3574          (produces_flags_concat
3575            (x64_cmpq_rm a_lo b_lo)
3576            (x64_produce_flags_side_effect (ProduceFlagsSideEffectOp.Sbb) $I64 a_hi b_hi))
3577          cc))
3578
3579;; CLIF's `fcmp` instruction always operates on XMM registers--both scalar and
3580;; vector. For the scalar versions, we use the flag-setting behavior of the
3581;; `UCOMIS*`.
3582;;
3583;; Checking the result of `UCOMIS*` is unfortunately difficult in some cases
3584;; because we do not have a single condition code to check for the condition
3585;; (i.e., `eq`, `le`, `gt`, etc.) *and* orderedness. Instead, we must check
3586;; the flags multiple times. The UCOMIS* documentation (see Intel's Software
3587;; Developer's Manual, volume 2, chapter 4)
3588;; is helpful:
3589;;  - unordered assigns    Z = 1, P = 1, C = 1
3590;;  - greater than assigns Z = 0, P = 0, C = 0
3591;;  - less than assigns    Z = 0, P = 0, C = 1
3592;;  - equal assigns        Z = 1, P = 0, C = 0
3593(decl emit_fcmp (FloatCC Value Value) CondResult)
3594
3595(rule (emit_fcmp (FloatCC.Equal) a @ (value_type (ty_scalar_float ty)) b)
3596      (CondResult.And (x64_ucomis ty a b) (CC.NP) (CC.Z)))
3597
3598(rule (emit_fcmp (FloatCC.NotEqual) a @ (value_type (ty_scalar_float ty)) b)
3599      (CondResult.Or (x64_ucomis ty a b) (CC.P) (CC.NZ)))
3600
3601;; Some scalar lowerings correspond to one condition code.
3602
3603(rule (emit_fcmp (FloatCC.Ordered) a @ (value_type (ty_scalar_float ty)) b)
3604      (CondResult.CC (x64_ucomis ty a b) (CC.NP)))
3605(rule (emit_fcmp (FloatCC.Unordered) a @ (value_type (ty_scalar_float ty)) b)
3606      (CondResult.CC (x64_ucomis ty a b) (CC.P)))
3607(rule (emit_fcmp (FloatCC.OrderedNotEqual) a @ (value_type (ty_scalar_float ty)) b)
3608      (CondResult.CC (x64_ucomis ty a b) (CC.NZ)))
3609(rule (emit_fcmp (FloatCC.UnorderedOrEqual) a @ (value_type (ty_scalar_float ty)) b)
3610      (CondResult.CC (x64_ucomis ty a b) (CC.Z)))
3611(rule (emit_fcmp (FloatCC.GreaterThan) a @ (value_type (ty_scalar_float ty)) b)
3612      (CondResult.CC (x64_ucomis ty a b) (CC.NBE)))
3613(rule (emit_fcmp (FloatCC.GreaterThanOrEqual) a @ (value_type (ty_scalar_float ty)) b)
3614      (CondResult.CC (x64_ucomis ty a b) (CC.NB)))
3615(rule (emit_fcmp (FloatCC.UnorderedOrLessThan) a @ (value_type (ty_scalar_float ty)) b)
3616      (CondResult.CC (x64_ucomis ty a b) (CC.B)))
3617(rule (emit_fcmp (FloatCC.UnorderedOrLessThanOrEqual) a @ (value_type (ty_scalar_float ty)) b)
3618      (CondResult.CC (x64_ucomis ty a b) (CC.BE)))
3619
3620;; Other scalar lowerings are made possible by flipping the operands and
3621;; reversing the condition code.
3622
3623(rule (emit_fcmp (FloatCC.LessThan) a @ (value_type (ty_scalar_float ty)) b)
3624      ;; Same flags as `GreaterThan`.
3625      (CondResult.CC (x64_ucomis ty b a) (CC.NBE)))
3626(rule (emit_fcmp (FloatCC.LessThanOrEqual) a @ (value_type (ty_scalar_float ty)) b)
3627      ;; Same flags as `GreaterThanOrEqual`.
3628      (CondResult.CC (x64_ucomis ty b a) (CC.NB)))
3629(rule (emit_fcmp (FloatCC.UnorderedOrGreaterThan) a @ (value_type (ty_scalar_float ty)) b)
3630      ;; Same flags as `UnorderedOrLessThan`.
3631      (CondResult.CC (x64_ucomis ty b a) (CC.B)))
3632(rule (emit_fcmp (FloatCC.UnorderedOrGreaterThanOrEqual) a @ (value_type (ty_scalar_float ty)) b)
3633      ;; Same flags as `UnorderedOrLessThanOrEqual`.
3634      (CondResult.CC (x64_ucomis ty b a) (CC.BE)))
3635
3636;;;; Type Guards ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3637
3638;; A type guard for matching ints and bools up to 64 bits, or 64 bit references.
3639(decl ty_int_bool_or_ref () Type)
3640(extern extractor ty_int_bool_or_ref ty_int_bool_or_ref)
3641
3642;;;; Atomics ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3643
3644(decl x64_cmpxchg (Type Gpr Gpr SyntheticAmode) Gpr)
3645(rule (x64_cmpxchg $I8 expected replacement addr) (x64_lock_cmpxchgb_mr addr replacement expected))
3646(rule (x64_cmpxchg $I16 expected replacement addr) (x64_lock_cmpxchgw_mr addr replacement expected))
3647(rule (x64_cmpxchg $I32 expected replacement addr) (x64_lock_cmpxchgl_mr addr replacement expected))
3648(rule (x64_cmpxchg $I64 expected replacement addr) (x64_lock_cmpxchgq_mr addr replacement expected))
3649
3650(decl x64_cmpxchg16b (ValueRegs ValueRegs SyntheticAmode) ValueRegs)
3651(rule (x64_cmpxchg16b expected replacement addr)
3652      (let ((expected_low Gpr (value_regs_get_gpr expected 0))
3653            (expected_high Gpr (value_regs_get_gpr expected 1))
3654            (replacement_low Gpr (value_regs_get_gpr replacement 0))
3655            (replacement_high Gpr (value_regs_get_gpr replacement 1)))
3656        (x64_lock_cmpxchg16b_m expected_low expected_high replacement_low replacement_high addr)))
3657
3658(decl x64_xadd (Type SyntheticAmode Gpr) Gpr)
3659(rule (x64_xadd $I8 addr operand) (x64_lock_xaddb_mr addr operand))
3660(rule (x64_xadd $I16 addr operand) (x64_lock_xaddw_mr addr operand))
3661(rule (x64_xadd $I32 addr operand) (x64_lock_xaddl_mr addr operand))
3662(rule (x64_xadd $I64 addr operand) (x64_lock_xaddq_mr addr operand))
3663
3664(decl x64_xchg (Type SyntheticAmode Gpr) Gpr)
3665(rule (x64_xchg $I8 addr operand) (x64_xchgb_rm operand addr))
3666(rule (x64_xchg $I16 addr operand) (x64_xchgw_rm operand addr))
3667(rule (x64_xchg $I32 addr operand) (x64_xchgl_rm operand addr))
3668(rule (x64_xchg $I64 addr operand) (x64_xchgq_rm operand addr))
3669
3670(decl x64_lock_add (OperandSize SyntheticAmode Gpr) SideEffectNoResult)
3671(rule (x64_lock_add (OperandSize.Size8) addr reg)   (x64_lock_addb_mr_mem addr reg))
3672(rule (x64_lock_add (OperandSize.Size16) addr reg)  (x64_lock_addw_mr_mem addr reg))
3673(rule (x64_lock_add (OperandSize.Size32) addr reg)  (x64_lock_addl_mr_mem addr reg))
3674(rule (x64_lock_add (OperandSize.Size64) addr reg)  (x64_lock_addq_mr_mem addr reg))
3675
3676(decl x64_lock_sub (OperandSize SyntheticAmode Gpr) SideEffectNoResult)
3677(rule (x64_lock_sub (OperandSize.Size8) addr reg)   (x64_lock_subb_mr_mem addr reg))
3678(rule (x64_lock_sub (OperandSize.Size16) addr reg)  (x64_lock_subw_mr_mem addr reg))
3679(rule (x64_lock_sub (OperandSize.Size32) addr reg)  (x64_lock_subl_mr_mem addr reg))
3680(rule (x64_lock_sub (OperandSize.Size64) addr reg)  (x64_lock_subq_mr_mem addr reg))
3681
3682(decl x64_lock_and (OperandSize SyntheticAmode Gpr) SideEffectNoResult)
3683(rule (x64_lock_and (OperandSize.Size8) addr reg)   (x64_lock_andb_mr_mem addr reg))
3684(rule (x64_lock_and (OperandSize.Size16) addr reg)  (x64_lock_andw_mr_mem addr reg))
3685(rule (x64_lock_and (OperandSize.Size32) addr reg)  (x64_lock_andl_mr_mem addr reg))
3686(rule (x64_lock_and (OperandSize.Size64) addr reg)  (x64_lock_andq_mr_mem addr reg))
3687
3688(decl x64_lock_or (OperandSize SyntheticAmode Gpr) SideEffectNoResult)
3689(rule (x64_lock_or (OperandSize.Size8) addr reg)   (x64_lock_orb_mr_mem addr reg))
3690(rule (x64_lock_or (OperandSize.Size16) addr reg)  (x64_lock_orw_mr_mem addr reg))
3691(rule (x64_lock_or (OperandSize.Size32) addr reg)  (x64_lock_orl_mr_mem addr reg))
3692(rule (x64_lock_or (OperandSize.Size64) addr reg)  (x64_lock_orq_mr_mem addr reg))
3693
3694(decl x64_lock_xor (OperandSize SyntheticAmode Gpr) SideEffectNoResult)
3695(rule (x64_lock_xor (OperandSize.Size8) addr reg)   (x64_lock_xorb_mr_mem addr reg))
3696(rule (x64_lock_xor (OperandSize.Size16) addr reg)  (x64_lock_xorw_mr_mem addr reg))
3697(rule (x64_lock_xor (OperandSize.Size32) addr reg)  (x64_lock_xorl_mr_mem addr reg))
3698(rule (x64_lock_xor (OperandSize.Size64) addr reg)  (x64_lock_xorq_mr_mem addr reg))
3699
3700(decl x64_atomic_rmw_seq (Type AtomicRmwSeqOp SyntheticAmode Gpr) Gpr)
3701(rule (x64_atomic_rmw_seq ty op mem input)
3702      (let ((dst WritableGpr (temp_writable_gpr))
3703            (tmp WritableGpr (temp_writable_gpr))
3704            (_ Unit (emit (MInst.AtomicRmwSeq ty op mem input tmp dst))))
3705        dst))
3706
3707(decl x64_atomic_128_rmw_seq (AtomicRmwOp SyntheticAmode MemFlags ValueRegs) ValueRegs)
3708(rule (x64_atomic_128_rmw_seq op mem_low flags input)
3709      (let ((dst_low WritableGpr (temp_writable_gpr))
3710            (dst_high WritableGpr (temp_writable_gpr))
3711            (tmp_low WritableGpr (temp_writable_gpr))
3712            (tmp_high WritableGpr (temp_writable_gpr))
3713            (input_low Gpr (value_regs_get_gpr input 0))
3714            (input_high Gpr (value_regs_get_gpr input 1))
3715            (mem_high SyntheticAmode (amode_offset mem_low flags 8))
3716            (args BoxAtomic128RmwSeqArgs (atomic128_rmw_seq_args (atomic_128_rmw_seq_op op) mem_low mem_high input_low input_high tmp_low tmp_high dst_low dst_high))
3717            (_ Unit (emit (MInst.Atomic128RmwSeq args))))
3718        (value_regs dst_low dst_high)))
3719
3720(rule 1 (x64_atomic_128_rmw_seq (AtomicRmwOp.Xchg) mem_low flags input)
3721        (let ((dst_low WritableGpr (temp_writable_gpr))
3722              (dst_high WritableGpr (temp_writable_gpr))
3723              (input_low Gpr (value_regs_get_gpr input 0))
3724              (input_high Gpr (value_regs_get_gpr input 1))
3725              (mem_high SyntheticAmode (amode_offset mem_low flags 8))
3726              (args BoxAtomic128XchgSeqArgs (atomic128_xchg_seq_args mem_low mem_high input_low input_high dst_low dst_high))
3727              (_ Unit (emit (MInst.Atomic128XchgSeq args))))
3728          (value_regs dst_low dst_high)))
3729
3730(decl x64_atomic_128_store_seq (SyntheticAmode MemFlags ValueRegs) SideEffectNoResult)
3731(rule (x64_atomic_128_store_seq mem_low flags input)
3732        (let ((dst_low WritableGpr (temp_writable_gpr))
3733              (dst_high WritableGpr (temp_writable_gpr))
3734              (input_low Gpr (value_regs_get_gpr input 0))
3735              (input_high Gpr (value_regs_get_gpr input 1))
3736              (mem_high SyntheticAmode (amode_offset mem_low flags 8))
3737              (args BoxAtomic128XchgSeqArgs (atomic128_xchg_seq_args mem_low mem_high input_low input_high dst_low dst_high)))
3738          (SideEffectNoResult.Inst (MInst.Atomic128XchgSeq args))))
3739
3740
3741(type AtomicRmwSeqOp
3742      (enum And
3743            Nand
3744            Or
3745            Xor
3746            Umin
3747            Umax
3748            Smin
3749            Smax))
3750
3751(decl atomic_rmw_seq_op (AtomicRmwOp) AtomicRmwSeqOp)
3752(rule (atomic_rmw_seq_op (AtomicRmwOp.And)) (AtomicRmwSeqOp.And))
3753(rule (atomic_rmw_seq_op (AtomicRmwOp.Nand)) (AtomicRmwSeqOp.Nand))
3754(rule (atomic_rmw_seq_op (AtomicRmwOp.Or)) (AtomicRmwSeqOp.Or))
3755(rule (atomic_rmw_seq_op (AtomicRmwOp.Xor)) (AtomicRmwSeqOp.Xor))
3756(rule (atomic_rmw_seq_op (AtomicRmwOp.Umin)) (AtomicRmwSeqOp.Umin))
3757(rule (atomic_rmw_seq_op (AtomicRmwOp.Umax)) (AtomicRmwSeqOp.Umax))
3758(rule (atomic_rmw_seq_op (AtomicRmwOp.Smin)) (AtomicRmwSeqOp.Smin))
3759(rule (atomic_rmw_seq_op (AtomicRmwOp.Smax)) (AtomicRmwSeqOp.Smax))
3760
3761(type Atomic128RmwSeqOp
3762      (enum Add
3763            Sub
3764            And
3765            Nand
3766            Or
3767            Xor
3768            Umin
3769            Umax
3770            Smin
3771            Smax))
3772
3773(decl atomic_128_rmw_seq_op (AtomicRmwOp) Atomic128RmwSeqOp)
3774(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Add)) (Atomic128RmwSeqOp.Add))
3775(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Sub)) (Atomic128RmwSeqOp.Sub))
3776(rule (atomic_128_rmw_seq_op (AtomicRmwOp.And)) (Atomic128RmwSeqOp.And))
3777(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Nand)) (Atomic128RmwSeqOp.Nand))
3778(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Or)) (Atomic128RmwSeqOp.Or))
3779(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Xor)) (Atomic128RmwSeqOp.Xor))
3780(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Umin)) (Atomic128RmwSeqOp.Umin))
3781(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Umax)) (Atomic128RmwSeqOp.Umax))
3782(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Smin)) (Atomic128RmwSeqOp.Smin))
3783(rule (atomic_128_rmw_seq_op (AtomicRmwOp.Smax)) (Atomic128RmwSeqOp.Smax))
3784
3785;;;; Casting ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3786
3787(decl bitcast_xmm_to_gpr (u8 Xmm) Gpr)
3788(rule (bitcast_xmm_to_gpr 16 src)
3789      (x64_pextrw src 0))
3790(rule (bitcast_xmm_to_gpr 32 src)
3791      (x64_movd_to_gpr src))
3792(rule (bitcast_xmm_to_gpr 64 src)
3793      (x64_movq_to_gpr src))
3794
3795(decl bitcast_xmm_to_gprs (Xmm) ValueRegs)
3796(rule (bitcast_xmm_to_gprs src)
3797      (value_regs (x64_movq_to_gpr src) (x64_movq_to_gpr (x64_pshufd src 0b11101110))))
3798
3799;; This rule zeroes out the upper bits of the XMM register; we need this to
3800;; avoid undefined bits in scaler_to_vector.
3801(decl bitcast_gpr_to_xmm (u8 Gpr) Xmm)
3802(rule (bitcast_gpr_to_xmm 16 src)
3803      (x64_pinsrw (xmm_zero $I16X8) src 0))
3804(rule (bitcast_gpr_to_xmm 32 src)
3805      (x64_movd_to_xmm src))
3806(rule (bitcast_gpr_to_xmm 64 src)
3807      (x64_movq_to_xmm src))
3808
3809(decl bitcast_gprs_to_xmm (ValueRegs) Xmm)
3810(rule (bitcast_gprs_to_xmm src)
3811      (x64_punpcklqdq (x64_movq_to_xmm (value_regs_get_gpr src 0)) (x64_movq_to_xmm (value_regs_get_gpr src 1))))
3812
3813;;;; Stack Addresses ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3814
3815(decl stack_addr_impl (StackSlot Offset32) Gpr)
3816(rule (stack_addr_impl stack_slot offset)
3817      (let ((dst WritableGpr (temp_writable_gpr))
3818            (_ Unit (emit (abi_stackslot_addr dst stack_slot offset))))
3819        dst))
3820
3821;;;; Division/Remainders ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3822
3823;; Helper for creating `CheckedSRemSeq` instructions.
3824(decl x64_checked_srem_seq (OperandSize Gpr Gpr Gpr) ValueRegs)
3825(rule (x64_checked_srem_seq size dividend_lo dividend_hi divisor)
3826      (let ((dst_quotient WritableGpr (temp_writable_gpr))
3827            (dst_remainder WritableGpr (temp_writable_gpr))
3828            (_ Unit (emit (MInst.CheckedSRemSeq size dividend_lo dividend_hi divisor dst_quotient dst_remainder))))
3829        (value_regs dst_quotient dst_remainder)))
3830
3831(decl x64_checked_srem_seq8 (Gpr Gpr) Gpr)
3832(rule (x64_checked_srem_seq8 dividend divisor)
3833      (let ((dst WritableGpr (temp_writable_gpr))
3834            (_ Unit (emit (MInst.CheckedSRemSeq8 dividend divisor dst))))
3835        dst))
3836
3837;; Helper for creating `Div` instructions
3838;;
3839;; Two registers are returned through `ValueRegs` where the first is the
3840;; quotient and the second is the remainder.
3841(decl x64_div (Type Gpr Gpr GprMem TrapCode) ValueRegs)
3842(rule (x64_div $I16 lo hi divisor code) (x64_divw_m lo hi divisor code))
3843(rule (x64_div $I32 lo hi divisor code) (x64_divl_m lo hi divisor code))
3844(rule (x64_div $I64 lo hi divisor code) (x64_divq_m lo hi divisor code))
3845
3846(decl x64_idiv (Type Gpr Gpr GprMem TrapCode) ValueRegs)
3847(rule (x64_idiv $I16 lo hi divisor code) (x64_idivw_m lo hi divisor code))
3848(rule (x64_idiv $I32 lo hi divisor code) (x64_idivl_m lo hi divisor code))
3849(rule (x64_idiv $I64 lo hi divisor code) (x64_idivq_m lo hi divisor code))
3850
3851;;;; Pinned Register ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3852
3853(decl read_pinned_gpr () Gpr)
3854(rule (read_pinned_gpr)
3855      (mov_from_preg (preg_pinned)))
3856
3857(decl write_pinned_gpr (Gpr) SideEffectNoResult)
3858(rule (write_pinned_gpr val)
3859      (mov_to_preg (preg_pinned) val))
3860
3861;;;; Shuffle ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3862
3863;; Produce a mask suitable for use with `pshufb` for permuting the argument to
3864;; shuffle, when the arguments are the same (i.e. `shuffle a a mask`). This will
3865;; map all indices in the range 0..31 to the range 0..15.
3866(decl shuffle_0_31_mask (VecMask) VCodeConstant)
3867(extern constructor shuffle_0_31_mask shuffle_0_31_mask)
3868
3869;; Produce a mask suitable for use with `pshufb` for permuting the lhs of a
3870;; `shuffle` operation (lanes 0-15).
3871(decl shuffle_0_15_mask (VecMask) VCodeConstant)
3872(extern constructor shuffle_0_15_mask shuffle_0_15_mask)
3873
3874;; Produce a mask suitable for use with `pshufb` for permuting the rhs of a
3875;; `shuffle` operation (lanes 16-31).
3876(decl shuffle_16_31_mask (VecMask) VCodeConstant)
3877(extern constructor shuffle_16_31_mask shuffle_16_31_mask)
3878
3879;; Produce a permutation suitable for use with `vpermi2b`, for permuting two
3880;; I8X16 vectors simultaneously.
3881;;
3882;; NOTE: `vpermi2b` will mask the indices in each lane to 5 bits when indexing
3883;; into vectors, so this constructor makes no effort to handle indices that are
3884;; larger than 31. If you are lowering a clif opcode like `shuffle` that has
3885;; special behavior for out of bounds indices (emitting a `0` in the resulting
3886;; vector in the case of `shuffle`) you'll need to handle that behavior
3887;; separately.
3888(decl perm_from_mask (VecMask) VCodeConstant)
3889(extern constructor perm_from_mask perm_from_mask)
3890
3891;; If the mask that would be given to `shuffle` contains any out-of-bounds
3892;; indices, return a mask that will zero those.
3893(decl perm_from_mask_with_zeros (VCodeConstant VCodeConstant) VecMask)
3894(extern extractor perm_from_mask_with_zeros perm_from_mask_with_zeros)
3895
3896;;;; TLS Values ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3897
3898;; Helper for emitting ElfTlsGetAddr.
3899(decl elf_tls_get_addr (ExternalName) Gpr)
3900(rule (elf_tls_get_addr name)
3901      (let ((dst WritableGpr (temp_writable_gpr))
3902            (_ Unit (emit (MInst.ElfTlsGetAddr name dst))))
3903        dst))
3904
3905;; Helper for emitting MachOTlsGetAddr.
3906(decl macho_tls_get_addr (ExternalName) Gpr)
3907(rule (macho_tls_get_addr name)
3908      (let ((dst WritableGpr (temp_writable_gpr))
3909            (_ Unit (emit (MInst.MachOTlsGetAddr name dst))))
3910        dst))
3911
3912;; Helper for emitting CoffTlsGetAddr.
3913(decl coff_tls_get_addr (ExternalName) Gpr)
3914(rule (coff_tls_get_addr name)
3915      (let ((dst WritableGpr (temp_writable_gpr))
3916            (tmp WritableGpr (temp_writable_gpr))
3917            (_ Unit (emit (MInst.CoffTlsGetAddr name dst tmp))))
3918        dst))
3919
3920;;;; Label Addresses ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3921
3922(decl x64_label_address (MachLabel) Gpr)
3923(rule (x64_label_address label)
3924      (let ((dst WritableGpr (temp_writable_gpr))
3925            (_ Unit (emit (MInst.LabelAddress dst label))))
3926        dst))
3927
3928;;;; Automatic conversions ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
3929
3930(convert Gpr InstOutput output_gpr)
3931(convert Value Gpr put_in_gpr)
3932(convert Value GprMem put_in_gpr_mem)
3933(convert Value GprMemImm put_in_gpr_mem_imm)
3934(convert Value RegMem put_in_reg_mem)
3935(convert Value RegMemImm put_in_reg_mem_imm)
3936(convert Gpr GprMemImm gpr_to_gpr_mem_imm)
3937(convert Gpr GprMem gpr_to_gpr_mem)
3938(convert Gpr Reg gpr_to_reg)
3939(convert GprMem RegMem gpr_mem_to_reg_mem)
3940(convert Reg Gpr gpr_new)
3941(convert WritableGpr Gpr writable_gpr_to_gpr)
3942(convert RegMemImm GprMemImm gpr_mem_imm_new)
3943(convert RegMem GprMem reg_mem_to_gpr_mem)
3944(convert RegMem RegMemImm reg_mem_to_reg_mem_imm)
3945(convert Reg GprMem reg_to_gpr_mem)
3946(convert Reg GprMemImm reg_to_gpr_mem_imm)
3947(convert WritableGpr WritableReg writable_gpr_to_reg)
3948(convert WritableGpr Reg writable_gpr_to_r_reg)
3949(convert WritableGpr GprMem writable_gpr_to_gpr_mem)
3950(convert WritableGpr GprMemImm writable_gpr_to_gpr_mem_imm)
3951(convert WritableGpr ValueRegs writable_gpr_to_value_regs)
3952
3953(convert Xmm InstOutput output_xmm)
3954(convert Value Xmm put_in_xmm)
3955(convert Value XmmMem put_in_xmm_mem)
3956(convert Value XmmMemAligned put_in_xmm_mem_aligned)
3957(convert Value XmmMemImm put_in_xmm_mem_imm)
3958(convert Xmm Reg xmm_to_reg)
3959(convert Xmm RegMem xmm_to_reg_mem)
3960(convert Reg Xmm xmm_new)
3961(convert Reg XmmMem reg_to_xmm_mem)
3962(convert Reg RegMemImm reg_to_reg_mem_imm)
3963(convert RegMem XmmMem reg_mem_to_xmm_mem)
3964(convert Xmm XmmMem xmm_to_xmm_mem)
3965(convert Xmm XmmMemImm xmm_to_xmm_mem_imm)
3966(convert Xmm XmmMemAligned xmm_to_xmm_mem_aligned)
3967(convert XmmMem XmmMemImm xmm_mem_to_xmm_mem_imm)
3968(convert XmmMem RegMem xmm_mem_to_reg_mem)
3969(convert RegMemImm XmmMemImm xmm_mem_imm_new)
3970(convert WritableXmm Xmm writable_xmm_to_xmm)
3971(convert WritableXmm WritableReg writable_xmm_to_reg)
3972(convert WritableXmm Reg writable_xmm_to_r_reg)
3973(convert WritableXmm XmmMem writable_xmm_to_xmm_mem)
3974(convert WritableXmm XmmMemAligned writable_xmm_to_xmm_mem_aligned)
3975(convert WritableXmm ValueRegs writable_xmm_to_value_regs)
3976
3977;; Note that these conversions will introduce a `movupd` instruction if
3978;; the memory location is not aligned to a 16-byte boundary. This is primarily
3979;; used to convert `XmmMem` inputs, which themselves were typically created
3980;; via the `put_in_xmm_mem` constructor, into operands of SSE instructions.
3981;; Most pre-AVX instructions working with 16-bytes of data (e.g. full xmm
3982;; registers) require 16-byte alignment.
3983(convert XmmMem XmmMemAligned xmm_mem_to_xmm_mem_aligned)
3984(convert XmmMemImm XmmMemAlignedImm xmm_mem_imm_to_xmm_mem_aligned_imm)
3985
3986(convert Gpr Imm8Gpr gpr_to_imm8_gpr)
3987
3988(convert Amode SyntheticAmode amode_to_synthetic_amode)
3989(convert Amode GprMem amode_to_gpr_mem)
3990(convert SyntheticAmode GprMem synthetic_amode_to_gpr_mem)
3991(convert Amode XmmMem amode_to_xmm_mem)
3992(convert SyntheticAmode XmmMem synthetic_amode_to_xmm_mem)
3993(convert Amode XmmMemAligned amode_to_xmm_mem_aligned)
3994(convert SyntheticAmode XmmMemAligned synthetic_amode_to_xmm_mem_aligned)
3995(convert VCodeConstant SyntheticAmode const_to_synthetic_amode)
3996(convert VCodeConstant XmmMem const_to_xmm_mem)
3997(convert VCodeConstant RegMem const_to_reg_mem)
3998
3999(convert IntCC CC intcc_to_cc)
4000
4001(convert SinkableLoad RegMem sink_load_to_reg_mem)
4002(convert SinkableLoad GprMem sink_load_to_gpr_mem)
4003(convert SinkableLoad RegMemImm sink_load_to_reg_mem_imm)
4004(convert SinkableLoad GprMemImm sink_load_to_gpr_mem_imm)
4005(convert SinkableLoad XmmMem sink_load_to_xmm_mem)
4006(convert SinkableLoad SyntheticAmode sink_load)
4007
4008(decl reg_to_xmm_mem (Reg) XmmMem)
4009(rule (reg_to_xmm_mem r)
4010      (xmm_to_xmm_mem (xmm_new r)))
4011(decl xmm_to_reg_mem (Reg) XmmMem)
4012(rule (xmm_to_reg_mem r)
4013      (RegMem.Reg (xmm_to_reg r)))
4014
4015(decl writable_gpr_to_r_reg (WritableGpr) Reg)
4016(rule (writable_gpr_to_r_reg w_gpr)
4017      (writable_reg_to_reg (writable_gpr_to_reg w_gpr)))
4018(decl writable_gpr_to_gpr_mem (WritableGpr) GprMem)
4019(rule (writable_gpr_to_gpr_mem w_gpr)
4020      (gpr_to_gpr_mem w_gpr))
4021(decl writable_gpr_to_gpr_mem_imm (WritableGpr) GprMemImm)
4022(rule (writable_gpr_to_gpr_mem_imm w_gpr)
4023      (gpr_to_gpr_mem_imm w_gpr))
4024(decl writable_gpr_to_value_regs (WritableGpr) ValueRegs)
4025(rule (writable_gpr_to_value_regs w_gpr)
4026      (value_reg w_gpr))
4027(decl writable_xmm_to_r_reg (WritableXmm) Reg)
4028(rule (writable_xmm_to_r_reg w_xmm)
4029      (writable_reg_to_reg (writable_xmm_to_reg w_xmm)))
4030(decl writable_xmm_to_xmm_mem (WritableXmm) XmmMem)
4031(rule (writable_xmm_to_xmm_mem w_xmm)
4032      (xmm_to_xmm_mem (writable_xmm_to_xmm w_xmm)))
4033(decl writable_xmm_to_xmm_mem_aligned (WritableXmm) XmmMemAligned)
4034(rule (writable_xmm_to_xmm_mem_aligned w_xmm)
4035      (xmm_to_xmm_mem_aligned (writable_xmm_to_xmm w_xmm)))
4036(decl writable_xmm_to_value_regs (WritableXmm) ValueRegs)
4037(rule (writable_xmm_to_value_regs w_xmm)
4038      (value_reg w_xmm))
4039
4040(decl synthetic_amode_to_gpr_mem (SyntheticAmode) GprMem)
4041
4042(spec (amode_to_gpr_mem amode)
4043      (provide (= result amode)))
4044(decl amode_to_gpr_mem (Amode) GprMem)
4045(rule (amode_to_gpr_mem amode)
4046      (amode_to_synthetic_amode amode))
4047(rule (synthetic_amode_to_gpr_mem amode)
4048      (synthetic_amode_to_reg_mem amode))
4049(decl amode_to_xmm_mem (Amode) XmmMem)
4050(rule (amode_to_xmm_mem amode)
4051      (amode_to_synthetic_amode amode))
4052(decl synthetic_amode_to_xmm_mem (SyntheticAmode) XmmMem)
4053(rule (synthetic_amode_to_xmm_mem amode)
4054      (synthetic_amode_to_reg_mem amode))
4055(decl const_to_synthetic_amode (VCodeConstant) SyntheticAmode)
4056(extern constructor const_to_synthetic_amode const_to_synthetic_amode)
4057(decl const_to_xmm_mem (VCodeConstant) XmmMem)
4058(rule (const_to_xmm_mem c) (const_to_synthetic_amode c))
4059(decl const_to_reg_mem (VCodeConstant) RegMem)
4060(rule (const_to_reg_mem c) (RegMem.Mem (const_to_synthetic_amode c)))
4061
4062(decl xmm_to_xmm_mem_aligned (Xmm) XmmMemAligned)
4063(rule (xmm_to_xmm_mem_aligned reg) (xmm_mem_to_xmm_mem_aligned reg))
4064(decl amode_to_xmm_mem_aligned (Amode) XmmMemAligned)
4065(rule (amode_to_xmm_mem_aligned mode) (amode_to_xmm_mem mode))
4066(decl synthetic_amode_to_xmm_mem_aligned (SyntheticAmode) XmmMemAligned)
4067(rule (synthetic_amode_to_xmm_mem_aligned mode) (synthetic_amode_to_xmm_mem mode))
4068(decl put_in_xmm_mem_aligned (Value) XmmMemAligned)
4069(rule (put_in_xmm_mem_aligned val) (put_in_xmm_mem val))
4070
4071(decl mov_to_preg (PReg Gpr) SideEffectNoResult)
4072(rule (mov_to_preg dst src)
4073      (SideEffectNoResult.Inst (MInst.MovToPReg src dst)))
4074
4075(decl preg_rbp () PReg)
4076(extern constructor preg_rbp preg_rbp)
4077
4078(decl preg_rsp () PReg)
4079(extern constructor preg_rsp preg_rsp)
4080
4081(decl preg_pinned () PReg)
4082(extern constructor preg_pinned preg_pinned)
4083
4084(decl x64_rbp () Reg)
4085(rule (x64_rbp)
4086      (mov_from_preg (preg_rbp)))
4087
4088(decl x64_rsp () Reg)
4089(rule (x64_rsp)
4090      (mov_from_preg (preg_rsp)))
4091
4092;;;; Helpers for Emitting LibCalls ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
4093
4094(type LibCall extern
4095      (enum
4096        FmaF32
4097        FmaF64
4098        CeilF32
4099        CeilF64
4100        FloorF32
4101        FloorF64
4102        NearestF32
4103        NearestF64
4104        TruncF32
4105        TruncF64
4106        X86Pshufb))
4107
4108(decl libcall_1 (LibCall Reg) Reg)
4109(extern constructor libcall_1 libcall_1)
4110
4111(decl libcall_2 (LibCall Reg Reg) Reg)
4112(extern constructor libcall_2 libcall_2)
4113
4114(decl libcall_3 (LibCall Reg Reg Reg) Reg)
4115(extern constructor libcall_3 libcall_3)
4116
4117;; Helper for creating a `SequencePoint` instruction.
4118(decl x64_sequence_point () SideEffectNoResult)
4119(rule (x64_sequence_point)
4120      (SideEffectNoResult.Inst (MInst.SequencePoint)))
4121