Skip to content

Commit

Permalink
update submod
Browse files Browse the repository at this point in the history
  • Loading branch information
shiinamiyuki committed Nov 6, 2023
1 parent 05b6875 commit 8d191d3
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 2 deletions.
7 changes: 6 additions & 1 deletion luisa_compute/src/runtime.rs
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ impl Drop for DeviceHandle {
pub mod extension {
use super::*;
use api::denoiser_ext::{Feature, Image};
pub use api::denoiser_ext::{ImageColorSpace, ImageFormat, PrefilterMode};
pub use api::denoiser_ext::{ImageColorSpace, ImageFormat, PrefilterMode, FilterQuality};
pub struct DenoiserInput {
inner: api::denoiser_ext::DenoiserInput,
inputs: Vec<Image>,
Expand All @@ -121,6 +121,7 @@ pub mod extension {
features: std::ptr::null_mut(),
features_count: 0,
prefilter_mode: PrefilterMode::None,
filter_quality: FilterQuality::Default,
noisy_features: false,
width,
height,
Expand Down Expand Up @@ -198,6 +199,10 @@ pub mod extension {
self.inner.prefilter_mode = mode;
self
}
pub fn filter_quality(&mut self, quality: FilterQuality) -> &mut Self {
self.inner.filter_quality = quality;
self
}
/// set to false **only** if feature images are **noise-free**.
pub fn noisy_features(&mut self, noisy_features: bool) -> &mut Self {
self.inner.noisy_features = noisy_features;
Expand Down

0 comments on commit 8d191d3

Please sign in to comment.