diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-27 09:42:22 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-27 09:42:22 +0100 |
| commit | 5320aa6b7d339ff594d3886dd29634ea8cde6f17 (patch) | |
| tree | b8dc53eaea3966c288ac3c4a597b4d36c2deefa4 /candle-core/tests/custom_op_tests.rs | |
| parent | a8b39dd7b784b3c3cdd4d228813bd48b2d0d79bb (diff) | |
| download | candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.tar.gz candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.tar.bz2 candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.zip | |
Move the test-utils bits to a shared place. (#619)
Diffstat (limited to 'candle-core/tests/custom_op_tests.rs')
| -rw-r--r-- | candle-core/tests/custom_op_tests.rs | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/candle-core/tests/custom_op_tests.rs b/candle-core/tests/custom_op_tests.rs index 7ec04c6a..cff0aebe 100644 --- a/candle-core/tests/custom_op_tests.rs +++ b/candle-core/tests/custom_op_tests.rs @@ -1,10 +1,8 @@ use candle_core::backend::BackendStorage; use candle_core::cpu_backend; +use candle_core::test_utils::to_vec1_round; use candle_core::{CpuStorage, CustomOp1, DType, Device, Error, Layout, Result, Shape, Tensor}; -mod test_utils; -use test_utils::to_vec1_round; - fn fwd<T: num_traits::Float>(v: T, alpha: f64) -> T { if v.is_sign_positive() { v |