xref: /aosp_15_r20/external/pytorch/aten/src/ATen/test/dlconvertor_test.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <gtest/gtest.h>
2 
3 #include <ATen/ATen.h>
4 #include <ATen/DLConvertor.h>
5 
6 #include <iostream>
7 // NOLINTNEXTLINE(modernize-deprecated-headers)
8 #include <string.h>
9 #include <sstream>
10 
11 using namespace at;
TEST(TestDlconvertor,TestDlconvertor)12 TEST(TestDlconvertor, TestDlconvertor) {
13   manual_seed(123);
14 
15   Tensor a = rand({3, 4});
16   DLManagedTensor* dlMTensor = toDLPack(a);
17 
18   Tensor b = fromDLPack(dlMTensor);
19 
20   ASSERT_TRUE(a.equal(b));
21 }
22 
TEST(TestDlconvertor,TestDlconvertorNoStrides)23 TEST(TestDlconvertor, TestDlconvertorNoStrides) {
24   manual_seed(123);
25 
26   Tensor a = rand({3, 4});
27   DLManagedTensor* dlMTensor = toDLPack(a);
28   dlMTensor->dl_tensor.strides = nullptr;
29 
30   Tensor b = fromDLPack(dlMTensor);
31 
32   ASSERT_TRUE(a.equal(b));
33 }
34