/src/Simd/src/Simd/SimdAvx512bwSynetConvolution32f.cpp
Line | Count | Source |
1 | | /* |
2 | | * Simd Library (http://ermig1979.github.io/Simd). |
3 | | * |
4 | | * Copyright (c) 2011-2024 Yermalayeu Ihar. |
5 | | * |
6 | | * Permission is hereby granted, free of charge, to any person obtaining a copy |
7 | | * of this software and associated documentation files (the "Software"), to deal |
8 | | * in the Software without restriction, including without limitation the rights |
9 | | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
10 | | * copies of the Software, and to permit persons to whom the Software is |
11 | | * furnished to do so, subject to the following conditions: |
12 | | * |
13 | | * The above copyright notice and this permission notice shall be included in |
14 | | * all copies or substantial portions of the Software. |
15 | | * |
16 | | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
17 | | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
18 | | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
19 | | * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
20 | | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
21 | | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
22 | | * SOFTWARE. |
23 | | */ |
24 | | #include "Simd/SimdSynetConvolution32f.h" |
25 | | |
26 | | namespace Simd |
27 | | { |
28 | | #if defined(SIMD_AVX512BW_ENABLE) && defined(SIMD_SYNET_ENABLE) |
29 | | namespace Avx512bw |
30 | | { |
31 | | void * SynetConvolution32fInit(size_t batch, const SimdConvolutionParameters * conv) |
32 | 0 | { |
33 | 0 | ConvParam param(batch, conv, SimdSynetCompatibilityDefault); |
34 | 0 | if (!param.Valid(SimdTensorData32f)) |
35 | 0 | return NULL; |
36 | 0 | if (Avx2::SynetConvolution32fDepthwiseDotProduct::Preferable(param)) |
37 | 0 | return new Avx2::SynetConvolution32fDepthwiseDotProduct(param); |
38 | 0 | else if (SynetConvolution32fWinograd::Preferable(param)) |
39 | 0 | return new SynetConvolution32fWinograd(param); |
40 | 0 | else if (SynetConvolution32fGemmNT::Preferable(param)) |
41 | 0 | return new SynetConvolution32fGemmNT(param); |
42 | 0 | else if (SynetConvolution32fDirectNchw::Preferable(param)) |
43 | 0 | return new SynetConvolution32fDirectNchw(param); |
44 | 0 | else if (SynetConvolution32fNhwcDirect::Preferable(param)) |
45 | 0 | return new SynetConvolution32fNhwcDirect(param); |
46 | 0 | else if (SynetConvolution32fNhwcDepthwise::Preferable(param)) |
47 | 0 | return new SynetConvolution32fNhwcDepthwise(param); |
48 | 0 | else if (SynetConvolution32fNhwcGroupedBlock1x2::Preferable(param)) |
49 | 0 | return new SynetConvolution32fNhwcGroupedBlock1x2(param); |
50 | 0 | else |
51 | 0 | return new SynetConvolution32fGemmNN(param); |
52 | 0 | } |
53 | | } |
54 | | #endif |
55 | | } |