2 个版本
0.1.1 | 2024年3月21日 |
---|---|
0.1.0 | 2024年3月9日 |
237 在 机器学习 中排名
每月下载量 72
305KB
7K SLoC
zyx-opencl
zyx 机器学习库的 OpenCL 后端。
此后端实现
- 形状特殊化
- 操作融合
如果想知道使用哪个 zyx 后端,请先尝试使用此后端。
有关 README 和源代码,请访问 github。
有关更多详细信息,请参阅 书籍。
Cargo 功能
- std - 启用 zyx-core/std
- debug1 - 启用打印有关设备和编译的 opencl 内核的信息
- CL_VERSION_1_1 - 启用 opencl v1.1
- CL_VERSION_1_2 - 启用 opencl v1.2
- CL_VERSION_2_1 - 启用 opencl v2.1
lib.rs
:
zyx 的 OpenCL 后端
初始化后端。如果您想更改后端的设置,可以使用构建器。
let dev = zyx_opencl::device()?;
let dev = zyx_opencl::device_builder().build()?;
有关 README、快速教程和源代码,请访问 https://www.github.com/zk4x/zyx。
有关更多详细信息,请参阅 书籍。
依赖关系
~675KB
~13K SLoC