H A D | SparseGPUCodegen.cpp | 1 //===- SparseGPUCodegen.cpp - Generates GPU code -------- 147 genAllocMemRef(OpBuilder & builder,Location loc,Value mem,Value token) genAllocMemRef() argument 172 genAllocBuffer(OpBuilder & builder,Location loc,Type type,Value size,Value token) genAllocBuffer() argument 180 genAllocBuffer(OpBuilder & builder,Location loc,Value size,Value token) genAllocBuffer() argument 186 genDeallocMemRef(OpBuilder & builder,Location loc,Value mem,Value token) genDeallocMemRef() argument 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...] |