H A D | SparseGPUCodegen.cpp | 147 Value token) { in genAllocMemRef() argument 159 return builder.create<gpu::AllocOp>(loc, TypeRange({memTp, token.getType()}), in genAllocMemRef() 160 token, dynamicSizes, ValueRange()); in genAllocMemRef() 172 Value size, Value token) { in genAllocBuffer() argument 174 return builder.create<gpu::AllocOp>(loc, TypeRange({memTp, token.getType()}), in genAllocBuffer() 175 token, size, ValueRange()); in genAllocBuffer() 180 Value token) { in genAllocBuffer() argument 181 return genAllocBuffer(builder, loc, builder.getI8Type(), size, token); in genAllocBuffer() 186 Value token) { in genDeallocMemRef() argument 187 return builder.create<gpu::DeallocOp>(loc, token in genDeallocMemRef() 193 genCopyMemRef(OpBuilder & builder,Location loc,Value dst,Value src,Value token) genCopyMemRef() argument 519 genSpMat(OpBuilder & builder,Location loc,SparseTensorType & aTp,Type handleTp,Type tokenTp,Value token,Value sz1,Value sz2,Value nseA,Value rowA,Value colA,Value valA,CuSparseFormat format,bool enableRT) genSpMat() argument 603 Value token = genFirstWait(rewriter, loc); rewriteSpMV() local 701 Value token = genFirstWait(rewriter, loc); rewriteSpMM() local 806 Value token = genFirstWait(rewriter, loc); rewriteSpGEMM() local 979 Value token = genFirstWait(rewriter, loc); rewrite2To4SpMM() local 1096 Value token = genFirstWait(rewriter, loc); rewriteSDDMM() local [all...] |