You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.2 KiB
87 lines
2.2 KiB
// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
|
|
//
|
|
// 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.
|
|
#include <thrust/device_vector.h>
|
|
#include <sstream>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "paddle/fluid/framework/dim.h"
|
|
|
|
__global__ void test(paddle::framework::Dim<2>* o) {
|
|
o[0] = paddle::framework::make_dim(5, 6);
|
|
}
|
|
|
|
__global__ void dyn_idx_gpu(int64_t* o) {
|
|
auto d = paddle::framework::make_dim(5, 6);
|
|
o[0] = d[1];
|
|
}
|
|
|
|
TEST(Dim, Equality) {
|
|
// construct a Dim on the CPU
|
|
auto a = paddle::framework::make_dim(3, 4);
|
|
EXPECT_EQ(a[0], 3);
|
|
EXPECT_EQ(a[1], 4);
|
|
|
|
// construct a Dim on the GPU
|
|
thrust::device_vector<paddle::framework::Dim<2>> t(2);
|
|
test<<<1, 1>>>(thrust::raw_pointer_cast(t.data()));
|
|
a = t[0];
|
|
EXPECT_EQ(a[0], 5);
|
|
EXPECT_EQ(a[1], 6);
|
|
|
|
// product
|
|
EXPECT_EQ(paddle::framework::product(a), 30);
|
|
|
|
// mutate a Dim
|
|
auto b = paddle::framework::make_dim(7, 8);
|
|
b[1] = 10;
|
|
EXPECT_EQ(b[0], 7);
|
|
EXPECT_EQ(b[1], 10);
|
|
|
|
b[0] = 8;
|
|
b[1] = 11;
|
|
EXPECT_EQ(b[0], 8);
|
|
EXPECT_EQ(b[1], 11);
|
|
|
|
// dynamic access on GPU
|
|
thrust::device_vector<int64_t> r(1);
|
|
dyn_idx_gpu<<<1, 1>>>(thrust::raw_pointer_cast(r.data()));
|
|
int64_t res = r[0];
|
|
EXPECT_EQ(res, 6);
|
|
}
|
|
|
|
TEST(Dim, Bool) {
|
|
auto a = paddle::framework::make_dim(3, 4);
|
|
auto b = paddle::framework::make_dim(5, 6);
|
|
auto c = paddle::framework::make_dim(3, 4);
|
|
|
|
// comparison
|
|
EXPECT_TRUE(a == a);
|
|
EXPECT_FALSE(a == b);
|
|
EXPECT_TRUE(a == c);
|
|
}
|
|
|
|
TEST(Dim, Print) {
|
|
{
|
|
std::stringstream ss;
|
|
auto a = paddle::framework::make_dim(2, 3);
|
|
ss << a;
|
|
EXPECT_EQ(ss.str(), "2, 3");
|
|
}
|
|
{
|
|
std::stringstream ss;
|
|
ss << paddle::framework::make_dim(8);
|
|
EXPECT_EQ(ss.str(), "8");
|
|
}
|
|
}
|