cubecl

简短摘要

#[cube] 函数是使用普通 Rust 编写的,而不是使用单独的着色器语言。这意味着它可以进行类型检查、借用检查、组合和测试,您无需在运行时切换到另一种语言或通过字符串连接构建着色器源代码。在不牺牲性能的情况下实现了这些功能。Comptime 在编译时专门化内核,autotune 在首次运行时搜索配置空间,并且在支持它们的硬件上会自动采用张量核心路径。

同样的特性使得 CubeCL 成为 Rust 中科学计算的合理基础。它位于 wgpu 和 cudarc 等低级封装和 Burn 等高级框架之间，因此可以一次编写内核库，并在整个生态系统中重复使用。