vinx13 commented on a change in pull request #9492:
URL: https://github.com/apache/tvm/pull/9492#discussion_r761493148
##########
File path: python/tvm/script/tir/ty.py
##########
@@ -67,6 +68,71 @@ def __getitem__(self, vtypes):
return ConcreteType(tvm.ir.TupleType([vtype.evaluate() for vtype in
vtypes]))
+class GenericBufferType(SpecialStmt): # pylint:
disable=too-few-public-methods, abstract-method
+ """TVM script typing class for uniform Type objects"""
+
+ def __init__(self, vtype):
+ def match_buffer_syntax_sugar(
+ shape,
+ dtype="float32",
+ name: str = "Buffer",
+ data=None,
+ strides=None,
+ elem_offset=None,
+ scope="global",
+ align=-1,
+ offset_factor=0,
+ buffer_type="default",
+ span=None,
+ ):
+ if strides is None:
+ strides = []
+ align = convert_to_int(align, "align", self.context.report_error,
self.node.span)
+ offset_factor = convert_to_int(
+ offset_factor, "offset_factor", self.context.report_error,
self.node.span
+ )
+ buffer = tvm.tir.decl_buffer(
+ shape,
+ dtype,
+ name,
+ data,
+ strides,
+ elem_offset,
+ scope,
+ align,
+ offset_factor,
+ buffer_type,
+ span=span,
+ )
+ return buffer
+
+ self.type = vtype
+ super().__init__(match_buffer_syntax_sugar, def_symbol=True)
+
+ @staticmethod
+ def __call__(
+ shape,
+ dtype: str = "float32",
+ name: str = "Buffer",
+ data=None,
+ strides=None,
+ elem_offset=None,
+ scope="global",
+ align=-1,
+ offset_factor=0,
+ buffer_type="default",
+ span=None,
+ ):
+ pass
+
+ @staticmethod
+ def __getitem__(args):
Review comment:
can you comment the usage of this one? how does it relate to `__call__`?
From the current usage like `T.Buffer[...]` I think this should be defined as
`__getitem__(self, args)` instead of static method as `Buffer` is an instance
of `GenericBufferType`
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
