Adding annotations for tir.allocate

[manupak]

This commit is adding annotations for tir.allocate node to be used as hints for future transformations.

Please refer to the discussion : apache/tvm-rfcs#23

cc: @tqchen@junrushao1994@areusch

[Mousius]

Makes sense to me 😸

[areusch]

please tag the tracking issue @manupa-arm

[areusch]

deferring to @vinx13 for approval but lgtm

[vinx13]

Some minor comments, otherwise LGTM

[manupak]

Thanks @vinx13 .
I have addressed them now, PTAL.

[Hzfengsy]

Please also update the Script syntax and add a testcase.

The related codes are here:

tvm/python/tvm/script/tir/scope_handler.py

Lines 106 to 115 in d9a5ff5

	classAllocate(WithScopeHandler):
	"""With scope handler T.allocate(extents, dtype, scope, condition)"""
	def__init__(self):
	defallocate(extents, dtype, scope, condition=True, span=None):
	condition=tvm.runtime.convert(condition)
	scope=tvm.runtime.convert(scope)
	returntvm.tir.Allocate(
	self.buffer_var, dtype, extents, condition, self.body, span=span
	)

[Hzfengsy]

Please also update the three printers:

tvm/src/tir/ir/stmt.cc

Line 385 in d9a5ff5

.set_dispatch<AllocateNode>([](const ObjectRef& node, ReprPrinter* p) {

tvm/src/printer/tvmscript_printer.cc

Line 762 in d9a5ff5

Doc TVMScriptPrinter::VisitStmt_(const AllocateNode* op) {

tvm/src/printer/tir_text_printer.cc

Line 449 in d9a5ff5

Doc TIRTextPrinter::VisitStmt_(const AllocateNode* op) {

[manupak]

@Hzfengsy thanks for the review!

I think I've done the necessary changes now. PTAL.

[Hzfengsy]

LGTM. Thanks, @manupa-arm

[manupak]

@vinx13@Hzfengsy merge?

[Hzfengsy]

Sorry for the late response. Thanks @manupa-arm.