From 1ed0f9ff6c98e90dfe5433ce2e4fde7e32581063 Mon Sep 17 00:00:00 2001
From: ivanlele <ivanlele402@gmail.com>
Date: Tue, 6 Jan 2026 18:46:27 +0200
Subject: [PATCH] force JetBuffer and FlatValue to use UWORD

---
 jets-bench/src/buffer.rs | 2 +-
 jets-bench/src/input.rs  | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/jets-bench/src/buffer.rs b/jets-bench/src/buffer.rs
index 7ed1e45e..b6d04acc 100644
--- a/jets-bench/src/buffer.rs
+++ b/jets-bench/src/buffer.rs
@@ -17,7 +17,7 @@ use simplicity_sys::CFrameItem;
 /// and to the target_type of jet.
 pub struct JetBuffer {
     #[allow(dead_code)]
-    cells: Vec<usize>,
+    cells: Vec<UWORD>,
     src_bit_width: usize,
     dst_bit_width: usize,
     src_ptr_begin: *mut UWORD,
diff --git a/jets-bench/src/input.rs b/jets-bench/src/input.rs
index 0aaed71d..8b6ed0ba 100644
--- a/jets-bench/src/input.rs
+++ b/jets-bench/src/input.rs
@@ -5,6 +5,7 @@ use rand::distributions as dist;
 use rand::rngs::ThreadRng;
 use rand::{Rng, RngCore};
 use simplicity::ffi::c_jets::frame_ffi::c_writeBit;
+use simplicity::ffi::ffi::UWORD;
 use simplicity::ffi::CFrameItem;
 use simplicity::hashes::Hash;
 use simplicity::jet::Elements;
@@ -309,8 +310,8 @@ impl FlatValue {
         unsafe {
             use simplicity::elements::hashes::Hash as _;
 
-            let mut dst_inner = [0usize; MAX_VALUE_BYTES / mem::size_of::<usize>()];
-            let mut src_inner = [0usize; MAX_VALUE_BYTES / mem::size_of::<usize>()];
+            let mut dst_inner = [UWORD::from(0u8); MAX_VALUE_BYTES / mem::size_of::<UWORD>()];
+            let mut src_inner = [UWORD::from(0u8); MAX_VALUE_BYTES / mem::size_of::<UWORD>()];
 
             let mut src_bytes = self.inner;
             // See below block comment on the write frame for justification of this
@@ -322,7 +323,7 @@ impl FlatValue {
                 MAX_VALUE_BYTES,
             );
             for us in &mut src_inner {
-                *us = usize::from_be(us.swap_bytes());
+                *us = UWORD::from_be(us.swap_bytes());
             }
 
             let src_read_frame = CFrameItem::new_read(self.len_bits, src_inner.as_ptr());