CuPy稀疏核
CuPy稀疏核
提问于 2022-09-05 02:06:18
用于编写在cuSPARSE实例上工作的自定义(按元素或原始)内核的API是什么？如果我想编写一个可以将cupyx.scipy.sparse.csr_matrix实例作为参数的内核，那么底层的CUDA代码需要使用哪些参数？
回答 1
Stack Overflow用户
回答已采纳
发布于 2022-09-05 04:31:14
找到了(至少大约)。对于CSR稀疏矩阵s，字段s.data、s.indices和s.indptr分别包含非零项的长度s.nnz的密集数组、列索引的长度s.nnz的密集数组以及indices中每一行起始位置(相对地址)的列表。这些都是普通的cupy.ndarray，可以作为普通的传递到内核，然后传递到cuSPARSE函数。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/73603963
复制相关文章
相似问题
腾讯云开发者
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 
粤公网安备44030502008569号
腾讯云计算（北京）有限责任公司 京ICP证150476号 | 京ICP备11018762号