diff --git a/PyTorchSimFrontend/mlir/mlir_codegen_backend.py b/PyTorchSimFrontend/mlir/mlir_codegen_backend.py
index 9c20311c..b163ad1a 100644
--- a/PyTorchSimFrontend/mlir/mlir_codegen_backend.py
+++ b/PyTorchSimFrontend/mlir/mlir_codegen_backend.py
@@ -906,9 +906,6 @@ def codegen_loops(self):
         if (self.reduction_depth==0):
             loops = LoopNest([LoopLevel("dummy", 1)])
 
-        if len(reductions.loops) > 1:
-            raise NotImplementedError("Not support multiple reduction axis..")
-
         code.splice(self.const_buffer)
         code.splice(self.alloc_buffer)
         code.splice(self.spad_buffer)