2个不稳定版本

使用旧Rust 2015

0.2.0 2016年2月4日
0.1.0 2016年1月18日

机器学习 中排名第 719

每月下载量 24
2 个crates中使用(通过 collenchyma-blas

MIT/Apache

41KB
763

rust-cuBLAS • Join the chat at https://gitter.im/autumnai/collenchyma Build Status Coverage Status License

rust-cuBLAS为CUDA的cuBLAS库提供了一个安全包装器,这样您就可以在Rust应用程序中舒适且安全地使用cuBLAS。

由于cuBLAS目前依赖于CUDA在GPU上分配内存,您可能还需要查看rust-cuda

rust-cublas是在[Autumn][autumn]为Rust机器智能框架Leaf开发的。

rust-cublas是高性能计算框架Collenchyma的一部分,用于BLAS插件。对于BLAS操作(如cuBLAS提供的)的简单、统一接口,您可能需要查看Collenchyma

有关更多信息,

入门指南

如果您使用Cargo,只需将rust-cuBLAS添加到您的Cargo.toml

[dependencies]
cublas = "0.2.0"

如果您使用Cargo Edit,可以调用

$ cargo add cublas

构建

rust-cublas 依赖于 cuBLAS 运行时库,可以从 NVIDIA 获取。

手动配置

rust-cublas 的构建脚本默认会尝试通过 pkg-config 定位 cublas。在某些情况下,这可能不起作用,例如:

  • 在没有 pkg-config 的系统上,
  • 交叉编译时,或
  • cuBLAS 没有安装到默认的系统库目录(例如 /usr/lib)。

因此,可以通过导出以下环境变量来配置构建脚本

  • CUBLAS_LIB_DIR
    如果指定,将使用此目录来查找 cuBLAS 运行时库,例如 /opt/cuda/lib

  • CUBLAS_STATIC
    如果指定,cuBLAS 库将作为静态链接而不是动态链接。

  • CUBLAS_LIBS
    如果指定,将用于查找具有不同名称的 cuBLAS 库。

如果指定了 CUBLAS_LIB_DIRCUBLAS_INCLUDE_DIR,则构建脚本将跳过 pkg-config 步骤。

如果您还需要运行编译的二进制文件,请确保它们可用

# Linux; for other platforms consult the instructions that come with cuBLAS
cd <cublas_installpath>
export LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH

贡献

想要贡献吗?太棒了!我们提供了 指导您开始贡献代码或文档的说明

我们有一个主要是实时协作的文化,在 Github 和 Collenchyma Gitter 频道 上进行。您还可以联系维护者 {@MJ@hobofan}。

变更日志

您可以在根目录的 CHANGELOG.md 文件中找到发布历史。

变更日志是记录项目所有变更的日志或记录,如网站或软件项目,通常包括错误修复、新功能等记录 - 维基百科

我们使用 Clog,Rust 自动生成 CHANGELOG 文件的工具。

许可证

根据以下任一项许可

任选其一。

贡献

除非您明确声明,否则根据 Apache-2.0 许可证定义的,您有意提交的任何贡献,均将根据上述方式双许可,而无需附加条款或条件。

依赖

~160KB