Function TEST

tests/cpp/extra/test_c_env_api.cc:62–68

}

TEST(CEnvAPI, TVMFFIEnvSetDLPackManagedTensorAllocator) {
  auto old_allocator = TVMFFIEnvGetDLPackManagedTensorAllocator();
  DLPackManagedTensorAllocator pre_allocator;
  TVMFFIEnvSetDLPackManagedTensorAllocator(TestDLPackManagedTensorAllocator, 0, &pre_allocator);
  EXPECT_EQ(old_allocator, pre_allocator);
  TVMFFIEnvSetDLPackManagedTensorAllocator(old_allocator, 0, nullptr);
}

TEST(CEnvAPI, TVMFFIEnvTensorAlloc) {
  auto old_allocator = TVMFFIEnvGetDLPackManagedTensorAllocator();