huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the test-utils bits to a shared place. (#619)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-27 09:42:22 +01:00
committed by GitHub
parent a8b39dd7b7
commit 5320aa6b7d
17 changed files with 34 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-nn/tests/group_norm.rs
View File

@ -25,10 +25,9 @@ extern crate intel_mkl_src;
extern crate accelerate_src;
use anyhow::Result;
use candle::test_utils::to_vec3_round;
use candle::{Device, Tensor};
use candle_nn::{GroupNorm, Module};
mod test_utils;
use test_utils::to_vec3_round;
#[test]
fn group_norm() -> Result<()> {
@ -60,7 +59,7 @@ fn group_norm() -> Result<()> {
device,
)?;
assert_eq!(
to_vec3_round(gn2.forward(&input)?, 4)?,
to_vec3_round(&gn2.forward(&input)?, 4)?,
&[
[
[-0.1653, 0.3748, -0.7866],
@ -81,7 +80,7 @@ fn group_norm() -> Result<()> {
]
);
assert_eq!(
to_vec3_round(gn3.forward(&input)?, 4)?,
to_vec3_round(&gn3.forward(&input)?, 4)?,
&[
[
[0.4560, 1.4014, -0.6313],