2 releases
0.1.1 | Mar 21, 2024 |
---|---|
0.1.0 | Mar 9, 2024 |
#494 in Machine learning
72 downloads per month
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