Home
last modified time | relevance | path

Searched defs:enableRT (Results 1 – 3 of 3) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp472 SparseTensorType cTp, bool enableRT, in getCuSparseFormat() argument
495 CuSparseFormat format, bool enableRT) { in genFirstPosOrCrds() argument
508 CuSparseFormat format, bool enableRT) { in genSecondCrds() argument
521 CuSparseFormat format, bool enableRT) { in genSpMat() argument
563 linalg::GenericOp op, bool enableRT) { in rewriteSpMV() argument
660 rewriteSpMM(PatternRewriter & rewriter,linalg::GenericOp op,bool enableRT) rewriteSpMM() argument
761 rewriteSpGEMM(PatternRewriter & rewriter,linalg::GenericOp op,bool enableRT) rewriteSpGEMM() argument
1053 rewriteSDDMM(PatternRewriter & rewriter,linalg::GenericOp op,bool enableRT) rewriteSDDMM() argument
1310 bool enableRT; global() member
1331 populateSparseGPULibgenPatterns(RewritePatternSet & patterns,bool enableRT) populateSparseGPULibgenPatterns() argument
[all...]
H A DSparseTensorPasses.cpp458 createLowerSparseOpsToForeachPass(bool enableRT,bool enableConvert) createLowerSparseOpsToForeachPass() argument
511 createSparseGPUCodegenPass(unsigned numThreads,bool enableRT) createSparseGPUCodegenPass() argument
H A DSparseTensorRewriting.cpp1576 populateLowerSparseOpsToForeachPatterns(RewritePatternSet & patterns,bool enableRT,bool enableConvert) populateLowerSparseOpsToForeachPatterns() argument