binius_field/arch/portable/
packed_aes_32.rs

1// Copyright 2024-2025 Irreducible Inc.
2
3use cfg_if::cfg_if;
4
5use super::{
6	packed::{impl_broadcast, PackedPrimitiveType},
7	packed_arithmetic::{alphas, impl_tower_constants},
8};
9use crate::{
10	arch::{PackedStrategy, PairwiseRecursiveStrategy, PairwiseStrategy, PairwiseTableStrategy},
11	arithmetic_traits::{
12		impl_invert_with, impl_mul_alpha_with, impl_mul_with, impl_square_with,
13		impl_transformation_with_strategy,
14	},
15	AESTowerField16b, AESTowerField32b, AESTowerField8b,
16};
17
18// Define 32 bit packed field types
19pub type PackedAESBinaryField4x8b = PackedPrimitiveType<u32, AESTowerField8b>;
20pub type PackedAESBinaryField2x16b = PackedPrimitiveType<u32, AESTowerField16b>;
21pub type PackedAESBinaryField1x32b = PackedPrimitiveType<u32, AESTowerField32b>;
22
23// Define broadcast
24impl_broadcast!(u32, AESTowerField8b);
25impl_broadcast!(u32, AESTowerField16b);
26impl_broadcast!(u32, AESTowerField32b);
27
28// Define constants
29impl_tower_constants!(AESTowerField8b, u32, 0x00d300d3);
30impl_tower_constants!(AESTowerField16b, u32, { alphas!(u32, 4) });
31
32// Define multiplication
33cfg_if! {
34	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
35		impl_mul_with!(PackedAESBinaryField4x8b => crate::PackedAESBinaryField16x8b);
36		impl_mul_with!(PackedAESBinaryField2x16b => crate::PackedAESBinaryField8x16b);
37		impl_mul_with!(PackedAESBinaryField1x32b => crate::PackedAESBinaryField4x32b);
38	} else {
39		impl_mul_with!(PackedAESBinaryField4x8b @ PairwiseTableStrategy);
40		impl_mul_with!(PackedAESBinaryField2x16b @ PairwiseRecursiveStrategy);
41		impl_mul_with!(PackedAESBinaryField1x32b @ PairwiseRecursiveStrategy);
42	}
43}
44
45// Define square
46impl_square_with!(PackedAESBinaryField4x8b @ PairwiseTableStrategy);
47impl_square_with!(PackedAESBinaryField2x16b @ PairwiseRecursiveStrategy);
48impl_square_with!(PackedAESBinaryField1x32b @ PairwiseRecursiveStrategy);
49
50// Define invert
51cfg_if! {
52	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
53		impl_invert_with!(PackedAESBinaryField4x8b => crate::PackedAESBinaryField16x8b);
54		impl_invert_with!(PackedAESBinaryField2x16b => crate::PackedAESBinaryField8x16b);
55		impl_invert_with!(PackedAESBinaryField1x32b => crate::PackedAESBinaryField4x32b);
56	} else {
57		impl_invert_with!(PackedAESBinaryField4x8b @ PairwiseTableStrategy);
58		impl_invert_with!(PackedAESBinaryField2x16b @ PairwiseRecursiveStrategy);
59		impl_invert_with!(PackedAESBinaryField1x32b @ PairwiseRecursiveStrategy);
60	}
61}
62
63// Define multiply by alpha
64impl_mul_alpha_with!(PackedAESBinaryField4x8b @ PairwiseTableStrategy);
65impl_mul_alpha_with!(PackedAESBinaryField2x16b @ PackedStrategy);
66impl_mul_alpha_with!(PackedAESBinaryField1x32b @ PairwiseRecursiveStrategy);
67
68// Define linear transformations
69impl_transformation_with_strategy!(PackedAESBinaryField4x8b, PackedStrategy);
70impl_transformation_with_strategy!(PackedAESBinaryField2x16b, PackedStrategy);
71impl_transformation_with_strategy!(PackedAESBinaryField1x32b, PairwiseStrategy);
binius_field/arch/portable/packed_aes_32.rs

binius_field/arch/portable/
packed_aes_32.rs
