binius_field/arch/portable/
packed_32.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, BinaryField8b,
16};
17
18// Define 32 bit packed field types
19pub type PackedBinaryField32x1b = PackedPrimitiveType<u32, BinaryField1b>;
20pub type PackedBinaryField16x2b = PackedPrimitiveType<u32, BinaryField2b>;
21pub type PackedBinaryField8x4b = PackedPrimitiveType<u32, BinaryField4b>;
22pub type PackedBinaryField4x8b = PackedPrimitiveType<u32, BinaryField8b>;
23pub type PackedBinaryField2x16b = PackedPrimitiveType<u32, BinaryField16b>;
24pub type PackedBinaryField1x32b = PackedPrimitiveType<u32, BinaryField32b>;
25
26// Define broadcast
27impl_broadcast!(u32, BinaryField1b);
28impl_broadcast!(u32, BinaryField2b);
29impl_broadcast!(u32, BinaryField4b);
30impl_broadcast!(u32, BinaryField8b);
31impl_broadcast!(u32, BinaryField16b);
32impl_broadcast!(u32, BinaryField32b);
33
34// Define operations for height 0
35impl_ops_for_zero_height!(PackedBinaryField32x1b);
36
37// Define constants
38impl_tower_constants!(BinaryField1b, u32, { alphas!(u32, 0) });
39impl_tower_constants!(BinaryField2b, u32, { alphas!(u32, 1) });
40impl_tower_constants!(BinaryField4b, u32, { alphas!(u32, 2) });
41impl_tower_constants!(BinaryField8b, u32, { alphas!(u32, 3) });
42impl_tower_constants!(BinaryField16b, u32, { alphas!(u32, 4) });
43
44// Define multiplication
45impl_mul_with!(PackedBinaryField16x2b @ PackedStrategy);
46impl_mul_with!(PackedBinaryField8x4b @ PackedStrategy);
47cfg_if! {
48	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
49		impl_mul_with!(PackedBinaryField4x8b => crate::PackedBinaryField16x8b);
50		impl_mul_with!(PackedBinaryField2x16b => crate::PackedBinaryField8x16b);
51		impl_mul_with!(PackedBinaryField1x32b => crate::PackedBinaryField4x32b);
52	} else {
53		use crate::arch::PairwiseTableStrategy;
54
55		impl_mul_with!(PackedBinaryField4x8b @ PairwiseTableStrategy);
56		impl_mul_with!(PackedBinaryField2x16b @ PairwiseRecursiveStrategy);
57		impl_mul_with!(PackedBinaryField1x32b @ PairwiseRecursiveStrategy);
58	}
59}
60
61// Define square
62impl_square_with!(PackedBinaryField16x2b @ PackedStrategy);
63impl_square_with!(PackedBinaryField8x4b @ PackedStrategy);
64impl_square_with!(PackedBinaryField4x8b @ PackedStrategy);
65impl_square_with!(PackedBinaryField2x16b @ PackedStrategy);
66impl_square_with!(PackedBinaryField1x32b @ PairwiseRecursiveStrategy);
67
68// Define invert
69impl_invert_with!(PackedBinaryField16x2b @ PairwiseRecursiveStrategy);
70impl_invert_with!(PackedBinaryField8x4b @ PairwiseRecursiveStrategy);
71cfg_if! {
72	if #[cfg(all(target_arch = "x86_64", target_feature = "sse2", target_feature = "gfni"))] {
73		impl_invert_with!(PackedBinaryField4x8b => crate::PackedBinaryField16x8b);
74		impl_invert_with!(PackedBinaryField2x16b => crate::PackedBinaryField8x16b);
75		impl_invert_with!(PackedBinaryField1x32b => crate::PackedBinaryField4x32b);
76	} else {
77		impl_invert_with!(PackedBinaryField4x8b @ PairwiseTableStrategy);
78		impl_invert_with!(PackedBinaryField2x16b @ PairwiseStrategy);
79		impl_invert_with!(PackedBinaryField1x32b @ PairwiseRecursiveStrategy);
80	}
81}
82
83// Define multiply by alpha
84impl_mul_alpha_with!(PackedBinaryField16x2b @ PackedStrategy);
85impl_mul_alpha_with!(PackedBinaryField8x4b @ PackedStrategy);
86impl_mul_alpha_with!(PackedBinaryField4x8b @ PackedStrategy);
87impl_mul_alpha_with!(PackedBinaryField2x16b @ PackedStrategy);
88impl_mul_alpha_with!(PackedBinaryField1x32b @ PairwiseRecursiveStrategy);
89
90// Define linear transformations
91impl_transformation_with_strategy!(PackedBinaryField32x1b, PackedStrategy);
92impl_transformation_with_strategy!(PackedBinaryField16x2b, PackedStrategy);
93impl_transformation_with_strategy!(PackedBinaryField8x4b, PackedStrategy);
94impl_transformation_with_strategy!(PackedBinaryField4x8b, PackedStrategy);
95impl_transformation_with_strategy!(PackedBinaryField2x16b, PackedStrategy);
96impl_transformation_with_strategy!(PackedBinaryField1x32b, PairwiseStrategy);
binius_field/arch/portable/packed_32.rs

binius_field/arch/portable/
packed_32.rs
