Expand description
AVX2 and AVX-512 SIMD Implementations
Provides x86_64-specific SIMD implementations for CNN operations. Includes AVX2 (8 floats) and AVX-512 (16 floats) variants.
Functions§
- batch_
norm_ ⚠avx2 - AVX2 batch normalization
- conv_
3x3_ ⚠avx2 - AVX2 3x3 convolution (without FMA, 4x loop unrolling + multiple accumulators)
- conv_
3x3_ ⚠avx2_ fma - AVX2 3x3 convolution with FMA (4x loop unrolling + multiple accumulators)
- depthwise_
conv_ ⚠3x3_ avx2 - AVX2 depthwise 3x3 convolution (kernel position unrolling)
- dot_
product_ ⚠avx2 - AVX2 dot product without FMA
- dot_
product_ ⚠avx2_ fma - AVX2 dot product with FMA
- dot_
product_ ⚠avx512 - AVX-512 dot product
- global_
avg_ ⚠pool_ avx2 - AVX2 global average pooling
- max_
pool_ ⚠2x2_ avx2 - AVX2 max pooling 2x2
- relu6_
avx2 ⚠ - AVX2 ReLU6 activation
- relu_
avx2 ⚠ - AVX2 ReLU activation