TestUtil.h 2.8 KB
Newer Older
1
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
Z
zhangjinchao01 已提交
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

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://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. */

#pragma once

#include <gtest/gtest.h>
#include "paddle/math/Matrix.h"

namespace paddle {

std::string randStr(const int len);

inline int uniformRandom(int n) { return n == 0 ? 0 : rand() % n; }

inline bool approximatelyEqual(float a, float b, float epsilon) {
  return fabs(a - b) <= ((fabs(a) < fabs(b) ? fabs(b) : fabs(a)) * epsilon);
}

30 31 32 33 34
MatrixPtr makeRandomSparseMatrix(size_t height,
                                 size_t width,
                                 bool withValue,
                                 bool useGpu,
                                 bool equalNnzPerSample = false);
Z
zhangjinchao01 已提交
35 36 37 38 39 40 41 42 43

/**
 * @brief generate sequenceStartPositions for INPUT_SEQUENCE_DATA,
 *        INPUT_HASSUB_SEQUENCE_DATA and INPUT_SEQUENCE_LABEL
 *
 * @param batchSize                      batchSize
 *        sequenceStartPositions[out] generation output
 */
void generateSequenceStartPositions(size_t batchSize,
44
                                    IVectorPtr& sequenceStartPositions);
Z
zhangjinchao01 已提交
45 46

void generateSequenceStartPositions(size_t batchSize,
47
                                    ICpuGpuVectorPtr& sequenceStartPositions);
Z
zhangjinchao01 已提交
48 49 50 51 52 53 54 55

/**
 * @brief generate subSequenceStartPositions for INPUT_HASSUB_SEQUENCE_DATA
 *        according to sequenceStartPositions
 *
 * @param sequenceStartPositions[in]     input
 *        subSequenceStartPositions[out] generation output
 */
56 57
void generateSubSequenceStartPositions(const IVectorPtr& sequenceStartPositions,
                                       IVectorPtr& subSequenceStartPositions);
Z
zhangjinchao01 已提交
58 59 60 61 62 63 64 65 66 67 68 69

void generateSubSequenceStartPositions(
    const ICpuGpuVectorPtr& sequenceStartPositions,
    ICpuGpuVectorPtr& subSequenceStartPositions);

/**
 * @brief generate cpuSequenceDims for INPUT_SEQUENCE_MDIM_DATA according to
 *        sequenceStartPositions
 *
 * @param sequenceStartPositions[in]     input
 *        cpuSequenceDims[out]              generation output
 */
70 71 72 73
void generateMDimSequenceData(const IVectorPtr& sequenceStartPositions,
                              IVectorPtr& cpuSequenceDims);
void generateMDimSequenceData(const ICpuGpuVectorPtr& sequenceStartPositions,
                              IVectorPtr& cpuSequenceDims);
Z
zhangjinchao01 已提交
74 75 76 77 78

void checkMatrixEqual(const MatrixPtr& a, const MatrixPtr& b);

void checkVectorEqual(const IVectorPtr& a, const IVectorPtr& b);
}  // namespace paddle