From 9b098a9fbe0b322cafa1522e0359ea3fc12e3dc8 Mon Sep 17 00:00:00 2001
From: wborgeaud <williamborgeaud@gmail.com>
Date: Wed, 6 Oct 2021 16:36:30 +0200
Subject: [PATCH] Trivial random access

---
 src/gadgets/random_access.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/gadgets/random_access.rs b/src/gadgets/random_access.rs
index 13e42ab8..ab0db68c 100644
--- a/src/gadgets/random_access.rs
+++ b/src/gadgets/random_access.rs
@@ -14,6 +14,10 @@ impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
         claimed_element: ExtensionTarget<D>,
         v: Vec<ExtensionTarget<D>>,
     ) {
+        debug_assert!(!v.is_empty());
+        if v.len() == 1 {
+            return self.connect_extension(claimed_element, v[0]);
+        }
         let gate = RandomAccessGate::new(v.len());
         let gate_index = self.add_gate(gate.clone(), vec![]);
 
@@ -42,6 +46,10 @@ impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
         mut v: Vec<ExtensionTarget<D>>,
         min_length: usize,
     ) {
+        debug_assert!(!v.is_empty());
+        if v.len() == 1 {
+            return self.connect_extension(claimed_element, v[0]);
+        }
         let zero = self.zero_extension();
         if v.len() < min_length {
             v.resize(8, zero);