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

refactor: shorten execution_providers

Browse Source 
frankly, working on documentation has made me tired of typing out `execution_providers` and `ExecutionProvider` all the time
This commit is contained in:
Carson M.
2025-11-14 23:57:09 -06:00
parent 47e5667d6e
commit 3b408b1b44
43 changed files with 508 additions and 435 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/leak-check/main.rs
View File

@@ -1,6 +1,6 @@
use ort::{
adapter::Adapter,
execution_providers::CPUExecutionProvider,
ep,
memory::{AllocationDevice, Allocator, AllocatorType, MemoryInfo, MemoryType},
operator::{
Operator, OperatorDomain,
@@ -77,7 +77,7 @@ impl Operator for CustomOpTwo {
}
fn main() -> ort::Result<()> {
ort::init().with_execution_providers([CPUExecutionProvider::default().build()]).commit();
ort::init().with_execution_providers([ep::CPU::default().build()]).commit();
let mut session = Session::builder()?
.with_operators(OperatorDomain::new("test.customop")?.add(CustomOpOne)?.add(CustomOpTwo)?)?