megenginelite-sys 1.8.2

A safe megenginelite wrapper in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

/**
 * \file dnn/src/x86/simd_macro/fma_helper.h
 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
 *
 * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */
#include <immintrin.h>
#include <xmmintrin.h>

#define MEGDNN_SIMD_NAME              FMA
#define MEGDNN_SIMD_TARGET            fma
#define MEGDNN_SIMD_ATTRIBUTE_TARGET  MEGDNN_ATTRIBUTE_TARGET("fma")
#define MEGDNN_SIMD_WIDTH             8
#define MEGDNN_SIMD_TYPE              __m256
#define MEGDNN_SIMD_LOADU(addr)       _mm256_loadu_ps(addr)
#define MEGDNN_SIMD_STOREU(addr, reg) _mm256_storeu_ps(addr, reg)
#define MEGDNN_SIMD_SETZERO()         _mm256_setzero_ps()
#define MEGDNN_SIMD_SET1(num)         _mm256_set1_ps(num)
#define MEGDNN_SIMD_FMADD(a, b, c)    _mm256_fmadd_ps(a, b, c)

#define MEGDNN_SIMD_ADD(a, b)             _mm256_add_ps(a, b)
#define MEGDNN_SIMD_SUB(a, b)             _mm256_sub_ps(a, b)
#define MEGDNN_SIMD_MUL(a, b)             _mm256_mul_ps(a, b)
#define MEGDNN_SIMD_FNMADD(a, b, c)       _mm256_fnmadd_ps(a, b, c)
#define MEGDNN_SIMD_UNPACKLO(a, b)        _mm256_unpacklo_ps(a, b)
#define MEGDNN_SIMD_UNPACKHI(a, b)        _mm256_unpackhi_ps(a, b)
#define MEGDNN_SIMD_SHUFFLE(a, b, c)      _mm256_shuffle_ps(a, b, c)
#define MEGDNN_SIMD_BLEND(a, b, c)        _mm256_blend_ps(a, b, c)
#define MEGDNN_SIMD_PERMUTE2F128(a, b, c) _mm256_permute2f128_ps(a, b, c)