avx.rs - source

Files

ahash

convert.rs fallback_hash.rs folded_multiply.rs lib.rs random_state.rs

aho_corasick

packed

teddy

compile.rs mod.rs runtime.rs

api.rs mod.rs pattern.rs rabinkarp.rs vector.rs

ahocorasick.rs automaton.rs buffer.rs byte_frequencies.rs classes.rs dfa.rs error.rs lib.rs nfa.rs prefilter.rs state_id.rs

ansi_term

ansi.rs debug.rs difference.rs display.rs lib.rs style.rs util.rs windows.rs write.rs

anyhow

backtrace.rs chain.rs context.rs ensure.rs error.rs fmt.rs kind.rs lib.rs macros.rs ptr.rs wrapper.rs

atty

bech32

bincode

config

endian.rs int.rs legacy.rs limit.rs mod.rs trailing.rs

de

ser

byteorder.rs error.rs internal.rs lib.rs

bit_set

bit_vec

bitcoin

bitcoin_hashes

cmp.rs error.rs hash160.rs hex.rs hmac.rs impls.rs lib.rs ripemd160.rs serde_macros.rs sha1.rs sha256.rs sha256d.rs sha256t.rs sha512.rs siphash24.rs util.rs

bitflags

cfg_if

clap

convert_case

case.rs converter.rs lib.rs pattern.rs segmentation.rs

core2

io

cursor.rs error.rs impls.rs mod.rs traits.rs util.rs

error.rs lib.rs

crunchy

cryptoxide

enum_primitive

fancy_regex

analyze.rs compile.rs error.rs expand.rs lib.rs parse.rs replacer.rs vm.rs

hashbrown

external_trait_impls

raw

bitmask.rs mod.rs sse2.rs

lib.rs macros.rs map.rs scopeguard.rs set.rs

hex

error.rs lib.rs

hex_literal

comments.rs lib.rs

itoa

lib.rs udiv128.rs

libc

unix

linux_like

linux

arch

generic

gnu

b64

x86_64

align.rs mod.rs not_x32.rs

align.rs mod.rs

align.rs mod.rs non_exhaustive.rs

align.rs mod.rs

fixed_width_ints.rs lib.rs macros.rs

libloading

os

unix

changelog.rs lib.rs util.rs

memchr

memchr

x86

avx.rs mod.rs sse2.rs

fallback.rs iter.rs mod.rs naive.rs

memmem

prefilter

x86

avx.rs mod.rs sse.rs

fallback.rs genericsimd.rs mod.rs

x86

avx.rs mod.rs sse.rs

byte_frequencies.rs genericsimd.rs mod.rs rabinkarp.rs rarebytes.rs twoway.rs util.rs vector.rs

num

num_bigint

num_complex

cast.rs lib.rs pow.rs

num_integer

average.rs lib.rs roots.rs

num_iter

num_rational

num_traits

ops

checked.rs inv.rs mod.rs mul_add.rs overflowing.rs saturating.rs wrapping.rs

bounds.rs cast.rs float.rs identities.rs int.rs lib.rs macros.rs pow.rs real.rs sign.rs

ordered_float

paste

attr.rs error.rs lib.rs segment.rs

proc_macro2

detection.rs fallback.rs lib.rs marker.rs parse.rs wrapper.rs

proc_macro_error

imp

diagnostic.rs dummy.rs lib.rs macros.rs sealed.rs

proc_macro_error_attr

lib.rs parse.rs settings.rs

qimalloc

quote

ext.rs format.rs ident_fragment.rs lib.rs runtime.rs spanned.rs to_tokens.rs

regex

literal

backtrack.rs compile.rs dfa.rs error.rs exec.rs expand.rs find_byte.rs input.rs lib.rs pikevm.rs pool.rs prog.rs re_builder.rs re_bytes.rs re_set.rs re_trait.rs re_unicode.rs sparse.rs utf8.rs

regex_syntax

remain

atom.rs check.rs compare.rs emit.rs format.rs lib.rs parse.rs visit.rs

rust_ssvm

ryu

buffer

pretty

exponent.rs mantissa.rs mod.rs

common.rs d2s.rs d2s_full_table.rs d2s_intrinsics.rs digit_table.rs f2s.rs f2s_intrinsics.rs lib.rs

secp256k1

constants.rs context.rs ecdh.rs key.rs lib.rs macros.rs schnorrsig.rs

secp256k1_sys

lib.rs macros.rs types.rs

serde

de

format.rs ignored_any.rs impls.rs mod.rs seed.rs utf8.rs value.rs

private

de.rs doc.rs mod.rs ser.rs size_hint.rs

ser

fmt.rs impls.rs impossible.rs mod.rs

integer128.rs lib.rs macros.rs

serde_derive

internals

ast.rs attr.rs case.rs check.rs ctxt.rs mod.rs receiver.rs respan.rs symbol.rs

bound.rs de.rs dummy.rs fragment.rs lib.rs pretend.rs ser.rs try.rs

serde_json

features_check

io

value

de.rs from.rs index.rs mod.rs partial_eq.rs ser.rs

de.rs error.rs iter.rs lib.rs macros.rs map.rs number.rs read.rs ser.rs

serde_value

de.rs lib.rs ser.rs

sewup

kv

traits

key.rs mod.rs value.rs vec.rs

bucket.rs errors.rs mod.rs

rdb

db.rs errors.rs mod.rs table.rs traits.rs

runtimes

handler.rs mod.rs test.rs traits.rs

token

erc1155.rs erc20.rs erc721.rs helpers.rs mod.rs

types

address.rs errors.rs mod.rs raw.rs row.rs sized_str.rs

errors.rs lib.rs primitives.rs utils.rs

sewup_derive

ss_ewasm_api

lib.rs native.rs types.rs utils.rs

ssvm_evmc_client

host.rs lib.rs loader.rs types.rs

ssvm_evmc_sys

strsim

syn

textwrap

indentation.rs lib.rs splitting.rs

thiserror

aserror.rs display.rs lib.rs

thiserror_impl

ast.rs attr.rs expand.rs fmt.rs generics.rs lib.rs prop.rs valid.rs

tiny_keccak

keccak.rs keccakf.rs lib.rs sha3.rs

toml

datetime.rs de.rs lib.rs macros.rs map.rs ser.rs spanned.rs tokens.rs value.rs

unicode_width

lib.rs tables.rs

unicode_xid

lib.rs tables.rs

vec_map

use core::arch::x86_64::__m256i;

use crate::memmem::{
    prefilter::{PrefilterFnTy, PrefilterState},
    NeedleInfo,
};

// Check that the functions below satisfy the Prefilter function type.
const _: PrefilterFnTy = find;

/// An AVX2 accelerated candidate finder for single-substring search.
///
/// # Safety
///
/// Callers must ensure that the avx2 CPU feature is enabled in the current
/// environment.
#[target_feature(enable = "avx2")]
pub(crate) unsafe fn find(
    prestate: &mut PrefilterState,
    ninfo: &NeedleInfo,
    haystack: &[u8],
    needle: &[u8],
) -> Option<usize> {
    super::super::genericsimd::find::<__m256i>(
        prestate,
        ninfo,
        haystack,
        needle,
        super::sse::find,
    )
}

#[cfg(test)]
mod tests {
    #[test]
    #[cfg(not(miri))]
    fn prefilter_permutations() {
        use crate::memmem::prefilter::tests::PrefilterTest;
        if !is_x86_feature_detected!("avx2") {
            return;
        }
        // SAFETY: The safety of super::find only requires that the current
        // CPU support AVX2, which we checked above.
        unsafe { PrefilterTest::run_all_tests(super::find) };
    }
}