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

fixes

Browse Source
This commit is contained in:
Daniel Lubarov 2022-07-31 20:43:58 -07:00
parent 61a9839f2f
commit 94183f723c
1 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
evm/src/cpu/kernel/optimizer.rs
View File

@ -48,9 +48,9 @@ fn constant_propagation(code: &mut Vec<Item>) {
replace_windows(code, |window| {
if let [Push(Literal(y)), Push(Literal(x)), StandardOp(op)] = window {
match op.as_str() {
"ADD" => Some(vec![Push(Literal(x + y))]),
"SUB" => Some(vec![Push(Literal(x - y))]),
"MUL" => Some(vec![Push(Literal(x * y))]),
"ADD" => Some(vec![Push(Literal(x.overflowing_add(y).0))]),
"SUB" => Some(vec![Push(Literal(x.overflowing_sub(y).0))]),
"MUL" => Some(vec![Push(Literal(x.overflowing_mul(y).0))]),
"DIV" => Some(vec![Push(Literal(
x.checked_div(y).unwrap_or(U256::zero()),
))]),
@ -99,6 +99,28 @@ mod tests {
assert_eq!(code, vec![Push(Literal(0.into()))]);
}
#[test]
fn test_constant_propagation_add_overflowing() {
let mut code = vec![
Push(Literal(U256::max_value())),
Push(Literal(U256::max_value())),
StandardOp("ADD".into()),
];
constant_propagation(&mut code);
assert_eq!(code, vec![Push(Literal(U256::max_value() - 1))]);
}
#[test]
fn test_constant_propagation_sub_underflowing() {
let mut code = vec![
Push(Literal(U256::one())),
Push(Literal(U256::zero())),
StandardOp("SUB".into()),
];
constant_propagation(&mut code);
assert_eq!(code, vec![Push(Literal(U256::max_value()))]);
}
#[test]
fn test_constant_propagation_mul() {
let mut code = vec![