eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: separate graph optimization level 3

Browse Source
This commit is contained in:
Carson M.
2025-11-26 14:14:52 -06:00
parent b64e9f8946
commit 1e628680f1
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
ort-sys/src/lib.rs
View File

@@ -345,6 +345,7 @@ pub enum GraphOptimizationLevel {
ORT_DISABLE_ALL = 0,
ORT_ENABLE_BASIC = 1,
ORT_ENABLE_EXTENDED = 2,
ORT_ENABLE_LAYOUT = 3,
ORT_ENABLE_ALL = 99
}
#[repr(i32)]

7
src/session/builder/impl_options.rs
View File

@@ -370,7 +370,9 @@ pub enum GraphOptimizationLevel {
Level2,
/// Level 3 optimizations include memory layout optimizations, which may optimize the graph to use the NCHWc memory
/// layout rather than NCHW to improve spatial locality for some targets.
Level3
Level3,
/// Enable all optimizations.
All
}
impl From<GraphOptimizationLevel> for ort_sys::GraphOptimizationLevel {
@@ -379,7 +381,8 @@ impl From<GraphOptimizationLevel> for ort_sys::GraphOptimizationLevel {
GraphOptimizationLevel::Disable => ort_sys::GraphOptimizationLevel::ORT_DISABLE_ALL,
GraphOptimizationLevel::Level1 => ort_sys::GraphOptimizationLevel::ORT_ENABLE_BASIC,
GraphOptimizationLevel::Level2 => ort_sys::GraphOptimizationLevel::ORT_ENABLE_EXTENDED,
GraphOptimizationLevel::Level3 => ort_sys::GraphOptimizationLevel::ORT_ENABLE_ALL
GraphOptimizationLevel::Level3 => ort_sys::GraphOptimizationLevel::ORT_ENABLE_LAYOUT,
GraphOptimizationLevel::All => ort_sys::GraphOptimizationLevel::ORT_ENABLE_ALL
}
}
}