Ndarry是Rust编程语言中的高性能多维、多类型数组库,类似numpy。作者使用cublas库实现了在CUDA上运行的矩阵乘法函数,并通过FFI连接到Rust中。作者还实现了一个trait来给NdArray数组添加cuda_dot方法,实现了1D和2D矩阵之间的点乘。cublas在高维度矩阵乘法场景下展现出巨大优势。
完成下面两步后,将自动完成登录并继续当前操作。