chore: import upstream snapshot with attribution

test/cpp/phi/core/test_dim.cu

@@ -0,0 +1,102 @@
+// Copyright (c) 2022 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/common/dim.h"
+
+namespace phi {
+namespace tests {
+
+__global__ void test(phi::Dim<2>* o) { o[0] = common::make_dim(5, 6); }
+
+__global__ void dyn_idx_gpu(int64_t* o) {
+  auto d = common::make_dim(5, 6);
+  o[0] = d[1];
+}
+
+TEST(Dim, Equality) {
+  // construct a Dim on the CPU
+  auto a = common::make_dim(3, 4);
+  EXPECT_EQ(a[0], 3);
+  EXPECT_EQ(a[1], 4);
+
+  // construct a Dim on the GPU
+  thrust::device_vector<phi::Dim<2>> t(2);
+#ifdef PADDLE_WITH_HIP
+  hipLaunchKernelGGL(
+      test, dim3(1), dim3(1), 0, 0, thrust::raw_pointer_cast(t.data()));
+#else
+  test<<<1, 1>>>(thrust::raw_pointer_cast(t.data()));
+#endif
+  a = t[0];
+  EXPECT_EQ(a[0], 5);
+  EXPECT_EQ(a[1], 6);
+
+  // product
+  EXPECT_EQ(common::product(a), 30);
+
+  // mutate a Dim
+  auto b = common::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);
+#ifdef PADDLE_WITH_HIP
+  hipLaunchKernelGGL(
+      dyn_idx_gpu, dim3(1), dim3(1), 0, 0, thrust::raw_pointer_cast(r.data()));
+#else
+  dyn_idx_gpu<<<1, 1>>>(thrust::raw_pointer_cast(r.data()));
+#endif
+  int64_t res = r[0];
+  EXPECT_EQ(res, 6);
+}
+
+TEST(Dim, Bool) {
+  auto a = common::make_dim(3, 4);
+  auto b = common::make_dim(5, 6);
+  auto c = common::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 = common::make_dim(2, 3);
+    ss << a;
+    EXPECT_EQ(ss.str(), "2, 3");
+  }
+  {
+    std::stringstream ss;
+    ss << common::make_dim(8);
+    EXPECT_EQ(ss.str(), "8");
+  }
+}
+
+}  // namespace tests
+}  // namespace phi