[TVMSCRIPT] Add tir.min node in tvm script (apache#8219)

Co-authored-by: honghua.cao <[email protected]>
u99127 · Jun 9, 2021 · 55459e7 · 55459e7
1 parent 9899f1e
commit 55459e7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/tvm/script/intrin.py b/python/tvm/script/intrin.py
@@ -151,6 +151,11 @@ def max(a, b, span):  # pylint: disable=redefined-builtin
     return tvm.tir.Max(a, b, span)
 
 
+@register
+def min(a, b, span):  # pylint: disable=redefined-builtin
+    return tvm.tir.Min(a, b, span)
+
+
 def get_axis(begin, end, iter_type, span):
     ana = tvm.arith.Analyzer()
     extent = ana.simplify(end - begin)