From 399903e0d8c353fe57ed02c387c84ff389d0e202 Mon Sep 17 00:00:00 2001
From: Roshan <19766713+rpalakkal@users.noreply.github.com>
Date: Mon, 6 May 2024 23:41:15 -0400
Subject: [PATCH] chore: make poseidon hasher thread safe (#269)

---
 halo2-base/src/poseidon/hasher/mod.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/halo2-base/src/poseidon/hasher/mod.rs b/halo2-base/src/poseidon/hasher/mod.rs
index 68cf64c6..b477edcf 100644
--- a/halo2-base/src/poseidon/hasher/mod.rs
+++ b/halo2-base/src/poseidon/hasher/mod.rs
@@ -10,7 +10,7 @@ use crate::{
 
 use getset::{CopyGetters, Getters};
 use num_bigint::BigUint;
-use std::{cell::OnceCell, mem};
+use std::{mem, sync::OnceLock};
 
 #[cfg(test)]
 mod tests;
@@ -28,7 +28,7 @@ pub struct PoseidonHasher<F: ScalarField, const T: usize, const RATE: usize> {
     /// Spec, contains round constants and optimized matrices.
     #[getset(get = "pub")]
     spec: OptimizedPoseidonSpec<F, T, RATE>,
-    consts: OnceCell<PoseidonHasherConsts<F, T, RATE>>,
+    consts: OnceLock<PoseidonHasherConsts<F, T, RATE>>,
 }
 #[derive(Clone, Debug, Getters)]
 struct PoseidonHasherConsts<F: ScalarField, const T: usize, const RATE: usize> {
@@ -123,7 +123,7 @@ pub struct PoseidonCompactOutput<F: ScalarField> {
 impl<F: ScalarField, const T: usize, const RATE: usize> PoseidonHasher<F, T, RATE> {
     /// Create a poseidon hasher from an existing spec.
     pub fn new(spec: OptimizedPoseidonSpec<F, T, RATE>) -> Self {
-        Self { spec, consts: OnceCell::new() }
+        Self { spec, consts: OnceLock::new() }
     }
     /// Initialize necessary consts of hasher. Must be called before any computation.
     pub fn initialize_consts(&mut self, ctx: &mut Context<F>, gate: &impl GateInstructions<F>) {