Laurent Mazare e27b4700ad Indexing with max-value results in zero/no-op. (#2940) ...

* Indexing with max-value results in zero/no-op.

* Add some testing.

* Also adapt the metal kernels.

* Another test.

* Fix.

2025-05-03 11:36:31 +02:00

..

cpu

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

cpu_backend

Indexing with max-value results in zero/no-op. (#2940)

2025-05-03 11:36:31 +02:00

cuda_backend

Add the scatter in place ops. (#2923)

2025-04-26 07:36:49 +02:00

metal_backend

Remove redundant mlx gemm dtype check (#2925)

2025-04-27 06:14:57 +02:00

quantized

Cuda cleanup. (#2880)

2025-04-11 21:43:35 +02:00

accelerate.rs

feat: add silu activation function (#1706)

2024-02-14 10:27:22 +01:00

backend.rs

Add the scatter in place ops. (#2923)

2025-04-26 07:36:49 +02:00

backprop.rs

Add the scatter op. (#2921)

2025-04-25 21:46:58 +02:00

conv.rs

Expose the cudnn algo in the conv ops. (#2892)

2025-04-14 08:25:32 +02:00

convert.rs

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

custom_op.rs

Update to cudarc 0.14 (breaking change). (#2858)

2025-04-03 09:12:19 +02:00

device.rs

Optimize Tensor::new when called on nested Vec<..>. (#2927)

2025-04-28 09:19:45 +02:00

display.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

dtype.rs

Indexing with max-value results in zero/no-op. (#2940)

2025-05-03 11:36:31 +02:00

dummy_cuda_backend.rs

Add the scatter in place ops. (#2923)

2025-04-26 07:36:49 +02:00

dummy_metal_backend.rs

Add the scatter in place ops. (#2923)

2025-04-26 07:36:49 +02:00

error.rs

fix: place ug dep behind not wasm32 flag (#2760)

2025-02-01 23:05:52 +01:00

indexer.rs

Complete the missing backticks in the comments (#2469)

2024-09-11 16:37:05 +02:00

layout.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

lib.rs

Add a Context trait similar to anyhow::Context. (#2676)

2024-12-22 09:18:13 +01:00

mkl.rs

feat: add silu activation function (#1706)

2024-02-14 10:27:22 +01:00

npy.rs

Update zip requirement from 0.6.6 to 1.1.1 (#2103)

2024-04-22 16:23:27 +02:00

op.rs

Add the scatter op. (#2921)

2025-04-25 21:46:58 +02:00

pickle.rs

Fix for clippy 1.86. (#2864)

2025-04-03 19:38:27 +02:00

safetensors.rs

Lint fixes introduced with Rust 1.83 (#2646)

2024-11-28 23:00:21 +01:00

scalar.rs

Add an enum for scalar values. (#2909)

2025-04-18 22:13:38 +02:00

shape.rs

Refactor From<Tuple> implementations by using macros, add tests (#2762)

2025-02-19 10:58:29 +01:00

sort.rs

Cuda cleanup. (#2880)

2025-04-11 21:43:35 +02:00

storage.rs

Add the scatter in place ops. (#2923)

2025-04-26 07:36:49 +02:00

streaming.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

strided_index.rs

Clippy fixes for 1.84. (#2710)

2025-01-10 10:15:15 +01:00

tensor_cat.rs

Check the bounds in the cuda indexing kernels. (#2908)

2025-04-18 20:08:17 +02:00

tensor.rs

Switch Tensor::full to return a contiguous tensor. (#2929)

2025-04-28 21:36:39 +02:00

test_utils.rs

Add flip to tensor (#2855)

2025-04-01 09:07:16 +02:00

utils.rs

20241118 docs (#2629)

2024-11-19 04:07:07 +01:00

variable.rs

Bug Fix: When converting a tensor to a variable, clone if the tensor is already a variable. (#2124)

2024-04-29 11:21:53 +02:00