From 695458daea140a6413987da55ea3d8950cfff731 Mon Sep 17 00:00:00 2001 From: Yuanqiang Liu Date: Sun, 28 Apr 2024 02:24:55 +0800 Subject: [PATCH] Fix ArgAnnotation with boolean flag which instructs value semantics (#3238) --- projects/pt1/python/torch_mlir_e2e_test/annotations.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/projects/pt1/python/torch_mlir_e2e_test/annotations.py b/projects/pt1/python/torch_mlir_e2e_test/annotations.py index 5a07b306e..e34b0f85d 100644 --- a/projects/pt1/python/torch_mlir_e2e_test/annotations.py +++ b/projects/pt1/python/torch_mlir_e2e_test/annotations.py @@ -44,7 +44,7 @@ def export(fn): return fn -ArgAnnotation = Tuple[List[int], torch.dtype] +ArgAnnotation = Tuple[List[int], torch.dtype, bool] # TODO: Replace with py3 extended argument annotations when available. @@ -55,9 +55,9 @@ def annotate_args(annotations: List[Optional[ArgAnnotation]]): The `annotations` should be a list of the same length as the number of argument to the method (including `self`). Each list entry is either: - None, corresponding to providing the compiler with no information. - - A 2-tuple consisting of a shape and a dtype, such as - `([2, 3, 4], torch.float32)`. A dimension with an unknown size can be - indicated by using `-1` as the size. This provides the compiler a + - A 3-tuple consisting of a shape, a dtype and a flag of value semantics, + such as `([2, 3, 4], torch.float32, True)`. A dimension with an unknown size + can be indicated by using `-1` as the size. This provides the compiler a guarantee that the argument will always dynamically have the described shape and dtype. """