huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Also run the backprop tests on cuda.

Browse Source
This commit is contained in:
laurent
2023-06-28 08:15:03 +01:00
parent a457020d50
commit 395c84e80a
3 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/tests/tensor_tests.rs
View File

@ -1,4 +1,3 @@
// TODO: Also test the cuda backend.
mod test_utils;
use candle::{DType, Device, Result, Tensor};
use test_utils::to_vec3_round;