no-std zyx-opencl

Zyx OpenCL backend

2 releases

0.1.1 Mar 21, 2024
0.1.0 Mar 9, 2024

#494 in Machine learning

Download history 7/week @ 2024-09-23 3/week @ 2024-09-30

72 downloads per month

MIT/Apache

305KB
7K SLoC

zyx-opencl

OpenCL backend for zyx machine learning library.

This backend implements:

  • shape specialization
  • ops fusion

If wondering which zyx backend to use, try using this backend first.

For README and source code, please visit github.

For more details, there is a book.

Cargo features

  • std - enables zyx-core/std
  • debug1 - enables printing of information about devices and prints compiled opencl kernels
  • CL_VERSION_1_1 - enables opencl v1.1
  • CL_VERSION_1_2 - enables opencl v1.2
  • CL_VERSION_2_1 - enables opencl v2.1

lib.rs:

OpenCL backend for zyx

Initialize backend. You can use builder if you want to change settings of the backend.

let dev = zyx_opencl::device()?;
let dev = zyx_opencl::device_builder().build()?;

For README, quick tutorial and source code, please visit https://www.github.com/zk4x/zyx.

For more details, there is a book.

Dependencies

~670KB
~13K SLoC