Adding annotation for tir.allocate

* fix defaults Change-Id: I844d1fa28bce797a37da313d776138b4fdc00e0f
apache · Oct 1, 2021 · 506c2f2 · 506c2f2
1 parent 2d12303
commit 506c2f2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/tvm/tir/stmt.py b/python/tvm/tir/stmt.py
@@ -326,15 +326,17 @@ class Allocate(Stmt):
     """
 
     def __init__(self, buffer_var, dtype, extents, condition, body, annotations=None, span=None):
+        if annotations is None:
+            annotations = dict()
         self.__init_handle_by_constructor__(
-            _ffi_api.Allocate,
+            _ffi_api.Allocate,  # type: ignore
             buffer_var,
             dtype,
             extents,
             condition,
             body,
             annotations,
-            span,  # type: ignore
+            span,
         )