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

Co-authored-by: honghua.cao <[email protected]>
trevor-m · Jun 17, 2021 · 542f0d3 · 542f0d3
1 parent 7e4928e
commit 542f0d3
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)