2 个版本

0.1.1 2024年3月21日
0.1.0 2024年3月9日

237机器学习 中排名

Download history 4/week @ 2024-04-05 3/week @ 2024-04-26 45/week @ 2024-05-03

每月下载量 72

MIT/Apache

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