binius_field/arch/portable/
packed_aes_64.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, AESTowerField64b, AESTowerField8b,
16};
17
18// Define 64 bit packed field types
19pub type PackedAESBinaryField8x8b = PackedPrimitiveType<u64, AESTowerField8b>;
20pub type PackedAESBinaryField4x16b = PackedPrimitiveType<u64, AESTowerField16b>;
21pub type PackedAESBinaryField2x32b = PackedPrimitiveType<u64, AESTowerField32b>;
22pub type PackedAESBinaryField1x64b = PackedPrimitiveType<u64, AESTowerField64b>;
23
24// Define broadcast
25impl_broadcast!(u64, AESTowerField8b);
26impl_broadcast!(u64, AESTowerField16b);
27impl_broadcast!(u64, AESTowerField32b);
28impl_broadcast!(u64, AESTowerField64b);
29
30// Define constants
31impl_tower_constants!(AESTowerField8b, u64, 0x00d300d300d300d3);
32impl_tower_constants!(AESTowerField16b, u64, { alphas!(u64, 4) });
33impl_tower_constants!(AESTowerField32b, u64, { alphas!(u64, 5) });
34
35// Define multiplication
36cfg_if! {
37	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
38		impl_mul_with!(PackedAESBinaryField8x8b => crate::PackedAESBinaryField16x8b);
39		impl_mul_with!(PackedAESBinaryField4x16b => crate::PackedAESBinaryField8x16b);
40		impl_mul_with!(PackedAESBinaryField2x32b => crate::PackedAESBinaryField4x32b);
41		impl_mul_with!(PackedAESBinaryField1x64b => crate::PackedAESBinaryField2x64b);
42	} else {
43		impl_mul_with!(PackedAESBinaryField8x8b @ PairwiseTableStrategy);
44		impl_mul_with!(PackedAESBinaryField4x16b @ PairwiseRecursiveStrategy);
45		impl_mul_with!(PackedAESBinaryField2x32b @ PairwiseRecursiveStrategy);
46		impl_mul_with!(PackedAESBinaryField1x64b @ PairwiseRecursiveStrategy);
47	}
48}
49
50// Define square
51cfg_if! {
52	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
53		impl_square_with!(PackedAESBinaryField8x8b => crate::PackedAESBinaryField16x8b);
54		impl_square_with!(PackedAESBinaryField4x16b => crate::PackedAESBinaryField8x16b);
55		impl_square_with!(PackedAESBinaryField2x32b => crate::PackedAESBinaryField4x32b);
56		impl_square_with!(PackedAESBinaryField1x64b => crate::PackedAESBinaryField2x64b);
57	} else {
58		impl_square_with!(PackedAESBinaryField8x8b @ PairwiseTableStrategy);
59		impl_square_with!(PackedAESBinaryField4x16b @ PairwiseRecursiveStrategy);
60		impl_square_with!(PackedAESBinaryField2x32b @ PairwiseRecursiveStrategy);
61		impl_square_with!(PackedAESBinaryField1x64b @ PairwiseRecursiveStrategy);
62	}
63}
64
65// Define invert
66cfg_if! {
67	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
68		impl_invert_with!(PackedAESBinaryField8x8b => crate::PackedAESBinaryField16x8b);
69		impl_invert_with!(PackedAESBinaryField4x16b => crate::PackedAESBinaryField8x16b);
70		impl_invert_with!(PackedAESBinaryField2x32b => crate::PackedAESBinaryField4x32b);
71		impl_invert_with!(PackedAESBinaryField1x64b => crate::PackedAESBinaryField2x64b);
72	} else {
73		impl_invert_with!(PackedAESBinaryField8x8b @ PairwiseTableStrategy);
74		impl_invert_with!(PackedAESBinaryField4x16b @ PairwiseRecursiveStrategy);
75		impl_invert_with!(PackedAESBinaryField2x32b @ PairwiseRecursiveStrategy);
76		impl_invert_with!(PackedAESBinaryField1x64b @ PairwiseRecursiveStrategy);
77	}
78}
79
80// Define multiply by alpha
81impl_mul_alpha_with!(PackedAESBinaryField8x8b @ PairwiseTableStrategy);
82impl_mul_alpha_with!(PackedAESBinaryField4x16b @ PackedStrategy);
83impl_mul_alpha_with!(PackedAESBinaryField2x32b @ PackedStrategy);
84impl_mul_alpha_with!(PackedAESBinaryField1x64b @ PairwiseRecursiveStrategy);
85
86// Define linear transformations
87impl_transformation_with_strategy!(PackedAESBinaryField8x8b, PackedStrategy);
88impl_transformation_with_strategy!(PackedAESBinaryField4x16b, PackedStrategy);
89impl_transformation_with_strategy!(PackedAESBinaryField2x32b, PackedStrategy);
90impl_transformation_with_strategy!(PackedAESBinaryField1x64b, PairwiseStrategy);
binius_field/arch/portable/packed_aes_64.rs

binius_field/arch/portable/
packed_aes_64.rs
