diff --git a/python/tvm/script/intrin.py b/python/tvm/script/intrin.py index 2ca11ff30b4b..76ddbb1de697 100644 --- 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)