diff --git a/include/xten/Dialect/XTenNN/IR/XTenNNOps.td b/include/xten/Dialect/XTenNN/IR/XTenNNOps.td index 55220c5d..364655f8 100644 --- a/include/xten/Dialect/XTenNN/IR/XTenNNOps.td +++ b/include/xten/Dialect/XTenNN/IR/XTenNNOps.td @@ -277,6 +277,17 @@ def XTenNN_KernelOp : XTenNN_Op<"kernel", []> { ); let results = (outs Variadic:$results); + let builders = [ + OpBuilder<(ins + "::mlir::TypeRange":$results, + "::mlir::ValueRange":$arguments, + "::llvm::StringRef":$name), [{ + build($_builder, $_state, results, arguments, name, + ::mlir::ArrayAttr(), ::mlir::ArrayAttr()); + }] + > + ]; + let hasCustomAssemblyFormat = 1; let hasVerifier = 1; }