mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-26 19:52:38 +00:00
5781222456
* Convert ABI to a versioned interface. * Convert ORT_THROW_ON_ERROR to inline function to fix link errors.
36 lines
1.3 KiB
C++
36 lines
1.3 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/session/onnxruntime_cxx_api.h"
|
|
#include "core/framework/allocator.h"
|
|
#include "core/providers/cpu/cpu_provider_factory.h"
|
|
#include "test_fixture.h"
|
|
|
|
const OrtApi* g_ort = OrtGetApi(ORT_API_VERSION);
|
|
const OrtApi* Ort::g_api = OrtGetApi(ORT_API_VERSION);
|
|
|
|
using namespace onnxruntime;
|
|
|
|
TEST_F(CApiTest, allocation_info) {
|
|
OrtMemoryInfo *info1, *info2;
|
|
ORT_THROW_ON_ERROR(g_ort->CreateCpuMemoryInfo(OrtArenaAllocator, OrtMemTypeDefault, &info1));
|
|
ORT_THROW_ON_ERROR(g_ort->CreateCpuMemoryInfo(OrtArenaAllocator, OrtMemTypeDefault, &info2));
|
|
int result;
|
|
ORT_THROW_ON_ERROR(g_ort->CompareMemoryInfo(info1, info2, &result));
|
|
ASSERT_EQ(0, result);
|
|
g_ort->ReleaseMemoryInfo(info1);
|
|
g_ort->ReleaseMemoryInfo(info2);
|
|
}
|
|
|
|
TEST_F(CApiTest, DefaultAllocator) {
|
|
Ort::AllocatorWithDefaultOptions default_allocator;
|
|
char* p = (char*)default_allocator.Alloc(100);
|
|
ASSERT_NE(p, nullptr);
|
|
memset(p, 0, 100);
|
|
default_allocator.Free(p);
|
|
const OrtMemoryInfo* info1 = default_allocator.GetInfo();
|
|
const OrtMemoryInfo* info2 = static_cast<OrtAllocator*>(default_allocator)->Info(default_allocator);
|
|
int result;
|
|
ORT_THROW_ON_ERROR(g_ort->CompareMemoryInfo(info1, info2, &result));
|
|
ASSERT_EQ(0, result);
|
|
}
|