1d004b542a
* libnd4j first step of mkldnn for xw_plus_b and test of aurora crash in imageHelper * libnd4j sync folders with master * libnd4j merge master, raw implementation of xw_plus_b on mkldnn, clean up, need testing and adding checks for corresponded input shapes * libnd4j corrections and checks added to xw_plus_b mkl * libnd4j corrected dataType description based on mkl operation description, need more investigation * libnd4j fixe xw_blus_b mkl implementation, need testing Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j two unit tests added Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j fixed check input dimensions bug Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libndj4 one more test added to cover different order handling Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j added optional int arg support to define weights format, if arg == 1, mkldnn (do not need transpose in mkldnn implementation), else mmul weights format, corrected check points, added unit test Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j merge master Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j some improvements to avoid NDArray transpose in xw_plus_b operation Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j fixed issues connected with weights rank, also added support of one case based on tf (for mkldnn, cpu, cuda), test case added Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j added proper handling of empty inputs (all implementations) * libnd4j fixed compilation error * libnd4j several more corrections after conflict solve and fixed typos Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j removed unsupported data types Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j merge master and fixed issues Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j added propagation implementation for xw_plus_b, fixed issue connected with mkl weights data format, avoided data copy in transpose mode, test cases added, manually tested with gradCheck Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j one minor fix of double operation declaration Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j code clean up Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j minor tests fixes Signed-off-by: Oleg <oleg.semeniv@gmail.com> * libnd4j fixed build problem, integrate helpers changes Signed-off-by: Oleg <oleg.semeniv@gmail.com> Co-authored-by: raver119 <raver119@gmail.com>
89 lines
3.0 KiB
C++
89 lines
3.0 KiB
C++
/*******************************************************************************
|
|
* Copyright (c) 2015-2018 Skymind, Inc.
|
|
*
|
|
* This program and the accompanying materials are made available under the
|
|
* terms of the Apache License, Version 2.0 which is available at
|
|
* https://www.apache.org/licenses/LICENSE-2.0.
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
******************************************************************************/
|
|
|
|
|
|
//
|
|
// @author raver119@gmail.com
|
|
//
|
|
|
|
#include "testlayers.h"
|
|
#include <initializer_list>
|
|
#include <ops/declarable/PlatformHelper.h>
|
|
|
|
#ifdef HAVE_MKLDNN
|
|
|
|
#include <ops/declarable/platform/mkldnn/mkldnnUtils.h>
|
|
|
|
#endif
|
|
|
|
class MklDnnTests : public testing::Test {
|
|
public:
|
|
|
|
};
|
|
|
|
static void printer(std::initializer_list<sd::ops::platforms::PlatformHelper*> helpers) {
|
|
|
|
for (auto v:helpers) {
|
|
nd4j_printf("Initialized [%s]\n", v->name().c_str());
|
|
}
|
|
}
|
|
|
|
|
|
TEST_F(MklDnnTests, helpers_includer) {
|
|
// we need this block, to make sure all helpers are still available within binary, and not optimized out by linker
|
|
#ifdef HAVE_MKLDNN
|
|
sd::ops::platforms::PLATFORM_conv2d_ENGINE_CPU conv2d;
|
|
sd::ops::platforms::PLATFORM_conv2d_bp_ENGINE_CPU conv2d_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_conv2d_ENGINE_CPU conv3d;
|
|
sd::ops::platforms::PLATFORM_conv2d_bp_ENGINE_CPU conv3d_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_avgpool2d_ENGINE_CPU avgpool2d;
|
|
sd::ops::platforms::PLATFORM_avgpool2d_bp_ENGINE_CPU avgpool2d_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_maxpool2d_ENGINE_CPU maxpool2d;
|
|
sd::ops::platforms::PLATFORM_maxpool2d_bp_ENGINE_CPU maxpool2d_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_avgpool3dnew_ENGINE_CPU avgpool3d;
|
|
sd::ops::platforms::PLATFORM_avgpool3dnew_bp_ENGINE_CPU avgpool3d_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_maxpool3dnew_ENGINE_CPU maxpool3d;
|
|
sd::ops::platforms::PLATFORM_maxpool3dnew_bp_ENGINE_CPU maxpool3d_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_lrn_ENGINE_CPU lrn;
|
|
|
|
sd::ops::platforms::PLATFORM_batchnorm_ENGINE_CPU batchnorm;
|
|
|
|
sd::ops::platforms::PLATFORM_matmul_ENGINE_CPU matmul;
|
|
|
|
sd::ops::platforms::PLATFORM_softmax_ENGINE_CPU softmax;
|
|
|
|
sd::ops::platforms::PLATFORM_softmax_bp_ENGINE_CPU softmax_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_tanh_ENGINE_CPU tanh;
|
|
|
|
sd::ops::platforms::PLATFORM_tanh_ENGINE_CPU tanh_bp;
|
|
|
|
sd::ops::platforms::PLATFORM_xw_plus_b_ENGINE_CPU xw_plus_b;
|
|
|
|
sd::ops::platforms::PLATFORM_xw_plus_b_bp_ENGINE_CPU xw_plus_b_bp;
|
|
|
|
|
|
printer({&conv2d, &conv2d_bp, &conv3d, &conv3d_bp, &avgpool2d, &avgpool2d_bp, &maxpool2d, &maxpool2d_bp, &avgpool3d, &avgpool3d_bp, &maxpool3d, &maxpool3d_bp, &lrn, &batchnorm, &matmul, &softmax, &softmax_bp, &tanh, &tanh_bp, &xw_plus_b, &xw_plus_b_bp });
|
|
|
|
#endif
|
|
}
|