Pass sparse tensor to tvm.build

[huangteng]

Hi I am struggling with sparse cmm usage and could you please help to take a look at this post ?

[Solved][Error] 'NDArray' object has no attribute 'a'

Hi, I am learning how to use csrmm topi but have encountered above error with the following code: with tvm.target.Target('llvm'): ctx = tvm.context(tgt, 0) A = tvm.contrib.sparse.placeholder((3, 3), name='A') B = te.placeholder((3, 5), name='B') D = topi.sparse.csrmm(A, B) s = te.create_schedule(D.op) # print(tvm.lower(s, [A.data, A.indices, A.indptr, B, D], simple_mode=True)) fcsrmm = tvm.build(s, [A.data, A.indices, A.indptr, B, D], tgt, target_host=tgt_host, name…

thanks