Skip to content

Commit aeec82a

Browse files
committed
Add unittest for reshape op
test=develop
1 parent 24c9fbd commit aeec82a

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

paddle/fluid/framework/tensor_util_test.cc

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,11 @@ TEST(TensorCopy, Tensor) {
4141
EXPECT_EQ(src_ptr[i], dst_ptr[i]);
4242
}
4343

44+
TensorCopy(dst_tensor, *cpu_place, &dst_tensor);
45+
for (size_t i = 0; i < 9; ++i) {
46+
EXPECT_EQ(src_ptr[i], dst_ptr[i]);
47+
}
48+
4449
EXPECT_TRUE(dst_tensor.layout() == src_tensor.layout());
4550

4651
Tensor slice_tensor = src_tensor.Slice(1, 2);
@@ -82,6 +87,15 @@ TEST(TensorCopy, Tensor) {
8287
EXPECT_EQ(src_ptr[i], dst_ptr[i]);
8388
}
8489

90+
// Copy the same tensor
91+
TensorCopy(gpu_tensor, *gpu_place, gpu_ctx, &gpu_tensor);
92+
gpu_ctx.Wait();
93+
const int* dst_ptr_tmp = dst_tensor.data<int>();
94+
EXPECT_NE(src_ptr, dst_ptr_tmp);
95+
for (size_t i = 0; i < 9; ++i) {
96+
EXPECT_EQ(src_ptr[i], dst_ptr_tmp[i]);
97+
}
98+
8599
Tensor slice_tensor = src_tensor.Slice(1, 2);
86100

87101
// CPU Slice Tensor to GPU Tensor

0 commit comments

Comments
 (0)