diff --git a/Cargo.toml b/Cargo.toml index b107e9b564eb4ab6598db96c624973b572bade79..f31310690932767d27cc7daacc9cdd2059632870 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -781,6 +781,7 @@ codegen-units = 16 [profile.dev.package] audio = { opt-level = 3 } +denoise = { opt-level = 3 } taffy = { opt-level = 3 } cranelift-codegen = { opt-level = 3 } cranelift-codegen-meta = { opt-level = 3 } diff --git a/crates/denoise/src/engine.rs b/crates/denoise/src/engine.rs index 5196b70b5ba02f665385c022a0dfa9cd22c1db9c..be0548c689e3b902342cd1cb6d6d8e29351e8be4 100644 --- a/crates/denoise/src/engine.rs +++ b/crates/denoise/src/engine.rs @@ -138,13 +138,13 @@ impl Engine { const SPECTRUM_INPUT: &str = "input_2"; const MEMORY_INPUT: &str = "input_3"; - let memory_input = + let spectrum = Tensor::from_slice::<_, f32>(&self.in_magnitude, (1, 1, FFT_OUT_SIZE), &Device::Cpu) .expect("the in magnitude has enough elements to fill the Tensor"); let inputs = HashMap::from([ - (MEMORY_INPUT.to_string(), memory_input), - (SPECTRUM_INPUT.to_string(), self.spectral_memory.clone()), + (SPECTRUM_INPUT.to_string(), spectrum), + (MEMORY_INPUT.to_string(), self.spectral_memory.clone()), ]); inputs } diff --git a/crates/denoise/src/lib.rs b/crates/denoise/src/lib.rs index 755f2a197be9cc9120eb39ffe0ecd5a68a24ba08..ae09acafdac42b2ba71b76af49af1f9419f20142 100644 --- a/crates/denoise/src/lib.rs +++ b/crates/denoise/src/lib.rs @@ -264,6 +264,7 @@ impl Denoiser { } } +#[inline] fn read_sub_block(s: &mut impl Source) -> Option<[f32; BLOCK_SHIFT]> { let mut res = [0f32; BLOCK_SHIFT]; for sample in &mut res {