Laurent Mazare 02b5c38049 Use cublas bf16. (#101)

2023-07-07 08:00:12 +01:00

..

backprop.rs

Sketch the conv1d op.

2023-07-04 10:52:34 +01:00

conv.rs

Proper conv1d dispatch.

2023-07-04 11:29:28 +01:00

cpu_backend.rs

MKL adjustments. (#87)

2023-07-06 11:37:27 +01:00

cuda_backend.rs

Use cublas bf16. (#101)

2023-07-07 08:00:12 +01:00

device.rs

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

display.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

dtype.rs

Add dtype support.

2023-07-02 20:12:26 +01:00

dummy_cuda_backend.rs

Add more of the conv1d op.

2023-07-04 11:15:45 +01:00

error.rs

MKL adjustments. (#87)

2023-07-06 11:37:27 +01:00

layout.rs

Get the cpu backend to compile.

2023-06-28 14:12:38 +01:00

lib.rs

Add a simpler way to specify the dim index for some ops.

2023-07-05 20:22:43 +01:00

npy.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

op.rs

Sketch the conv1d op.

2023-07-04 10:52:34 +01:00

safetensors.rs

Update candle-core/src/safetensors.rs

2023-07-06 23:59:54 +02:00

shape.rs

Add a simpler way to specify the dim index for some ops.

2023-07-05 20:22:43 +01:00

storage.rs

Add more of the conv1d op.

2023-07-04 11:15:45 +01:00

strided_index.rs

Get the cpu tests to run.

2023-06-28 14:32:02 +01:00

tensor.rs

Sketch the Falcon model. (#93)

2023-07-06 19:01:21 +01:00

utils.rs

PyTorch like display implementation.

2023-06-27 21:16:35 +01:00