saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-21 19:18:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add enable/disable mem pattern api for python and csharp. (#1227)

Browse source
This commit is contained in:
Zhang Lei 2019-06-19 11:17:21 -07:00 committed by Changming Sun
parent e26e11b9f7
commit 23838d9c2a
2 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
csharp/src/Microsoft.ML.OnnxRuntime/SessionOptions.cs
View file

@ -55,6 +55,24 @@ namespace Microsoft.ML.OnnxRuntime
NativeApiStatus.VerifySuccess(NativeMethods.OrtDisableSequentialExecution(_nativePtr));
}
/// <summary>
/// Enable Mem Pattern. By default, it is enabled
/// </summary>
/// </param>
public void EnableMemPattern()
{
NativeApiStatus.VerifySuccess(NativeMethods.OrtEnableMemPattern(_nativePtr));
}
/// <summary>
/// Disable Mem Pattern.
/// </summary>
/// </param>
public void DisableMemPattern()
{
NativeApiStatus.VerifySuccess(NativeMethods.OrtDisableMemPattern(_nativePtr));
}
/// <summary>
/// Default instance
/// </summary>

2
onnxruntime/python/onnxruntime_pybind_state.cc
View file

@ -389,6 +389,8 @@ void addObjectMethods(py::module& m) {
Set this option to false if you don't want it. Default is True.)pbdoc")
.def_readwrite("enable_profiling", &SessionOptions::enable_profiling,
R"pbdoc(Enable profiling for this session. Default is false.)pbdoc")
.def_readwrite("enable_mem_pattern", &SessionOptions::enable_mem_pattern,
R"pbdoc(Enable the memory pattern optimization. Default is true.)pbdoc")
.def_readwrite("enable_sequential_execution", &SessionOptions::enable_sequential_execution,
R"pbdoc(Enables sequential execution, disables parallel execution. Default is true.)pbdoc")
.def_readwrite("max_num_graph_transformation_steps", &SessionOptions::max_num_graph_transformation_steps,