Skip to content

Commit

Permalink
Merge pull request #2823 from o1-labs/o1vm/riscv32im/test-decoding-sltu
Browse files Browse the repository at this point in the history
o1vm/riscv32im: test decoding sltu
  • Loading branch information
dannywillems authored Dec 4, 2024
2 parents 0116f6c + 7fe1149 commit f25d9d5
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions o1vm/src/interpreters/riscv32im/tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -146,6 +146,29 @@ pub fn generate_random_slt_instruction<RNG: RngCore + CryptoRng>(rng: &mut RNG)
]
}

pub fn generate_random_sltu_instruction<RNG: RngCore + CryptoRng>(rng: &mut RNG) -> [u8; 4] {
let opcode = 0b0110011;
let rd = rng.gen_range(0..32);
let funct3 = 0b011;
let rs1 = rng.gen_range(0..32);
let rs2 = rng.gen_range(0..32);
let funct2 = 0b00;
let funct5 = 0b00000;
let instruction = opcode
| (rd << 7)
| (funct3 << 12)
| (rs1 << 15)
| (rs2 << 20)
| (funct2 << 25)
| (funct5 << 27);
[
instruction as u8,
(instruction >> 8) as u8,
(instruction >> 16) as u8,
(instruction >> 24) as u8,
]
}

#[test]
pub fn test_instruction_decoding_add() {
let mut env: Env<Fp> = dummy_env();
Expand Down Expand Up @@ -197,3 +220,19 @@ pub fn test_instruction_decoding_slt() {
let (opcode, _instruction) = env.decode_instruction();
assert_eq!(opcode, Instruction::RType(RInstruction::SetLessThan));
}

#[test]
pub fn test_instruction_decoding_sltu() {
let mut env: Env<Fp> = dummy_env();
let mut rng = o1_utils::tests::make_test_rng(None);
let instruction = generate_random_sltu_instruction(&mut rng);
env.memory[0].1[0] = instruction[0];
env.memory[0].1[1] = instruction[1];
env.memory[0].1[2] = instruction[2];
env.memory[0].1[3] = instruction[3];
let (opcode, _instruction) = env.decode_instruction();
assert_eq!(
opcode,
Instruction::RType(RInstruction::SetLessThanUnsigned)
);
}

0 comments on commit f25d9d5

Please sign in to comment.