Skip to content

Commit

Permalink
switch to anymap3
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Jan 17, 2025
1 parent 73c4235 commit 4ea708d
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ anstyle = "1.0.2"
anstyle-parse = "0.2.1"
anstyle-query = "1.0.0"
anyhow = "1.0.43"
anymap = "0.12.1"
anymap3 = "1.0"
approx = "0.5"
bit-set= "0.5.2"
blis-src = { version = "0.2", features = ["static", "pthreads"] }
Expand Down
2 changes: 1 addition & 1 deletion core/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ maintenance = { status = "actively-developed" }
[dependencies]
accelerate-src = { workspace = true, optional = true }
anyhow.workspace = true
anymap.workspace = true
anymap3.workspace = true
bit-set.workspace = true
blis-src = { version = "0.2", features = ["static", "pthreads"], optional = true }
cblas = { version = "0.4", optional = true }
Expand Down
8 changes: 4 additions & 4 deletions core/src/plan.rs
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ pub struct SessionState {
pub scenario: Option<usize>,
pub tensors: HashMap<String, Tensor>,
pub cached_mmm_scratch_space: RefCell<Option<Box<dyn tract_linalg::mmm::ScratchSpace>>>,
pub scratch_extensions: anymap::Map,
pub scratch_extensions: anymap3::Map,
}

impl Default for SessionState {
Expand All @@ -39,7 +39,7 @@ impl Default for SessionState {
tensors: HashMap::default(),
scenario: None,
cached_mmm_scratch_space: None.into(),
scratch_extensions: anymap::Map::new(),
scratch_extensions: anymap3::Map::new(),
}
}
}
Expand All @@ -52,7 +52,7 @@ impl Clone for SessionState {
tensors: self.tensors.clone(),
scenario: self.scenario,
cached_mmm_scratch_space: None.into(),
scratch_extensions: anymap::Map::new(),
scratch_extensions: anymap3::Map::new(),
}
}
}
Expand Down Expand Up @@ -681,7 +681,7 @@ where
scenario: self.scenario,
tensors: self.tensors.clone(),
cached_mmm_scratch_space: None.into(),
scratch_extensions: anymap::Map::new(),
scratch_extensions: anymap3::Map::new(),
},
states: self.states.iter().map(|s| s.as_ref().map(|s| s.unfreeze())).collect(),
values: self
Expand Down

0 comments on commit 4ea708d

Please sign in to comment.