binius_field/arch/portable/
packed_64.rs

1// Copyright 2024-2025 Irreducible Inc.
2
3use cfg_if::cfg_if;
4
5use super::{
6	packed::{impl_broadcast, impl_ops_for_zero_height, PackedPrimitiveType},
7	packed_arithmetic::{alphas, impl_tower_constants},
8};
9use crate::{
10	arch::{PackedStrategy, PairwiseRecursiveStrategy, PairwiseStrategy},
11	arithmetic_traits::{
12		impl_invert_with, impl_mul_alpha_with, impl_mul_with, impl_square_with,
13		impl_transformation_with_strategy,
14	},
15	BinaryField16b, BinaryField1b, BinaryField2b, BinaryField32b, BinaryField4b, BinaryField64b,
16	BinaryField8b,
17};
18
19// Define 64 bit packed field types
20pub type PackedBinaryField64x1b = PackedPrimitiveType<u64, BinaryField1b>;
21pub type PackedBinaryField32x2b = PackedPrimitiveType<u64, BinaryField2b>;
22pub type PackedBinaryField16x4b = PackedPrimitiveType<u64, BinaryField4b>;
23pub type PackedBinaryField8x8b = PackedPrimitiveType<u64, BinaryField8b>;
24pub type PackedBinaryField4x16b = PackedPrimitiveType<u64, BinaryField16b>;
25pub type PackedBinaryField2x32b = PackedPrimitiveType<u64, BinaryField32b>;
26pub type PackedBinaryField1x64b = PackedPrimitiveType<u64, BinaryField64b>;
27
28// Define broadcast
29impl_broadcast!(u64, BinaryField1b);
30impl_broadcast!(u64, BinaryField2b);
31impl_broadcast!(u64, BinaryField4b);
32impl_broadcast!(u64, BinaryField8b);
33impl_broadcast!(u64, BinaryField16b);
34impl_broadcast!(u64, BinaryField32b);
35impl_broadcast!(u64, BinaryField64b);
36
37// Define operations for height 0
38impl_ops_for_zero_height!(PackedBinaryField64x1b);
39
40// Define constants
41impl_tower_constants!(BinaryField1b, u64, { alphas!(u64, 0) });
42impl_tower_constants!(BinaryField2b, u64, { alphas!(u64, 1) });
43impl_tower_constants!(BinaryField4b, u64, { alphas!(u64, 2) });
44impl_tower_constants!(BinaryField8b, u64, { alphas!(u64, 3) });
45impl_tower_constants!(BinaryField16b, u64, { alphas!(u64, 4) });
46impl_tower_constants!(BinaryField32b, u64, { alphas!(u64, 5) });
47
48// Define multiplication
49impl_mul_with!(PackedBinaryField32x2b @ PackedStrategy);
50impl_mul_with!(PackedBinaryField16x4b @ PackedStrategy);
51cfg_if! {
52	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
53		impl_mul_with!(PackedBinaryField8x8b => crate::PackedBinaryField16x8b);
54		impl_mul_with!(PackedBinaryField4x16b => crate::PackedBinaryField8x16b);
55		impl_mul_with!(PackedBinaryField2x32b => crate::PackedBinaryField4x32b);
56		impl_mul_with!(PackedBinaryField1x64b => crate::PackedBinaryField2x64b);
57	} else {
58		impl_mul_with!(PackedBinaryField8x8b @ crate::arch::PairwiseTableStrategy);
59		impl_mul_with!(PackedBinaryField4x16b @ PairwiseRecursiveStrategy);
60		impl_mul_with!(PackedBinaryField2x32b @ PairwiseRecursiveStrategy);
61		impl_mul_with!(PackedBinaryField1x64b @ PairwiseRecursiveStrategy);
62	}
63}
64
65// Define square
66impl_square_with!(PackedBinaryField32x2b @ PackedStrategy);
67impl_square_with!(PackedBinaryField16x4b @ PackedStrategy);
68cfg_if! {
69	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
70		impl_square_with!(PackedBinaryField8x8b => crate::PackedBinaryField16x8b);
71		impl_square_with!(PackedBinaryField4x16b => crate::PackedBinaryField8x16b);
72		impl_square_with!(PackedBinaryField2x32b => crate::PackedBinaryField4x32b);
73		impl_square_with!(PackedBinaryField1x64b => crate::PackedBinaryField2x64b);
74	} else {
75		impl_square_with!(PackedBinaryField8x8b @ crate::arch::PairwiseTableStrategy);
76		impl_square_with!(PackedBinaryField4x16b @ PairwiseStrategy);
77		impl_square_with!(PackedBinaryField2x32b @ PairwiseRecursiveStrategy);
78		impl_square_with!(PackedBinaryField1x64b @ crate::arch::HybridRecursiveStrategy);
79	}
80}
81
82// Define invert
83impl_invert_with!(PackedBinaryField32x2b @ PackedStrategy);
84impl_invert_with!(PackedBinaryField16x4b @ PackedStrategy);
85cfg_if! {
86	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
87		impl_invert_with!(PackedBinaryField8x8b => crate::PackedBinaryField16x8b);
88		impl_invert_with!(PackedBinaryField4x16b => crate::PackedBinaryField8x16b);
89		impl_invert_with!(PackedBinaryField2x32b => crate::PackedBinaryField4x32b);
90		impl_invert_with!(PackedBinaryField1x64b => crate::PackedBinaryField2x64b);
91	} else {
92		impl_invert_with!(PackedBinaryField8x8b @ crate::arch::PairwiseTableStrategy);
93		impl_invert_with!(PackedBinaryField4x16b @ PairwiseStrategy);
94		impl_invert_with!(PackedBinaryField2x32b @ PairwiseStrategy);
95		impl_invert_with!(PackedBinaryField1x64b @ PairwiseRecursiveStrategy);
96	}
97}
98
99// Define multiply by alpha
100impl_mul_alpha_with!(PackedBinaryField32x2b @ PackedStrategy);
101impl_mul_alpha_with!(PackedBinaryField16x4b @ PackedStrategy);
102impl_mul_alpha_with!(PackedBinaryField8x8b @ PackedStrategy);
103impl_mul_alpha_with!(PackedBinaryField4x16b @ PackedStrategy);
104impl_mul_alpha_with!(PackedBinaryField2x32b @ PackedStrategy);
105impl_mul_alpha_with!(PackedBinaryField1x64b @ PairwiseRecursiveStrategy);
106
107// Define linear transformations
108impl_transformation_with_strategy!(PackedBinaryField64x1b, PackedStrategy);
109impl_transformation_with_strategy!(PackedBinaryField32x2b, PackedStrategy);
110impl_transformation_with_strategy!(PackedBinaryField16x4b, PackedStrategy);
111impl_transformation_with_strategy!(PackedBinaryField8x8b, PackedStrategy);
112impl_transformation_with_strategy!(PackedBinaryField4x16b, PackedStrategy);
113impl_transformation_with_strategy!(PackedBinaryField2x32b, PackedStrategy);
114impl_transformation_with_strategy!(PackedBinaryField1x64b, PairwiseStrategy);