Skip to content

Latest commit

 

History

History
19 lines (15 loc) · 704 Bytes

File metadata and controls

19 lines (15 loc) · 704 Bytes

Adding A Backend

To add a backend, the following must be done:

Create Backend Device

In device.h

  • Add a backend name to enum class Backend
  • Handle the to_string case for your new backend
  • Create a shortname constexpr Device

Implement Backend

In tinytensor/tensor/backend

  • Create a folder for all your implementation files
  • Your backend must derive and extend BackendBase

Register The Backend

Add your device-to-backend in get_backend

Register Backend For Testing

Add your backend device in runner_single_type