| 45 | } |
| 46 | |
| 47 | std::string MatmulKernel::GetKernelSymbol(TContext* context) const { |
| 48 | std::stringstream ss; |
| 49 | ss << "mlir_auto_matmul_"; |
| 50 | if (context->getAttrBool("transposeA")) { |
| 51 | ss << "t"; |
| 52 | } else { |
| 53 | ss << "n"; |
| 54 | } |
| 55 | if (context->getAttrBool("transposeB")) { |
| 56 | ss << "t"; |
| 57 | } else { |
| 58 | ss << "n"; |
| 59 | } |
| 60 | return ss.str(); |
| 61 | } |
| 62 | |
| 63 | void MatmulKernel::CreateCompute( |
| 64 | Block* entryBlock, mlir::OpBuilder& op_builder, mlir::MLIRContext* ctx, |