Ben_Kosytorz
3a67c0979c
- Updated stablediffusion crate path from "../stable-diffusion-burn" to "./crates/stable-diffusion-burn" for proper workspace resolution - Enhanced .gitignore to include generated model files (.mpk, .pt, .bin, .safetensors, .ckpt) and user_data directory - Added Cargo.lock to gitignore with appropriate comment - Reorganized IDE files section in gitignore for better clarity - Added newline at end of file for proper formatting
48 lines
1.5 KiB
Rust
48 lines
1.5 KiB
Rust
// Re-export
|
|
use super::FloatElemType;
|
|
|
|
// Default
|
|
#[cfg(feature = "ndarray")]
|
|
pub type TestBackend = burn_ndarray::NdArray<FloatElemType>;
|
|
|
|
#[cfg(feature = "tch")]
|
|
pub type TestBackend = burn_tch::LibTorch<FloatElemType>;
|
|
|
|
#[cfg(feature = "cuda")]
|
|
pub type TestBackend = burn_cuda::Cuda<FloatElemType, super::IntElemType>;
|
|
|
|
#[cfg(feature = "rocm")]
|
|
pub type TestBackend = burn_rocm::Rocm<FloatElemType, super::IntElemType>;
|
|
|
|
#[cfg(feature = "wgpu")]
|
|
pub type TestBackend = burn_wgpu::Wgpu<FloatElemType, super::IntElemType>;
|
|
|
|
#[cfg(feature = "cpu")]
|
|
pub type TestBackend = burn_cpu::Cpu<FloatElemType, super::IntElemType>;
|
|
|
|
#[cfg(feature = "router")]
|
|
pub type TestBackend = burn_router::BackendRouter<
|
|
burn_router::DirectByteChannel<(burn_ndarray::NdArray, burn_wgpu::Wgpu)>,
|
|
>;
|
|
|
|
/// Collection of types used across tests
|
|
#[allow(unused)]
|
|
pub mod prelude {
|
|
pub use burn_autodiff::Autodiff;
|
|
pub use burn_tensor::Tensor;
|
|
|
|
use super::*;
|
|
pub type TestTensor<const D: usize> = Tensor<TestBackend, D>;
|
|
pub type TestTensorInt<const D: usize> = Tensor<TestBackend, D, burn_tensor::Int>;
|
|
pub type TestTensorBool<const D: usize> = Tensor<TestBackend, D, burn_tensor::Bool>;
|
|
|
|
pub type FloatElem = burn_tensor::ops::FloatElem<TestBackend>;
|
|
pub type IntElem = burn_tensor::ops::IntElem<TestBackend>;
|
|
|
|
pub type TestAutodiffBackend = Autodiff<TestBackend>;
|
|
pub type TestAutodiffTensor<const D: usize> = Tensor<TestAutodiffBackend, D>;
|
|
}
|
|
|
|
#[allow(unused)]
|
|
pub use prelude::*;
|