Lines Matching defs:tokenTp

518                            SparseTensorType &aTp, Type handleTp, Type tokenTp,
526       return builder.create<gpu::CreateCooOp>(loc, handleTp, tokenTp, token,
531     return builder.create<gpu::CreateCooAoSOp>(loc, handleTp, tokenTp, token,
539     return builder.create<gpu::CreateCsrOp>(loc, handleTp, tokenTp, token, sz1,
542     return builder.create<gpu::CreateCscOp>(loc, handleTp, tokenTp, token, sz1,
556   return builder.create<gpu::CreateBsrOp>(loc, handleTp, tokenTp, token, bRows,
602   Type tokenTp = rewriter.getType<gpu::AsyncTokenType>();
605       genSpMat(rewriter, loc, aTp, spmatHandleTp, tokenTp, token, szY, szX,
610       loc, dnTensorHandleTp, tokenTp, token, vecX, szX);
614       loc, dnTensorHandleTp, tokenTp, token, vecY, szY);
621       loc, indexTp, tokenTp, token, spMatA, dnX, dnY,
631       loc, tokenTp, token, spMatA, dnX, dnY, /*computeType=*/dnYType, buffer);
635   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
637   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnX)
639   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnY)
700   Type tokenTp = rewriter.getType<gpu::AsyncTokenType>();
703       genSpMat(rewriter, loc, aTp, spMatHandleTp, tokenTp, token, szm, szk,
708       loc, dnTensorHandleTp, tokenTp, token, matB,
713       loc, dnTensorHandleTp, tokenTp, token, matC,
721       loc, indexTp, tokenTp, token, spMatA, dnB, dnC,
732       loc, tokenTp, token, spMatA, dnB, dnC, /*computeType=*/dnCType, buffer);
736   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
738   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnB)
740   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnC)
805   Type tokenTp = rewriter.getType<gpu::AsyncTokenType>();
808       genSpMat(rewriter, loc, aTp, spmatHandleTp, tokenTp, token, szm, szk,
813       genSpMat(rewriter, loc, bTp, spmatHandleTp, tokenTp, token, szk, szn,
832       genSpMat(rewriter, loc, cTp, spmatHandleTp, tokenTp, token, szm, szn,
839       rewriter.create<gpu::SpGEMMCreateDescrOp>(loc, descTp, tokenTp, token);
843       loc, indexTp, tokenTp, token, desc, gpu::TransposeMode::NON_TRANSPOSE,
852       loc, indexTp, tokenTp, token, desc, gpu::TransposeMode::NON_TRANSPOSE,
860       loc, indexTp, tokenTp, token, desc, gpu::TransposeMode::NON_TRANSPOSE,
869       loc, indexTp, tokenTp, token, desc, gpu::TransposeMode::NON_TRANSPOSE,
876       loc, indexTp, indexTp, indexTp, tokenTp, token, spMatC);
888       loc, tokenTp, token, spMatC, rowC, colC, valC);
891       loc, tokenTp, token, desc, gpu::TransposeMode::NON_TRANSPOSE,
901   token = rewriter.create<gpu::SpGEMMDestroyDescrOp>(loc, tokenTp, token, desc)
903   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
905   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatB)
907   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatC)
978   Type tokenTp = rewriter.getType<gpu::AsyncTokenType>();
981       loc, spMatHandleTp, tokenTp, token, szm, szk,
986       loc, dnTensorHandleTp, tokenTp, token, matB,
991       loc, dnTensorHandleTp, tokenTp, token, matC,
1001       loc, bufferTypes, tokenTp, token, gpu::TransposeMode::NON_TRANSPOSE,
1023       loc, tokenTp, token, spMatA, dnB, dnC, /*computeType=*/dnCType,
1028   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
1030   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnB)
1032   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnC)
1095   Type tokenTp = rewriter.getType<gpu::AsyncTokenType>();
1098       loc, dnMatHandleTp, tokenTp, token, matA, SmallVector<Value>{szm, szk});
1102       loc, dnMatHandleTp, tokenTp, token, matB, SmallVector<Value>{szk, szn});
1106       genSpMat(rewriter, loc, cTp, spMatHandleTp, tokenTp, token, szm, szn,
1114       loc, indexTp, tokenTp, token, dnA, dnB, spMatC, dnCType);
1122   auto sddmmComp = rewriter.create<gpu::SDDMMOp>(loc, tokenTp, token, dnA, dnB,
1127   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnA)
1129   token = rewriter.create<gpu::DestroyDnTensorOp>(loc, tokenTp, token, dnB)
1131   token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatC)