feat: toy tutorial chapter 5.

Signed-off-by: jackfiled <xcrenchangjun@outlook.com>
This commit is contained in:
2025-06-05 23:46:52 +08:00
parent 902915a57b
commit c5ab1a6bc0
5 changed files with 393 additions and 18 deletions

View File

@@ -14,6 +14,8 @@ namespace mlir
namespace hello
{
std::unique_ptr<Pass> createShapeInferencePass();
std::unique_ptr<Pass> createLowerToAffineLoopsPass();
}
}

View File

@@ -224,7 +224,7 @@ def PrintOp : Hello_Op<"print"> {
}];
// The print operation takes an input tensor to print.
let arguments = (ins F64Tensor:$input);
let arguments = (ins AnyTypeOf<[F64Tensor, F64MemRef]>:$input);
let assemblyFormat = "$input attr-dict `:` type($input)";
}