Update

python/nano/src/bigdl/nano/pytorch/strategies/ipex/ipex_strategy.py

@@ -94,11 +94,14 @@ def optimizer_step(self,
                        closure: Callable[[], Any],
                        **kwargs: Any) -> Any:
         """Hook to run the optimizer step."""
+        if type(optimizer) in IPEX_FUSED_OPTIMIZER_LIST:
+            return super().optimizer_step(model, optimizer, optimizer_idx, closure, **kwargs)
+
         if isinstance(model, pl.LightningModule):
             closure = partial(self._wrap_closure, model, optimizer, optimizer_idx, closure)
+
+        closure_result = closure()
+        optimizer.step(closure=None, **kwargs)
+
+        return closure_result
 
-        # Automatically call closure for optimizer not supported by IPEX
-        if type(optimizer) not in IPEX_FUSED_OPTIMIZER_LIST:
-            closure()
-
-        return optimizer.step(closure, **kwargs)