Add configuration for async pipeline creation on RenderPlugin (#11847)
# Objective Fixes #11846 ## Solution Add a `synchronous_pipeline_compilation ` field to `RenderPlugin`, defaulting to `false`. Most of the diff is whitespace. ## Changelog Added `synchronous_pipeline_compilation ` to `RenderPlugin` for disabling async pipeline creation. ## Migration Guide TODO: consider combining this with the guide for #11846 `RenderPlugin` has a new `synchronous_pipeline_compilation ` property. The default value is `false`. Set this to `true` if you want to retain the previous synchronous behavior. --------- Co-authored-by: JMS55 <47158642+JMS55@users.noreply.github.com> Co-authored-by: François <mockersf@gmail.com>
This commit is contained in:
parent
9a2ce8e31b
commit
ebaa347afe
@ -85,6 +85,9 @@ use std::{
|
|||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct RenderPlugin {
|
pub struct RenderPlugin {
|
||||||
pub render_creation: RenderCreation,
|
pub render_creation: RenderCreation,
|
||||||
|
/// If `true`, disables asynchronous pipeline compilation.
|
||||||
|
/// This has no effect on macOS, Wasm, or without the `multi-threaded` feature.
|
||||||
|
pub synchronous_pipeline_compilation: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The labels of the default App rendering sets.
|
/// The labels of the default App rendering sets.
|
||||||
@ -355,7 +358,10 @@ impl Plugin for RenderPlugin {
|
|||||||
|
|
||||||
render_app
|
render_app
|
||||||
.insert_resource(instance)
|
.insert_resource(instance)
|
||||||
.insert_resource(PipelineCache::new(device.clone()))
|
.insert_resource(PipelineCache::new(
|
||||||
|
device.clone(),
|
||||||
|
self.synchronous_pipeline_compilation,
|
||||||
|
))
|
||||||
.insert_resource(device)
|
.insert_resource(device)
|
||||||
.insert_resource(queue)
|
.insert_resource(queue)
|
||||||
.insert_resource(render_adapter)
|
.insert_resource(render_adapter)
|
||||||
|
@ -480,6 +480,9 @@ pub struct PipelineCache {
|
|||||||
pipelines: Vec<CachedPipeline>,
|
pipelines: Vec<CachedPipeline>,
|
||||||
waiting_pipelines: HashSet<CachedPipelineId>,
|
waiting_pipelines: HashSet<CachedPipelineId>,
|
||||||
new_pipelines: Mutex<Vec<CachedPipeline>>,
|
new_pipelines: Mutex<Vec<CachedPipeline>>,
|
||||||
|
/// If `true`, disables asynchronous pipeline compilation.
|
||||||
|
/// This has no effect on MacOS, wasm, or without the `multi_threaded` feature.
|
||||||
|
synchronous_pipeline_compilation: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PipelineCache {
|
impl PipelineCache {
|
||||||
@ -488,7 +491,7 @@ impl PipelineCache {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Create a new pipeline cache associated with the given render device.
|
/// Create a new pipeline cache associated with the given render device.
|
||||||
pub fn new(device: RenderDevice) -> Self {
|
pub fn new(device: RenderDevice, synchronous_pipeline_compilation: bool) -> Self {
|
||||||
Self {
|
Self {
|
||||||
shader_cache: Arc::new(Mutex::new(ShaderCache::new(&device))),
|
shader_cache: Arc::new(Mutex::new(ShaderCache::new(&device))),
|
||||||
device,
|
device,
|
||||||
@ -496,6 +499,7 @@ impl PipelineCache {
|
|||||||
waiting_pipelines: default(),
|
waiting_pipelines: default(),
|
||||||
new_pipelines: default(),
|
new_pipelines: default(),
|
||||||
pipelines: default(),
|
pipelines: default(),
|
||||||
|
synchronous_pipeline_compilation,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -679,7 +683,8 @@ impl PipelineCache {
|
|||||||
let device = self.device.clone();
|
let device = self.device.clone();
|
||||||
let shader_cache = self.shader_cache.clone();
|
let shader_cache = self.shader_cache.clone();
|
||||||
let layout_cache = self.layout_cache.clone();
|
let layout_cache = self.layout_cache.clone();
|
||||||
create_pipeline_task(async move {
|
create_pipeline_task(
|
||||||
|
async move {
|
||||||
let mut shader_cache = shader_cache.lock().unwrap();
|
let mut shader_cache = shader_cache.lock().unwrap();
|
||||||
let mut layout_cache = layout_cache.lock().unwrap();
|
let mut layout_cache = layout_cache.lock().unwrap();
|
||||||
|
|
||||||
@ -695,8 +700,12 @@ impl PipelineCache {
|
|||||||
|
|
||||||
let fragment_module = match &descriptor.fragment {
|
let fragment_module = match &descriptor.fragment {
|
||||||
Some(fragment) => {
|
Some(fragment) => {
|
||||||
match shader_cache.get(&device, id, fragment.shader.id(), &fragment.shader_defs)
|
match shader_cache.get(
|
||||||
{
|
&device,
|
||||||
|
id,
|
||||||
|
fragment.shader.id(),
|
||||||
|
&fragment.shader_defs,
|
||||||
|
) {
|
||||||
Ok(module) => Some(module),
|
Ok(module) => Some(module),
|
||||||
Err(err) => return Err(err),
|
Err(err) => return Err(err),
|
||||||
}
|
}
|
||||||
@ -760,7 +769,9 @@ impl PipelineCache {
|
|||||||
Ok(Pipeline::RenderPipeline(
|
Ok(Pipeline::RenderPipeline(
|
||||||
device.create_render_pipeline(&descriptor),
|
device.create_render_pipeline(&descriptor),
|
||||||
))
|
))
|
||||||
})
|
},
|
||||||
|
self.synchronous_pipeline_compilation,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn start_create_compute_pipeline(
|
fn start_create_compute_pipeline(
|
||||||
@ -771,7 +782,8 @@ impl PipelineCache {
|
|||||||
let device = self.device.clone();
|
let device = self.device.clone();
|
||||||
let shader_cache = self.shader_cache.clone();
|
let shader_cache = self.shader_cache.clone();
|
||||||
let layout_cache = self.layout_cache.clone();
|
let layout_cache = self.layout_cache.clone();
|
||||||
create_pipeline_task(async move {
|
create_pipeline_task(
|
||||||
|
async move {
|
||||||
let mut shader_cache = shader_cache.lock().unwrap();
|
let mut shader_cache = shader_cache.lock().unwrap();
|
||||||
let mut layout_cache = layout_cache.lock().unwrap();
|
let mut layout_cache = layout_cache.lock().unwrap();
|
||||||
|
|
||||||
@ -808,7 +820,9 @@ impl PipelineCache {
|
|||||||
Ok(Pipeline::ComputePipeline(
|
Ok(Pipeline::ComputePipeline(
|
||||||
device.create_compute_pipeline(&descriptor),
|
device.create_compute_pipeline(&descriptor),
|
||||||
))
|
))
|
||||||
})
|
},
|
||||||
|
self.synchronous_pipeline_compilation,
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Process the pipeline queue and create all pending pipelines if possible.
|
/// Process the pipeline queue and create all pending pipelines if possible.
|
||||||
@ -917,21 +931,34 @@ impl PipelineCache {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn create_pipeline_task(
|
|
||||||
task: impl Future<Output = Result<Pipeline, PipelineCacheError>> + Send + 'static,
|
|
||||||
) -> CachedPipelineState {
|
|
||||||
#[cfg(all(
|
#[cfg(all(
|
||||||
not(target_arch = "wasm32"),
|
not(target_arch = "wasm32"),
|
||||||
not(target_os = "macos"),
|
not(target_os = "macos"),
|
||||||
feature = "multi-threaded"
|
feature = "multi-threaded"
|
||||||
))]
|
))]
|
||||||
|
fn create_pipeline_task(
|
||||||
|
task: impl Future<Output = Result<Pipeline, PipelineCacheError>> + Send + 'static,
|
||||||
|
sync: bool,
|
||||||
|
) -> CachedPipelineState {
|
||||||
|
if !sync {
|
||||||
return CachedPipelineState::Creating(bevy_tasks::AsyncComputeTaskPool::get().spawn(task));
|
return CachedPipelineState::Creating(bevy_tasks::AsyncComputeTaskPool::get().spawn(task));
|
||||||
|
}
|
||||||
|
|
||||||
|
match futures_lite::future::block_on(task) {
|
||||||
|
Ok(pipeline) => CachedPipelineState::Ok(pipeline),
|
||||||
|
Err(err) => CachedPipelineState::Err(err),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(any(
|
#[cfg(any(
|
||||||
target_arch = "wasm32",
|
target_arch = "wasm32",
|
||||||
target_os = "macos",
|
target_os = "macos",
|
||||||
not(feature = "multi-threaded")
|
not(feature = "multi-threaded")
|
||||||
))]
|
))]
|
||||||
|
fn create_pipeline_task(
|
||||||
|
task: impl Future<Output = Result<Pipeline, PipelineCacheError>> + Send + 'static,
|
||||||
|
_sync: bool,
|
||||||
|
) -> CachedPipelineState {
|
||||||
match futures_lite::future::block_on(task) {
|
match futures_lite::future::block_on(task) {
|
||||||
Ok(pipeline) => CachedPipelineState::Ok(pipeline),
|
Ok(pipeline) => CachedPipelineState::Ok(pipeline),
|
||||||
Err(err) => CachedPipelineState::Err(err),
|
Err(err) => CachedPipelineState::Err(err),
|
||||||
|
@ -27,6 +27,7 @@ fn main() {
|
|||||||
features: WgpuFeatures::POLYGON_MODE_LINE,
|
features: WgpuFeatures::POLYGON_MODE_LINE,
|
||||||
..default()
|
..default()
|
||||||
}),
|
}),
|
||||||
|
..default()
|
||||||
}),
|
}),
|
||||||
// You need to add this plugin to enable wireframe rendering
|
// You need to add this plugin to enable wireframe rendering
|
||||||
WireframePlugin,
|
WireframePlugin,
|
||||||
|
@ -18,6 +18,7 @@ fn main() {
|
|||||||
..default()
|
..default()
|
||||||
}
|
}
|
||||||
.into(),
|
.into(),
|
||||||
|
..default()
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
.run();
|
.run();
|
||||||
|
Loading…
Reference in New Issue
Block a user