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,