Lines Matching defs:spMat

646 static gpu::Prune2To4SpMatFlag get2To4PruneFlag(Value spMat) {
647   return spMat.getDefiningOp<gpu::Create2To4SpMatOp>().getPruneFlag();
657 static bool is2To4Sparsity(Value spMat) {
658   if (auto op = spMat.getDefiningOp<gpu::Create2To4SpMatOp>())
660   if (auto op = spMat.getDefiningOp<gpu::CreateCooOp>())
662   if (auto op = spMat.getDefiningOp<gpu::CreateCooAoSOp>())
664   if (auto op = spMat.getDefiningOp<gpu::CreateCsrOp>())
666   if (auto op = spMat.getDefiningOp<gpu::CreateCscOp>())
668   if (auto op = spMat.getDefiningOp<gpu::CreateBsrOp>())
670   // Print the spMat defining op
671   spMat.getDefiningOp()->print(llvm::errs());