Skip to content

Commit

Permalink
Merge pull request #2824 from o1-labs/o1vm/riscv32im/test-decoding-xor
Browse files Browse the repository at this point in the history
o1vm/riscv32im: test decoding xor
  • Loading branch information
dannywillems authored Dec 4, 2024
2 parents f25d9d5 + c12f578 commit 55e9941
Showing 1 changed file with 36 additions and 0 deletions.
36 changes: 36 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_xor_instruction<RNG: RngCore + CryptoRng>(rng: &mut RNG) -> [u8; 4] {
let opcode = 0b0110011;
let rd = rng.gen_range(0..32);
let funct3 = 0b100;
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,
]
}

pub fn generate_random_sltu_instruction<RNG: RngCore + CryptoRng>(rng: &mut RNG) -> [u8; 4] {
let opcode = 0b0110011;
let rd = rng.gen_range(0..32);
Expand Down Expand Up @@ -236,3 +259,16 @@ pub fn test_instruction_decoding_sltu() {
Instruction::RType(RInstruction::SetLessThanUnsigned)
);
}

#[test]
pub fn test_instruction_decoding_xor() {
let mut env: Env<Fp> = dummy_env();
let mut rng = o1_utils::tests::make_test_rng(None);
let instruction = generate_random_xor_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::Xor));
}

0 comments on commit 55e9941

Please sign in to comment.