candle/candle-metal-kernels/tmp
Nicolas Patry 4349ff1fc2 Starting to fix some tests.
Few fixes.

Going back on remote metal-rs.

Reusing a single buffer (for now) to speed things up.

Adding some half kernels.

All tests are panicking instead of random failure.

Putting back f16 index select.

Add erf.

Working version for llama2-c.

Fixes + cache compute_pipeline_state.

BF16 metal fix.

Remove some prints.

new_owned -> new()..to_owned().

Better batched matmul.

Metal operational.

Reuse buffers on our own reference counts.

Tmp gemm.

Revert "Tmp gemm."

This reverts commit c65f68e988.

Interleave committing.

Speeding up copies using blit.

Fmt.

Fmt.

Remove the assert!

Fmt all.

Fixes after big rebase.

Add softmax for half and bfloat + tests

Fixing Llama example + accumulate softmax in float.
2023-11-30 11:30:31 +01:00
..
affine.rs Starting to fix some tests. 2023-11-30 11:30:31 +01:00
binary.rs Starting to fix some tests. 2023-11-30 11:30:31 +01:00
cast.rs Starting to fix some tests. 2023-11-30 11:30:31 +01:00
unary.rs Starting to fix some tests. 2023-11-30 11:30:31 +01:00