Skip to content

Commit

Permalink
more cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Jan 23, 2024
1 parent 5263373 commit 91522e5
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 2 deletions.
1 change: 0 additions & 1 deletion Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,6 @@ flate2 = "1.0.20"
fs2 = "0.4.3"
getrandom = "0.2"
half = { version="2.2.1", features = [ "std", "num-traits" ] }
hexf = "0.2.1"
image = "0.24.1"
itertools = "0.10.1"
home = "0.5.5"
Expand Down
3 changes: 2 additions & 1 deletion core/src/transform.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,8 @@ use crate::ops::nn::{Softmax, SoftmaxExp, TypedModel};

pub fn get_transformer(name: &str) -> Option<Box<dyn ModelTransformer>> {
match name {
"f32-to-f16" => Some(Box::new(FloatPrecisionTranslator::<f32, f16>::default())),
"f32-to-f16" => Some(Box::<FloatPrecisionTranslator<f32, f16>>::default()),
"f16-to-f32" => Some(Box::<FloatPrecisionTranslator<f32, f16>>::default()),
"softmax-fast-compact" => Some(Box::new(SoftmaxFastCompact)),
_ => None,
}
Expand Down
2 changes: 2 additions & 0 deletions linalg/src/lib.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
#![allow(clippy::missing_safety_doc)]
#![allow(clippy::redundant_closure_call)]
#![allow(clippy::len_zero)]
#![allow(clippy::excessive_precision)]
#![allow(clippy::approx_constant)]
#[macro_use]
extern crate derive_new;
extern crate lazy_static;
Expand Down

0 comments on commit 91522e5

Please sign in to comment.