|
|
@ -15,6 +15,7 @@ limitations under the License. */
|
|
|
|
#include <gflags/gflags.h>
|
|
|
|
#include <gflags/gflags.h>
|
|
|
|
#include <glog/logging.h>
|
|
|
|
#include <glog/logging.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
|
|
|
|
|
|
|
|
#include <cmath>
|
|
|
|
#include <cmath>
|
|
|
|
|
|
|
|
|
|
|
|
#include "paddle/fluid/inference/tests/api/tester_helper.h"
|
|
|
|
#include "paddle/fluid/inference/tests/api/tester_helper.h"
|
|
|
@ -49,17 +50,15 @@ TEST(AnalysisPredictor, use_gpu) {
|
|
|
|
ASSERT_TRUE(predictor->Run(inputs, &outputs));
|
|
|
|
ASSERT_TRUE(predictor->Run(inputs, &outputs));
|
|
|
|
|
|
|
|
|
|
|
|
const std::vector<float> truth_values = {
|
|
|
|
const std::vector<float> truth_values = {
|
|
|
|
127.780396f, 738.16656f, 1013.2264f, -438.17206f, 366.4022f,
|
|
|
|
127.779f, 738.165f, 1013.22f, -438.17f, 366.401f, 927.659f,
|
|
|
|
927.66187f, 736.2241f, -633.68567f, -329.92737f, -430.15637f,
|
|
|
|
736.222f, -633.684f, -329.927f, -430.155f, -633.062f, -146.548f,
|
|
|
|
-633.0639f, -146.54858f, -1324.2804f, -1349.3661f, -242.67671f,
|
|
|
|
-1324.28f, -1349.36f, -242.675f, 117.448f, -801.723f, -391.514f,
|
|
|
|
117.44864f, -801.7251f, -391.51495f, -404.8202f, 454.16132f,
|
|
|
|
-404.818f, 454.16f, 515.48f, -133.031f, 69.293f, 590.096f,
|
|
|
|
515.48206f, -133.03114f, 69.293076f, 590.09753f, -1434.6917f,
|
|
|
|
-1434.69f, -1070.89f, 307.074f, 400.525f, -316.12f, -587.125f,
|
|
|
|
-1070.8903f, 307.0744f, 400.52573f, -316.12177f, -587.1265f,
|
|
|
|
-161.056f, 800.363f, -96.4708f, 748.706f, 868.174f, -447.938f,
|
|
|
|
-161.05742f, 800.3663f, -96.47157f, 748.708f, 868.17645f,
|
|
|
|
112.737f, 1127.2f, 47.4355f, 677.72f, 593.186f, -336.4f,
|
|
|
|
-447.9403f, 112.73656f, 1127.1992f, 47.43518f, 677.7219f,
|
|
|
|
551.362f, 397.823f, 78.3979f, -715.398f, 405.969f, 404.256f,
|
|
|
|
593.1881f, -336.4011f, 551.3634f, 397.82474f, 78.39835f,
|
|
|
|
246.019f, -8.42969f, 131.365f, -648.051f};
|
|
|
|
-715.4006f, 405.96988f, 404.25684f, 246.01978f, -8.430191f,
|
|
|
|
|
|
|
|
131.36617f, -648.0528f};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
const size_t expected_size = 1;
|
|
|
|
const size_t expected_size = 1;
|
|
|
|
EXPECT_EQ(outputs.size(), expected_size);
|
|
|
|
EXPECT_EQ(outputs.size(), expected_size);
|
|
|
@ -108,17 +107,15 @@ TEST(Predictor, use_gpu) {
|
|
|
|
output_t->CopyToCpu(out_data.data());
|
|
|
|
output_t->CopyToCpu(out_data.data());
|
|
|
|
|
|
|
|
|
|
|
|
const std::vector<float> truth_values = {
|
|
|
|
const std::vector<float> truth_values = {
|
|
|
|
127.780396f, 738.16656f, 1013.2264f, -438.17206f, 366.4022f,
|
|
|
|
127.779f, 738.165f, 1013.22f, -438.17f, 366.401f, 927.659f,
|
|
|
|
927.66187f, 736.2241f, -633.68567f, -329.92737f, -430.15637f,
|
|
|
|
736.222f, -633.684f, -329.927f, -430.155f, -633.062f, -146.548f,
|
|
|
|
-633.0639f, -146.54858f, -1324.2804f, -1349.3661f, -242.67671f,
|
|
|
|
-1324.28f, -1349.36f, -242.675f, 117.448f, -801.723f, -391.514f,
|
|
|
|
117.44864f, -801.7251f, -391.51495f, -404.8202f, 454.16132f,
|
|
|
|
-404.818f, 454.16f, 515.48f, -133.031f, 69.293f, 590.096f,
|
|
|
|
515.48206f, -133.03114f, 69.293076f, 590.09753f, -1434.6917f,
|
|
|
|
-1434.69f, -1070.89f, 307.074f, 400.525f, -316.12f, -587.125f,
|
|
|
|
-1070.8903f, 307.0744f, 400.52573f, -316.12177f, -587.1265f,
|
|
|
|
-161.056f, 800.363f, -96.4708f, 748.706f, 868.174f, -447.938f,
|
|
|
|
-161.05742f, 800.3663f, -96.47157f, 748.708f, 868.17645f,
|
|
|
|
112.737f, 1127.2f, 47.4355f, 677.72f, 593.186f, -336.4f,
|
|
|
|
-447.9403f, 112.73656f, 1127.1992f, 47.43518f, 677.7219f,
|
|
|
|
551.362f, 397.823f, 78.3979f, -715.398f, 405.969f, 404.256f,
|
|
|
|
593.1881f, -336.4011f, 551.3634f, 397.82474f, 78.39835f,
|
|
|
|
246.019f, -8.42969f, 131.365f, -648.051f};
|
|
|
|
-715.4006f, 405.96988f, 404.25684f, 246.01978f, -8.430191f,
|
|
|
|
|
|
|
|
131.36617f, -648.0528f};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
float* data_o = out_data.data();
|
|
|
|
float* data_o = out_data.data();
|
|
|
|
for (size_t j = 0; j < out_num; j += 10) {
|
|
|
|
for (size_t j = 0; j < out_num; j += 10) {
|
|
|
|