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

test: create session so env is initialized

Browse Source
This commit is contained in:
Carson M.
2025-08-31 19:45:05 -05:00
parent f59858000a
commit c162a72a38
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/thread_manager.rs
View File

@@ -6,7 +6,10 @@ use std::{
thread::{self, JoinHandle}
};
use ort::environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker};
use ort::{
environment::{GlobalThreadPoolOptions, ThreadManager, ThreadWorker},
session::Session
};
struct ThreadStats {
active_threads: AtomicUsize
@@ -64,6 +67,8 @@ fn global_thread_manager() -> ort::Result<()> {
)
.commit();
let _session = Session::builder()?.commit_from_file("tests/data/upsample.ort")?;
assert_eq!(stats.active_threads.load(Ordering::Acquire), 4);
Ok(())