Laurent Mazare a4d5a414e3 Support gather on bf16 for metal. (#2035)

2024-04-10 12:49:25 +02:00

..

cpu

…

cpu_backend

Fix the matmul layout for accelerate & mkl. (#2011)

2024-04-04 19:18:03 +02:00

cuda_backend

Split the cuda error file. (#2003)

2024-04-04 08:27:23 +02:00

metal_backend

Support gather on bf16 for metal. (#2035)

2024-04-10 12:49:25 +02:00

quantized

Handle the batch dimension in quantized MMV on metal. (#2022)

2024-04-06 20:02:24 +02:00

accelerate.rs

feat: add silu activation function (#1706)

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

backend.rs

Add the alloc_uninit function. (#1901)

2024-03-22 07:25:23 +01:00

backprop.rs

Add support for "sign" on tensors (#2012)

2024-04-04 22:32:47 +02:00

conv.rs

Support for groups in conv-transpose1d. (#1731)

2024-02-18 21:28:07 +01:00

convert.rs

…

custom_op.rs

Preliminary support for inplace ops. (#1921)

2024-03-23 14:16:19 +01:00

device.rs

Add the alloc_uninit function. (#1901)

2024-03-22 07:25:23 +01:00

display.rs

Expose more printer options. (#1817)

2024-03-08 15:04:18 +01:00

dtype.rs

Implement the error trait for DTypeParseError. (#1852)

2024-03-15 08:37:27 +01:00

dummy_cuda_backend.rs

Add the alloc_uninit function. (#1901)

2024-03-22 07:25:23 +01:00

dummy_metal_backend.rs

Add the alloc_uninit function. (#1901)

2024-03-22 07:25:23 +01:00

error.rs

Metal part 1 - Scaffolding for metal. (#1308)

2023-11-10 08:35:48 +01:00

indexer.rs

Fix a couple typos (#1451)

2023-12-17 05:20:05 -06:00

layout.rs

Expose a couple layout methods. (#1816)

2024-03-08 10:52:22 +01:00

lib.rs

first commit (#2018)

2024-04-05 15:20:28 +02:00

mkl.rs

feat: add silu activation function (#1706)

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

npy.rs

Fix the npy read function and add some testing. (#1080)

2023-10-12 15:25:05 +02:00

op.rs

Add support for "sign" on tensors (#2012)

2024-04-04 22:32:47 +02:00

pickle.rs

Fix float unpickling. (#1730)

2024-02-18 19:33:55 +01:00

safetensors.rs

Add the buffered safetensor wrapper. (#948)

2023-09-23 22:57:42 +01:00

scalar.rs

…

shape.rs

Another fix for squeezing. (#1943)

2024-03-26 17:05:26 +01:00

storage.rs

Really unique identifier for metal device ids. (#1932)

2024-03-25 11:48:16 +01:00

strided_index.rs

…

tensor_cat.rs

Remove some unnecessary calls to contiguous. (#1968)

2024-03-30 13:22:00 +01:00

tensor.rs

Add support for "sign" on tensors (#2012)

2024-04-04 22:32:47 +02:00

test_utils.rs

Adding the test scaffolding.

2023-11-20 14:38:35 +01:00

utils.rs

Metal part 1 - Scaffolding for metal. (#1308)

2023-11-10 08:35:48 +01:00

variable.rs

Detach the tensors on batch-norm eval. (#1702)

2024-02-13 14:26:32 +01:00