2 releases
0.1.1 | Mar 21, 2024 |
---|---|
0.1.0 | Mar 9, 2024 |
#1455 in Machine learning
72 downloads per month
305KB
7K
SLoC
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.
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
Dependencies
~705KB
~13K SLoC