mirror of
https://github.com/yalue/onnxruntime_go.git
synced 2025-12-24 13:38:00 +08:00
75e3434f7e
- Sets up the onnxruntime environment, but doesn't load or run networks yet. - The things builds and run on Windows. - Still working on getting the Linux (arm64 for now) test to work.
28 lines
1.3 KiB
C
28 lines
1.3 KiB
C
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
/*
|
|
* This file defines RunOptions Config Keys and format of the Config Values.
|
|
*
|
|
* The Naming Convention for a RunOptions Config Key,
|
|
* "[Area][.[SubArea1].[SubArea2]...].[Keyname]"
|
|
* Such as "ep.cuda.use_arena"
|
|
* The Config Key cannot be empty
|
|
* The maximum length of the Config Key is 128
|
|
*
|
|
* The string format of a RunOptions Config Value is defined individually for each Config.
|
|
* The maximum length of the Config Value is 1024
|
|
*/
|
|
|
|
// Key for enabling shrinkages of user listed device memory arenas.
|
|
// Expects a list of semi-colon separated key value pairs separated by colon in the following format:
|
|
// "device_0:device_id_0;device_1:device_id_1"
|
|
// No white-spaces allowed in the provided list string.
|
|
// Currently, the only supported devices are : "cpu", "gpu" (case sensitive).
|
|
// If "cpu" is included in the list, DisableCpuMemArena() API must not be called (i.e.) arena for cpu should be enabled.
|
|
// Example usage: "cpu:0;gpu:0" (or) "gpu:0"
|
|
// By default, the value for this key is empty (i.e.) no memory arenas are shrunk
|
|
static const char* const kOrtRunOptionsConfigEnableMemoryArenaShrinkage = "memory.enable_memory_arena_shrinkage";
|