-
Notifications
You must be signed in to change notification settings - Fork 76
Expand file tree
/
Copy pathmain.cpp
More file actions
73 lines (60 loc) · 2.23 KB
/
main.cpp
File metadata and controls
73 lines (60 loc) · 2.23 KB
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#include <gtest/gtest.h>
#include <cstddef>
#include <cstdint>
#include <fstream>
#include <memory>
#include <string>
#include <vector>
#include "core/task/include/task.hpp"
#include "core/util/include/util.hpp"
#include "seq/example/include/ops_seq.hpp"
TEST(nesterov_a_test_task_seq, test_matmul_50) {
constexpr size_t kCount = 50;
// Create data
std::vector<int> in(kCount * kCount, 0);
std::vector<int> out(kCount * kCount, 0);
for (size_t i = 0; i < kCount; i++) {
in[(i * kCount) + i] = 1;
}
// Create task_data
auto task_data_seq = std::make_shared<ppc::core::TaskData>();
task_data_seq->inputs.emplace_back(reinterpret_cast<uint8_t *>(in.data()));
task_data_seq->inputs_count.emplace_back(in.size());
task_data_seq->outputs.emplace_back(reinterpret_cast<uint8_t *>(out.data()));
task_data_seq->outputs_count.emplace_back(out.size());
// Create Task
nesterov_a_test_task_seq::TestTaskSequential test_task_sequential(task_data_seq);
ASSERT_EQ(test_task_sequential.Validation(), true);
test_task_sequential.PreProcessing();
test_task_sequential.Run();
test_task_sequential.PostProcessing();
EXPECT_EQ(in, out);
}
TEST(nesterov_a_test_task_seq, test_matmul_100_from_file) {
std::string line;
std::ifstream test_file(ppc::util::GetAbsolutePath("seq/example/data/test.txt"));
if (test_file.is_open()) {
getline(test_file, line);
}
test_file.close();
const size_t count = std::stoi(line);
// Create data
std::vector<int> in(count * count, 0);
std::vector<int> out(count * count, 0);
for (size_t i = 0; i < count; i++) {
in[(i * count) + i] = 1;
}
// Create task_data
auto task_data_seq = std::make_shared<ppc::core::TaskData>();
task_data_seq->inputs.emplace_back(reinterpret_cast<uint8_t *>(in.data()));
task_data_seq->inputs_count.emplace_back(in.size());
task_data_seq->outputs.emplace_back(reinterpret_cast<uint8_t *>(out.data()));
task_data_seq->outputs_count.emplace_back(out.size());
// Create Task
nesterov_a_test_task_seq::TestTaskSequential test_task_sequential(task_data_seq);
ASSERT_EQ(test_task_sequential.Validation(), true);
test_task_sequential.PreProcessing();
test_task_sequential.Run();
test_task_sequential.PostProcessing();
EXPECT_EQ(in, out);
}