| 861 | }; |
| 862 | |
| 863 | TEST_F(MemoryManagerApi, E2ETest1D) { |
| 864 | size_t aSize = 8; |
| 865 | |
| 866 | array a = af::array(aSize, af::dtype::f32); |
| 867 | ASSERT_EQ(payload->table.size(), 1); |
| 868 | |
| 869 | ASSERT_EQ(payload->table[a.device<float>()], aSize * sizeof(float)); |
| 870 | ASSERT_EQ(payload->lastNdims, 1); |
| 871 | ASSERT_EQ(payload->lastDims, af::dim4(aSize)); |
| 872 | ASSERT_EQ(payload->lastElementSize, 4); |
| 873 | } |
| 874 | |
| 875 | TEST_F(MemoryManagerApi, E2ETest2D) { |
| 876 | size_t aSize = 8; |