candle/candle-core/tests
Laurent Mazare ab86cd37c8
Support i64 in index-select on metal. (#1951)
* Support i64 in index-select on metal.

* Add some testing of index-select for all dtypes.
2024-03-27 16:30:07 +01:00
..
conv_tests.rs Backwards for ConvTranspose2D (#1910) 2024-03-23 07:05:55 +01: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
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 a cuda kernel for dequantizing q8_0. (#1804) 2024-03-05 09:50:37 +01:00
serialization_tests.rs Fix the npy read function and add some testing. (#1080) 2023-10-12 15:25:05 +02:00
tensor_tests.rs Support i64 in index-select on metal. (#1951) 2024-03-27 16:30:07 +01: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