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

	class Allocate(WithScopeHandler):
	"""With scope handler T.allocate(extents, dtype, scope, condition)"""
	def __init__(self):
	def allocate(extents, dtype, scope, condition=True, span=None):
	condition = tvm.runtime.convert(condition)
	scope = tvm.runtime.convert(scope)
	return tvm.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.