Laurent Mazare 557b2c28dd Q6K quantization (#495) ...

* Print the detected arch options.

* Add the q6k quantization.

* Add a currently broken test.

* Bugfix.

* Bugfix.

* Another bugfix.

* Another bugfix + get the test to work.

2023-08-17 22:22:57 +01:00

..

conv_tests.rs

Add a naive conv2d cuda kernel. (#438)

2023-08-14 10:34:42 +01:00

custom_op_tests.rs

Relax the requirements on CustomOp. (#486)

2023-08-17 11:12:05 +01:00

display_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

grad_tests.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00

indexing_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

layout_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

pool_tests.rs

Add a cuda kernel for upsampling. (#441)

2023-08-14 13:12:17 +01:00

quantized_tests.rs

Q6K quantization (#495)

2023-08-17 22:22:57 +01:00

tensor_tests.rs

Further randn tweaks: use the appropriate rng rather than the f64 one, some cleanup. (#383)

2023-08-10 05:48:19 +01:00

test_utils.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00