cranelift_codegen/isa/riscv64/inst/
regs.rs

1//! Riscv64 ISA definitions: registers.
2//!
3
4use crate::machinst::{Reg, Writable};
5
6use alloc::vec;
7use alloc::vec::Vec;
8
9use regalloc2::{PReg, RegClass, VReg};
10
11// first argument of function call
12#[inline]
13pub fn a0() -> Reg {
14    x_reg(10)
15}
16
17// second argument of function call
18#[inline]
19#[allow(dead_code)]
20pub fn a1() -> Reg {
21    x_reg(11)
22}
23
24// third argument of function call
25#[inline]
26#[allow(dead_code)]
27pub fn a2() -> Reg {
28    x_reg(12)
29}
30
31#[inline]
32#[allow(dead_code)]
33pub fn writable_a0() -> Writable<Reg> {
34    Writable::from_reg(a0())
35}
36#[inline]
37#[allow(dead_code)]
38pub fn writable_a1() -> Writable<Reg> {
39    Writable::from_reg(a1())
40}
41#[inline]
42#[allow(dead_code)]
43pub fn writable_a2() -> Writable<Reg> {
44    Writable::from_reg(a2())
45}
46
47#[inline]
48#[allow(dead_code)]
49pub fn fa0() -> Reg {
50    f_reg(10)
51}
52#[inline]
53#[allow(dead_code)]
54pub fn writable_fa0() -> Writable<Reg> {
55    Writable::from_reg(fa0())
56}
57#[inline]
58#[allow(dead_code)]
59pub fn writable_fa1() -> Writable<Reg> {
60    Writable::from_reg(fa1())
61}
62#[inline]
63pub fn fa1() -> Reg {
64    f_reg(11)
65}
66
67/// Get a reference to the zero-register.
68#[inline]
69pub fn zero_reg() -> Reg {
70    x_reg(0)
71}
72
73/// Get a writable reference to the zero-register (this discards a result).
74#[inline]
75pub fn writable_zero_reg() -> Writable<Reg> {
76    Writable::from_reg(zero_reg())
77}
78#[inline]
79pub fn stack_reg() -> Reg {
80    x_reg(2)
81}
82
83/// Get a writable reference to the stack-pointer register.
84#[inline]
85pub fn writable_stack_reg() -> Writable<Reg> {
86    Writable::from_reg(stack_reg())
87}
88
89/// Get a reference to the link register (x1).
90pub fn link_reg() -> Reg {
91    x_reg(1)
92}
93
94/// Get a writable reference to the link register.
95#[inline]
96pub fn writable_link_reg() -> Writable<Reg> {
97    Writable::from_reg(link_reg())
98}
99
100/// Get a reference to the frame pointer (x8).
101#[inline]
102pub fn fp_reg() -> Reg {
103    x_reg(8)
104}
105
106/// Get a writable reference to the frame pointer.
107#[inline]
108pub fn writable_fp_reg() -> Writable<Reg> {
109    Writable::from_reg(fp_reg())
110}
111
112/// Get a reference to the first temporary, sometimes "spill temporary",
113/// register. This register is used in various ways as a temporary.
114#[inline]
115pub fn spilltmp_reg() -> Reg {
116    x_reg(31)
117}
118
119/// Get a writable reference to the spilltmp reg.
120#[inline]
121pub fn writable_spilltmp_reg() -> Writable<Reg> {
122    Writable::from_reg(spilltmp_reg())
123}
124
125///spilltmp2
126#[inline]
127pub fn spilltmp_reg2() -> Reg {
128    x_reg(30)
129}
130
131/// Get a writable reference to the spilltmp2 reg.
132#[inline]
133pub fn writable_spilltmp_reg2() -> Writable<Reg> {
134    Writable::from_reg(spilltmp_reg2())
135}
136
137#[inline]
138pub fn x_reg(enc: usize) -> Reg {
139    let p_reg = PReg::new(enc, RegClass::Int);
140    let v_reg = VReg::new(p_reg.index(), p_reg.class());
141    Reg::from(v_reg)
142}
143pub const fn px_reg(enc: usize) -> PReg {
144    PReg::new(enc, RegClass::Int)
145}
146
147#[inline]
148pub fn f_reg(enc: usize) -> Reg {
149    let p_reg = PReg::new(enc, RegClass::Float);
150    let v_reg = VReg::new(p_reg.index(), p_reg.class());
151    Reg::from(v_reg)
152}
153pub const fn pf_reg(enc: usize) -> PReg {
154    PReg::new(enc, RegClass::Float)
155}
156
157#[allow(dead_code)]
158pub(crate) fn x_reg_range(start: usize, end: usize) -> Vec<Writable<Reg>> {
159    let mut regs = vec![];
160    for i in start..=end {
161        regs.push(Writable::from_reg(x_reg(i)));
162    }
163    regs
164}
165
166pub const fn pv_reg(enc: usize) -> PReg {
167    PReg::new(enc, RegClass::Vector)
168}
cranelift_codegen/isa/riscv64/inst/regs.rs

cranelift_codegen/isa/riscv64/inst/
regs.rs
