burn/burn-candle
David Chavez f73136e3df
chore(candle): Allow enabling accelerate (#1009)
* chore(candle): Allow enabling accelerate

* Temporarily disable test for accelerate feature

* Allow enabling accelerate from upstream

* Update the README

* Have xtask also test using accelerate

* Renable failing test

* Fix matmul on candle when using accelerate

* Add additional comment to xtask method
2023-11-30 13:03:00 -05:00
..
src chore(candle): Allow enabling accelerate (#1009) 2023-11-30 13:03:00 -05:00
Cargo.toml chore(candle): Allow enabling accelerate (#1009) 2023-11-30 13:03:00 -05:00
README.md chore(candle): Allow enabling accelerate (#1009) 2023-11-30 13:03:00 -05:00

README.md

Burn Candle Backend

This crate provides a backend for Burn based on the Candle framework.

It is still in alpha stage, not all operations are supported. It is usable for some use cases, like for inference.

It can be used with CPU or CUDA. On macOS computations can be accelerated by using the Accelerate framework.

Feature Flags

The following features are supported:

  • cuda - Cuda GPU device (NVIDIA only)
  • accelerate - Accelerate framework (macOS only)