logos-storage/plonky2
1
0
Fork 0
mirror of https://github.com/logos-storage/plonky2.git synced 2026-01-05 23:33:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

merge successful

Browse Source
This commit is contained in:
Dmitry Vagner 2023-05-16 15:02:01 +02:00
parent 59ae7103a8
commit d3986e6b50
3 changed files with 34 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
evm/src/cpu/kernel/tests/bn254.rs
View File

@ -20,7 +20,7 @@ fn extract_stack(interpreter: Interpreter<'static>) -> Vec<U256> {
.collect::<Vec<U256>>() .collect::<Vec<U256>>()
} }
fn run_bn_mul_fp6(f: Fp6<BN254>, g: Fp6<BN254>, label: &str) -> Vec<U256> { fn run_bn_mul_fp6(f: Fp6<BN254>, g: Fp6<BN254>, label: &str) -> Fp6<BN254> {
let mut stack = f.to_stack(); let mut stack = f.to_stack();
if label == "mul_fp254_6" { if label == "mul_fp254_6" {
stack.extend(g.to_stack().to_vec()); stack.extend(g.to_stack().to_vec());
@ -52,7 +52,7 @@ fn test_bn_mul_fp6() -> Result<()> {
Ok(()) Ok(())
} }
fn run_bn_mul_fp12(f: Fp12<BN254>, g: Fp12<BN254>, label: &str) -> Vec<U256> { fn run_bn_mul_fp12(f: Fp12<BN254>, g: Fp12<BN254>, label: &str) -> Fp12<BN254> {
let in0: usize = 100; let in0: usize = 100;
let in1: usize = 112; let in1: usize = 112;
let out: usize = 124; let out: usize = 124;
@ -118,7 +118,7 @@ fn test_bn_frob_fp6() -> Result<()> {
Ok(()) Ok(())
} }
fn run_bn_frob_fp12(f: Fp12<BN254>, n: usize) -> Vec<U256> { fn run_bn_frob_fp12(f: Fp12<BN254>, n: usize) -> Fp12<BN254> {
let ptr: usize = 100; let ptr: usize = 100;
let setup = InterpreterMemoryInitialization { let setup = InterpreterMemoryInitialization {
label: format!("test_frob_fp254_12_{}", n), label: format!("test_frob_fp254_12_{}", n),

17
evm/src/curve_pairings.rs
View File

@ -25,12 +25,21 @@ impl<T: FieldExt> Curve<T> {
} }
} }
impl<T: FieldExt + Stack> Curve<T> { impl<T: FieldExt + Stack> Stack for Curve<T> {
pub fn on_stack(self) -> Vec<U256> { const SIZE: usize = 2 * T::SIZE;
let mut stack = self.x.on_stack();
stack.extend(self.y.on_stack()); fn to_stack(&self) -> Vec<U256> {
let mut stack = self.x.to_stack();
stack.extend(self.y.to_stack());
stack stack
} }
fn from_stack(stack: &[U256]) -> Self {
Curve {
x: T::from_stack(&stack[0..T::SIZE]),
y: T::from_stack(&stack[T::SIZE..2 * T::SIZE]),
}
}
} }
impl<T> Curve<T> impl<T> Curve<T>

56
evm/src/extension_tower.rs
View File

@ -1225,7 +1225,7 @@ where
pub trait Stack { pub trait Stack {
const SIZE: usize; const SIZE: usize;
fn to_stack(&self) -> &[U256]; fn to_stack(&self) -> Vec<U256>;
fn from_stack(stack: &[U256]) -> Self; fn from_stack(stack: &[U256]) -> Self;
} }
@ -1233,9 +1233,8 @@ pub trait Stack {
impl Stack for BN254 { impl Stack for BN254 {
const SIZE: usize = 1; const SIZE: usize = 1;
fn to_stack(&self) -> &[U256] { fn to_stack(&self) -> Vec<U256> {
let boxed: Box<[U256]> = Box::new([self.val]); vec![self.val]
Box::leak(boxed)
} }
fn from_stack(stack: &[U256]) -> BN254 { fn from_stack(stack: &[U256]) -> BN254 {
@ -1243,18 +1242,11 @@ impl Stack for BN254 {
} }
} }
impl Stack for BN254 {
fn on_stack(self) -> Vec<U256> {
vec![self.val]
}
}
impl Stack for BLS381 { impl Stack for BLS381 {
const SIZE: usize = 2; const SIZE: usize = 2;
fn to_stack(&self) -> &[U256] { fn to_stack(&self) -> Vec<U256> {
let boxed: Box<[U256]> = Box::new([self.lo(), self.hi()]); vec![self.lo(), self.hi()]
Box::leak(boxed)
} }
fn from_stack(stack: &[U256]) -> BLS381 { fn from_stack(stack: &[U256]) -> BLS381 {
@ -1268,13 +1260,10 @@ impl Stack for BLS381 {
impl<T: FieldExt + Stack> Stack for Fp2<T> { impl<T: FieldExt + Stack> Stack for Fp2<T> {
const SIZE: usize = 2 * T::SIZE; const SIZE: usize = 2 * T::SIZE;
fn to_stack(&self) -> &[U256] { fn to_stack(&self) -> Vec<U256> {
let re = self.re.to_stack(); let mut stack = self.re.to_stack();
let im = self.im.to_stack(); stack.extend(self.im.to_stack());
let mut combined: Vec<U256> = Vec::new(); stack
combined.extend_from_slice(re);
combined.extend_from_slice(im);
Box::leak(combined.into_boxed_slice())
} }
fn from_stack(stack: &[U256]) -> Fp2<T> { fn from_stack(stack: &[U256]) -> Fp2<T> {
@ -1293,16 +1282,11 @@ where
{ {
const SIZE: usize = 3 * Fp2::<T>::SIZE; const SIZE: usize = 3 * Fp2::<T>::SIZE;
fn to_stack(&self) -> &[U256] { fn to_stack(&self) -> Vec<U256> {
let t0 = self.t0.to_stack(); let mut stack = self.t0.to_stack();
let t1 = self.t1.to_stack(); stack.extend(self.t1.to_stack());
let t2 = self.t2.to_stack(); stack.extend(self.t2.to_stack());
stack
let mut combined: Vec<U256> = Vec::new();
combined.extend_from_slice(t0);
combined.extend_from_slice(t1);
combined.extend_from_slice(t2);
Box::leak(combined.into_boxed_slice())
} }
fn from_stack(stack: &[U256]) -> Self { fn from_stack(stack: &[U256]) -> Self {
@ -1322,14 +1306,10 @@ where
{ {
const SIZE: usize = 2 * Fp6::<T>::SIZE; const SIZE: usize = 2 * Fp6::<T>::SIZE;
fn to_stack(&self) -> &[U256] { fn to_stack(&self) -> Vec<U256> {
let z0 = self.z0.to_stack(); let mut stack = self.z0.to_stack();
let z1 = self.z1.to_stack(); stack.extend(self.z1.to_stack());
stack
let mut combined: Vec<U256> = Vec::new();
combined.extend_from_slice(z0);
combined.extend_from_slice(z1);
Box::leak(combined.into_boxed_slice())
} }
fn from_stack(stack: &[U256]) -> Self { fn from_stack(stack: &[U256]) -> Self {