From 55459e74dd241a573dcdcb69f3385a3fbd8bb31a Mon Sep 17 00:00:00 2001 From: Honghua Cao <49267856+Beya2019@users.noreply.github.com> Date: Wed, 9 Jun 2021 21:52:28 +0800 Subject: [PATCH] [TVMSCRIPT] Add tir.min node in tvm script (#8219) Co-authored-by: honghua.cao --- python/tvm/script/intrin.py | 5 +++++ 1 file changed, 5 insertions(+) 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)