From 02db6c93ecfd12e4d088b8a1d229d2ae40473f80 Mon Sep 17 00:00:00 2001
From: wborgeaud <williamborgeaud@gmail.com>
Date: Fri, 13 May 2022 14:20:59 +0200
Subject: [PATCH] Add permutation in mock STARKs for good measure

---
 starky2/src/all_stark.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/starky2/src/all_stark.rs b/starky2/src/all_stark.rs
index bcfe9935..262b3fb2 100644
--- a/starky2/src/all_stark.rs
+++ b/starky2/src/all_stark.rs
@@ -7,6 +7,7 @@ use plonky2::hash::hash_types::RichField;
 use crate::config::StarkConfig;
 use crate::constraint_consumer::{ConstraintConsumer, RecursiveConstraintConsumer};
 use crate::cross_table_lookup::CrossTableLookup;
+use crate::permutation::PermutationPair;
 use crate::stark::Stark;
 use crate::vars::{StarkEvaluationTargets, StarkEvaluationVars};
 
@@ -71,6 +72,10 @@ impl<F: RichField + Extendable<D>, const D: usize> Stark<F, D> for CpuStark<F, D
     fn constraint_degree(&self) -> usize {
         3
     }
+
+    fn permutation_pairs(&self) -> Vec<PermutationPair> {
+        vec![PermutationPair::singletons(8, 9)]
+    }
 }
 
 impl<F: RichField + Extendable<D>, const D: usize> Stark<F, D> for KeccakStark<F, D> {
@@ -98,6 +103,10 @@ impl<F: RichField + Extendable<D>, const D: usize> Stark<F, D> for KeccakStark<F
     fn constraint_degree(&self) -> usize {
         3
     }
+
+    fn permutation_pairs(&self) -> Vec<PermutationPair> {
+        vec![PermutationPair::singletons(0, 6)]
+    }
 }
 
 #[cfg(test)]