稀疏矩阵何时能在GPU背景下创建

ti.linalg.SparseMatrixBuilder 大概什么时间支持GPU背景?

Hi @dudu, 非常欢迎来到太极论坛。

目前CPU上的Sparse Matrix 接口还需要完善,2022的Q1我们会以CPU为主。GPU上的支持可能要在Q2或者之后了。

1 个赞