forked from starkware-libs/cairo
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathqm31.cairo
More file actions
121 lines (106 loc) · 3.54 KB
/
Copy pathqm31.cairo
File metadata and controls
121 lines (106 loc) · 3.54 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
//! Definition for the `qm31` type.
//! Only available for local proofs.
//!
//! The implementations defined in this module can be accessed by using the traits directly.
/// The `qm31` type, defining an extension field over 4 `m31`s.
pub extern type qm31;
/// The field for the Mersenne prime with `n` 31.
#[feature("bounded-int-utils")]
pub type m31 =
core::internal::bounded_int::BoundedInt<0, 0x7ffffffe>;
pub trait QM31Trait {
/// Returns a new `qm31` composed of the given parts.
fn new(w0: m31, w1: m31, w2: m31, w3: m31) -> qm31;
/// Returns the parts of the given `qm31` as `m31`s.
fn unpack(self: qm31) -> [m31; 4];
}
impl QM31Impl of QM31Trait {
fn new(w0: m31, w1: m31, w2: m31, w3: m31) -> qm31 {
qm31_pack(w0, w1, w2, w3)
}
fn unpack(self: qm31) -> [m31; 4] {
let (w0, w1, w2, w3) = qm31_unpack(self);
[w0, w1, w2, w3]
}
}
/// Returns a `qm31` given its values as constants.
pub extern fn qm31_const<
const W0: m31, const W1: m31, const W2: m31, const W3: m31,
>() -> qm31 nopanic;
extern fn qm31_is_zero(a: qm31) -> core::internal::OptionRev<NonZero<qm31>> nopanic;
extern fn qm31_add(a: qm31, b: qm31) -> qm31 nopanic;
extern fn qm31_sub(a: qm31, b: qm31) -> qm31 nopanic;
extern fn qm31_mul(a: qm31, b: qm31) -> qm31 nopanic;
extern fn qm31_div(a: qm31, b: NonZero<qm31>) -> qm31 nopanic;
extern fn qm31_pack(w0: m31, w1: m31, w2: m31, w3: m31) -> qm31 nopanic;
extern fn qm31_unpack(a: qm31) -> (m31, m31, m31, m31) implicits(crate::RangeCheck) nopanic;
extern fn qm31_from_m31(a: m31) -> qm31 nopanic;
impl qm31Copy of Copy<qm31>;
impl qm31Drop of Drop<qm31>;
impl qm31Add of Add<qm31> {
fn add(lhs: qm31, rhs: qm31) -> qm31 {
qm31_add(lhs, rhs)
}
}
impl qm31Sub of Sub<qm31> {
fn sub(lhs: qm31, rhs: qm31) -> qm31 {
qm31_sub(lhs, rhs)
}
}
impl qm31Mul of Mul<qm31> {
fn mul(lhs: qm31, rhs: qm31) -> qm31 {
qm31_mul(lhs, rhs)
}
}
impl qm31Div of Div<qm31> {
fn div(lhs: qm31, rhs: qm31) -> qm31 {
qm31_div(lhs, rhs.try_into().expect('qm31_div by 0'))
}
}
impl QM31TryIntoNonZero of TryInto<qm31, NonZero<qm31>> {
fn try_into(self: qm31) -> Option<NonZero<qm31>> {
match qm31_is_zero(self) {
core::internal::OptionRev::None => None,
core::internal::OptionRev::Some(x) => Some(x),
}
}
}
impl QM31PartialEq of PartialEq<qm31> {
fn eq(lhs: @qm31, rhs: @qm31) -> bool {
(*lhs - *rhs).is_zero()
}
}
impl QM31Zero of crate::num::traits::Zero<qm31> {
fn zero() -> qm31 {
qm31_const::<0, 0, 0, 0>()
}
#[inline]
fn is_zero(self: @qm31) -> bool {
match qm31_is_zero(*self) {
core::internal::OptionRev::None => true,
core::internal::OptionRev::Some(_) => false,
}
}
#[inline]
fn is_non_zero(self: @qm31) -> bool {
!self.is_zero()
}
}
impl M31IntoQM31 of Into<m31, qm31> {
fn into(self: m31) -> qm31 {
qm31_from_m31(self)
}
}
/// Additional `m31` actions for specific implementations based on `qm31` opcode.
pub mod m31_ops {
use super::m31;
/// Addition of `m31`s in field.
extern fn m31_add(a: m31, b: m31) -> m31 nopanic;
/// Subtraction of `m31`s in field.
extern fn m31_sub(a: m31, b: m31) -> m31 nopanic;
/// Multiplication of `m31`s in field.
extern fn m31_mul(a: m31, b: m31) -> m31 nopanic;
/// Division of `m31`s in field.
extern fn m31_div(a: m31, b: NonZero<m31>) -> m31 nopanic;
pub use {m31_add as add, m31_div as div, m31_mul as mul, m31_sub as sub};
}