diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-06-29 12:10:19 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-29 12:10:19 +0100 |
| commit | c0719b778186d4498fa855f4a997d2fd63ea037d (patch) | |
| tree | 7dce00b52392a2176725a5a6f6987fd095aaabd8 /candle-core/tests/tensor_tests.rs | |
| parent | 3872dc4751c45b625d71c6652c2854a3cc695fb3 (diff) | |
| parent | 2741b39ad37ecb58c110459739ee174fae5f1fa4 (diff) | |
| download | candle-c0719b778186d4498fa855f4a997d2fd63ea037d.tar.gz candle-c0719b778186d4498fa855f4a997d2fd63ea037d.tar.bz2 candle-c0719b778186d4498fa855f4a997d2fd63ea037d.zip | |
Merge pull request #35 from LaurentMazare/const-scalar
Use broadcasted scalars for const tensors.
Diffstat (limited to 'candle-core/tests/tensor_tests.rs')
| -rw-r--r-- | candle-core/tests/tensor_tests.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-core/tests/tensor_tests.rs b/candle-core/tests/tensor_tests.rs index 8ac0c9f2..78ca4b05 100644 --- a/candle-core/tests/tensor_tests.rs +++ b/candle-core/tests/tensor_tests.rs @@ -263,12 +263,12 @@ fn matmul(device: &Device) -> Result<()> { let a_tt = a.t()?.contiguous()?.t()?; assert!(!a_tt.is_contiguous()); assert_eq!(a.dims(), a_tt.dims()); - assert_eq!(a_tt.stride_tmp(), &[6, 1, 2]); + assert_eq!(a_tt.stride(), &[6, 1, 2]); let b_tt = b.t()?.contiguous()?.t()?; assert!(!b_tt.is_contiguous()); assert_eq!(b.dims(), b_tt.dims()); - assert_eq!(b_tt.stride_tmp(), &[6, 1, 3]); + assert_eq!(b_tt.stride(), &[6, 1, 3]); assert_eq!(a_tt.matmul(&b)?.to_vec3::<f32>()?, &expected); assert_eq!(a.matmul(&b_tt)?.to_vec3::<f32>()?, &expected); |