candle/candle-core/tests
Laurent Mazare 844d45cde4
Bugfix for the metal elu kernel. (#2490)
* Bugfix for the metal elu kernel.

* Add a test.
2024-09-21 15:03:19 +02:00
..
conv_tests.rs Fix for backprop in ConvTranspose2D with stride of 2 (#2337) 2024-07-17 19:22:23 +02:00
custom_op_tests.rs Preliminary support for inplace ops. (#1921) 2024-03-23 14:16:19 +01:00
display_tests.rs Rename the candle crate to candle-core (#301) 2023-08-02 08:20:22 +01:00
fortran_tensor_3d.pth Add support for loading Fortran contiguous tensors (#1672) 2024-02-07 21:49:59 +01:00
grad_tests.rs Optimize the cat operation on contiguous tensors (#1855) 2024-03-17 10:49:13 +01:00
indexing_tests.rs Add a basic implementation for slice-assign. (#1377) 2023-11-26 17:31:22 +00:00
layout_tests.rs Avoid copying the data on squeeze and unsqueeze. (#1884) 2024-03-20 13:04:36 +01:00
matmul_tests.rs Add a minimal test for the metal bf16 matmul. (#2381) 2024-08-01 11:22:46 +02:00
npy.py Fix the npy read function and add some testing. (#1080) 2023-10-12 15:25:05 +02:00
pool_tests.rs Add avg_pool2d metal implementation for the metal backend (#1869) 2024-03-18 18:50:14 +01:00
pth.py Add support for loading Fortran contiguous tensors (#1672) 2024-02-07 21:49:59 +01:00
pth_tests.rs Add support for loading Fortran contiguous tensors (#1672) 2024-02-07 21:49:59 +01:00
quantized_tests.rs Add the cuda dequantize f16 kernels. (#2137) 2024-04-28 20:05:05 +02:00
serialization_tests.rs Add SliceSafetensors. (#2179) 2024-05-11 13:15:42 +02:00
tensor_tests.rs Bugfix for the metal elu kernel. (#2490) 2024-09-21 15:03:19 +02:00
test.npy Fix the npy read function and add some testing. (#1080) 2023-10-12 15:25:05 +02:00
test.npz Fix the npy read function and add some testing. (#1080) 2023-10-12 15:25:05 +02:00
test.pt Fix: pth files don't load on Windows (#1661) 2024-02-06 08:50:55 +01:00
test_with_key.pt Enhance pickle to retrieve state_dict with a given key (#1671) 2024-02-06 21:17:33 +01:00