From c7d738acb70c80f37804608ec2bb45fcf46748de Mon Sep 17 00:00:00 2001 From: Håvard Pettersen Date: Wed, 26 May 2021 11:52:40 +0000 Subject: disable use of arena allocator --- eval/src/tests/tensor/onnx_wrapper/onnx_wrapper_test.cpp | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'eval/src/tests/tensor') diff --git a/eval/src/tests/tensor/onnx_wrapper/onnx_wrapper_test.cpp b/eval/src/tests/tensor/onnx_wrapper/onnx_wrapper_test.cpp index 9b44dd7519e..54f958f8111 100644 --- a/eval/src/tests/tensor/onnx_wrapper/onnx_wrapper_test.cpp +++ b/eval/src/tests/tensor/onnx_wrapper/onnx_wrapper_test.cpp @@ -436,4 +436,11 @@ TEST(OnnxTest, inspect_float_to_int8_conversion) { //------------------------------------------------------------------------- } +TEST(OnnxTest, default_allocator_type) { + Ort::AllocatorWithDefaultOptions default_alloc; + OrtAllocatorType res = Invalid; + Ort::ThrowOnError(Ort::GetApi().MemoryInfoGetType(default_alloc.GetInfo(), &res)); + fprintf(stderr, "default allocator type: %d\n", int(res)); +} + GTEST_MAIN_RUN_ALL_TESTS() -- cgit v1.2.3