Skip to content

Commit

Permalink
.
Browse files Browse the repository at this point in the history
  • Loading branch information
@rw0x0
rw0x0 committed Nov 27, 2024
1 parent f0d9be1 commit ecbe468
Show file tree
Hide file tree
Showing 2 changed files with 58 additions and 228 deletions.
1 change: 1 addition & 0 deletions co-noir/co-brillig/Cargo.toml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ noirc-abi.workspace = true
noirc-artifacts.workspace = true
num-bigint.workspace = true
num-traits.workspace = true
rand.workspace = true
rayon.workspace = true
serde.workspace = true
thiserror.workspace = true
Expand Down
285 changes: 57 additions & 228 deletions co-noir/co-brillig/src/mpc/rep3.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@ use mpc_core::protocols::rep3_ring::ring::int_ring::IntRing2k;
use mpc_core::protocols::rep3_ring::ring::ring_impl::RingElement;
use mpc_core::protocols::rep3_ring::{self, Rep3BitShare, Rep3RingShare};
use num_bigint::BigUint;
use num_traits::AsPrimitive;
use rand::distributions::{Distribution, Standard};
use std::marker::PhantomData;

use super::PlainBrilligType as Public;
Expand Down Expand Up @@ -145,6 +147,43 @@ macro_rules! bit_from_u128 {
}};
}

fn cast_ring<T, F: PrimeField, N: Rep3Network>(
share: Rep3RingShare<T>,
integer_bit_size: IntegerBitSize,
io_context: &mut IoContext<N>,
) -> eyre::Result<Rep3BrilligType<F>>
where
Standard: Distribution<T>,
T: IntRing2k
+ AsPrimitive<Bit>
+ AsPrimitive<u8>
+ AsPrimitive<u16>
+ AsPrimitive<u32>
+ AsPrimitive<u64>
+ AsPrimitive<u128>,
{
match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector::<_, Bit, _>(share, io_context)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector::<_, u8, _>(share, io_context)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector::<_, u16, _>(share, io_context)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector::<_, u32, _>(share, io_context)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector::<_, u64, _>(share, io_context)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector::<_, u128, _>(share, io_context)?,
))),
}
}

impl<F: PrimeField, N: Rep3Network> BrilligDriver<F> for Rep3BrilligDriver<F, N> {
type BrilligType = Rep3BrilligType<F>;

Expand Down Expand Up @@ -235,234 +274,24 @@ impl<F: PrimeField, N: Rep3Network> BrilligDriver<F> for Rep3BrilligDriver<F, N>
)?,
))),
},
Shared::Ring128(rep3_ring_share) => match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
},
Shared::Ring64(rep3_ring_share) => match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
},
Shared::Ring32(rep3_ring_share) => match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
},
Shared::Ring16(rep3_ring_share) => match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
},
Shared::Ring8(rep3_ring_share) => match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
},
Shared::Ring1(rep3_ring_share) => match integer_bit_size {
IntegerBitSize::U1 => Ok(Rep3BrilligType::Shared(Shared::Ring1(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U8 => Ok(Rep3BrilligType::Shared(Shared::Ring8(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U16 => Ok(Rep3BrilligType::Shared(Shared::Ring16(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U32 => Ok(Rep3BrilligType::Shared(Shared::Ring32(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U64 => Ok(Rep3BrilligType::Shared(Shared::Ring64(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
IntegerBitSize::U128 => Ok(Rep3BrilligType::Shared(Shared::Ring128(
rep3_ring::casts::ring_cast_selector(
rep3_ring_share,
&mut self.io_context,
)?,
))),
},
Shared::Ring128(rep3_ring_share) => {
cast_ring(rep3_ring_share, integer_bit_size, &mut self.io_context)
}
Shared::Ring64(rep3_ring_share) => {
cast_ring(rep3_ring_share, integer_bit_size, &mut self.io_context)
}
Shared::Ring32(rep3_ring_share) => {
cast_ring(rep3_ring_share, integer_bit_size, &mut self.io_context)
}
Shared::Ring16(rep3_ring_share) => {
cast_ring(rep3_ring_share, integer_bit_size, &mut self.io_context)
}
Shared::Ring8(rep3_ring_share) => {
cast_ring(rep3_ring_share, integer_bit_size, &mut self.io_context)
}
Shared::Ring1(rep3_ring_share) => {
cast_ring(rep3_ring_share, integer_bit_size, &mut self.io_context)
}
},
(Rep3BrilligType::Public(public), BitSize::Field) => {
let casted = self.plain_driver.cast(public, bit_size)?;
Expand Down

0 comments on commit ecbe468

Please sign in to comment.