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
41 lines
1.3 KiB
Rust
41 lines
1.3 KiB
Rust
use super::*;
|
|
use burn_tensor::Tolerance;
|
|
use burn_tensor::{Int, Tensor};
|
|
|
|
#[test]
|
|
pub fn into_contiguous_match_reference_backend_1() {
|
|
for shape in [
|
|
[4, 4, 4, 4],
|
|
[32, 42, 24, 48],
|
|
[8, 3, 7, 4],
|
|
[1, 4, 1, 1],
|
|
[1, 32, 256, 128],
|
|
] {
|
|
let num_elems = shape.iter().product::<usize>() as i64;
|
|
let tensor: Tensor<TestBackend, 4> =
|
|
Tensor::<TestBackend, 1, Int>::arange(0..num_elems, &Default::default())
|
|
.reshape(shape)
|
|
.float();
|
|
let tensor_ref =
|
|
Tensor::<ReferenceBackend, 4>::from_data(tensor.to_data(), &Default::default());
|
|
|
|
for (i, j) in get_combinations(shape.len()) {
|
|
let view = tensor.clone().swap_dims(i, j);
|
|
let view_ref = tensor_ref.clone().swap_dims(i, j);
|
|
let data = view.into_data();
|
|
let data_ref = view_ref.into_data();
|
|
|
|
data_ref.assert_approx_eq::<FloatElem>(&data, Tolerance::default());
|
|
}
|
|
}
|
|
}
|
|
|
|
fn get_combinations(n: usize) -> impl Iterator<Item = (usize, usize)> {
|
|
// Iterate from 0 up to n
|
|
(0..n).flat_map(move |i| {
|
|
// For each i, iterate from i + 1 up to n
|
|
// This ensures no repeats (i == j) and no duplicates (j, i)
|
|
(i + 1..n).map(move |j| (i, j))
|
|
})
|
|
}
|