dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore(tests): allow overriding autographics backend (#1005)

This commit is contained in:
David Chavez 2023-11-27 19:58:43 +01:00 committed by GitHub
parent 929b1786bb
commit 60c24430c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
burn-wgpu/src/graphics.rs
View File

@ -78,9 +78,27 @@ impl GraphicsApi for WebGpu {
impl GraphicsApi for AutoGraphicsApi { impl GraphicsApi for AutoGraphicsApi {
fn backend() -> wgpu::Backend { fn backend() -> wgpu::Backend {
// Allow overriding AutoGraphicsApi backend with ENV var in std test environments
#[cfg(not(no_std))]
#[cfg(test)]
if let Ok(backend_str) = std::env::var("AUTO_GRAPHICS_BACKEND") {
match backend_str.to_lowercase().as_str() {
"metal" => return wgpu::Backend::Metal,
"vulkan" => return wgpu::Backend::Vulkan,
"dx12" => return wgpu::Backend::Dx12,
"opengl" => return wgpu::Backend::Gl,
"webgpu" => return wgpu::Backend::BrowserWebGpu,
_ => {
eprintln!("Invalid graphics backend specified in GRAPHICS_BACKEND environment variable");
std::process::exit(1);
}
}
}
// In a no_std environment or if the environment variable is not set
#[cfg(target_os = "macos")] #[cfg(target_os = "macos")]
return wgpu::Backend::Metal; return wgpu::Backend::Metal;
#[cfg(not(target_os = "macos"))] #[cfg(not(target_os = "macos"))]
wgpu::Backend::Vulkan return wgpu::Backend::Vulkan;
} }
} }