From 0e69c39908ce4b07e48891f61a17700d748d24cd Mon Sep 17 00:00:00 2001 From: Hongyi Jin Date: Mon, 11 Mar 2024 13:32:44 -0400 Subject: [PATCH] add fallback for outer reduction --- python/tvm/dlight/gpu/low_batch_gemv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/dlight/gpu/low_batch_gemv.py b/python/tvm/dlight/gpu/low_batch_gemv.py index 1c27fdfb133a..84a9319248c5 100644 --- a/python/tvm/dlight/gpu/low_batch_gemv.py +++ b/python/tvm/dlight/gpu/low_batch_gemv.py @@ -295,7 +295,7 @@ def apply( # pylint: disable=too-many-locals,too-many-branches,too-many-return- ) return sch else: - raise NotImplementedError("Outer reduction is not supported yet") + return None def sch_inner_reduction( # pylint: disable=too-many-arguments, invalid-name, unused-argument self,