1 files changed, 17 insertions, 0 deletions
diff --git a/src/kernel.rs b/src/kernel.rs
index cd81bb5..a6cca9f 100644
--- a/src/kernel.rs
+++ b/src/kernel.rs
@@ -49,6 +49,14 @@ extern "C" {
         flags: u32,
         out: *mut [u8; 64],
     );
+    pub fn blake3_avx2_xof_stream_2(
+        cv: &[u32; 8],
+        block: &[u8; 64],
+        counter: u64,
+        block_len: u32,
+        flags: u32,
+        out: *mut [u8; 64 * 2],
+    );
     pub fn blake3_avx512_xof_stream_2(
         cv: &[u32; 8],
         block: &[u8; 64],
@@ -195,6 +203,15 @@ mod test {
 
     #[test]
     #[cfg(target_arch = "x86_64")]
+    fn test_avx2_xof_2() {
+        if !is_x86_feature_detected!("avx2") {
+            return;
+        }
+        test_xof_function(blake3_avx2_xof_stream_2);
+    }
+
+    #[test]
+    #[cfg(target_arch = "x86_64")]
     fn test_avx512_xof_2() {
         if !is_x86_feature_detected!("avx512f") || !is_x86_feature_detected!("avx512vl") {
             return;