use vyre::ir::{AtomicOp, BufferDecl, DataType, Expr, Node, Program};
use vyre_conform::{reference::interp, spec::value::Value};
#[test]
fn u64_atomic_add_propagates_carry_to_high_word() {
let program = Program {
entry_op_id: None,
buffers: vec![BufferDecl::read_write("u64_buf", 0, DataType::U64)],
workgroup_size: [1, 1, 1],
entry: vec![Node::store(
"u64_buf",
Expr::u32(0),
Expr::Atomic {
op: AtomicOp::Add,
buffer: "u64_buf".to_string(),
index: Box::new(Expr::u32(0)),
expected: None,
value: Box::new(Expr::u32(1)),
},
)],
};
let bytes = vec![0xFF, 0xFF, 0xFF, 0xFF, 0x00, 0x00, 0x00, 0x00];
let outputs =
interp::run(&program, &[Value::Bytes(bytes)]).expect("u64 atomic add must propagate carry");
let expected = vec![0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00];
assert_eq!(outputs, vec![Value::Bytes(expected)]);
}