From 0427373571fc5d3c253dfcd26d403fbce6ef0626 Mon Sep 17 00:00:00 2001
From: mgam <312065559@qq.com>
Date: Sat, 1 Nov 2025 14:33:32 +0800
Subject: [PATCH 1/3] Fix the deprecated torch.cuda.amp module

---
 mmengine/optim/optimizer/amp_optimizer_wrapper.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mmengine/optim/optimizer/amp_optimizer_wrapper.py b/mmengine/optim/optimizer/amp_optimizer_wrapper.py
index 4f3323f2cc..0534627991 100644
--- a/mmengine/optim/optimizer/amp_optimizer_wrapper.py
+++ b/mmengine/optim/optimizer/amp_optimizer_wrapper.py
@@ -1,6 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from contextlib import contextmanager
 from typing import Union
+from functools import partial
 
 import torch
 import torch.nn as nn
@@ -17,7 +18,8 @@
 elif is_mlu_available():
     from torch.mlu.amp import GradScaler
 else:
-    from torch.cuda.amp import GradScaler
+    from torch.amp import GradScaler
+    GradScaler = partial(GradScaler, device='cuda')
 
 
 @OPTIM_WRAPPERS.register_module()

From 47e5298255bdc67e1b86aa942738d75e32c82f57 Mon Sep 17 00:00:00 2001
From: mgam <312065559@qq.com>
Date: Sat, 1 Nov 2025 15:21:42 +0800
Subject: [PATCH 2/3] Fix test according to the latest GradScaler definition.

---
 tests/test_optim/test_optimizer/test_optimizer_wrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_optim/test_optimizer/test_optimizer_wrapper.py b/tests/test_optim/test_optimizer/test_optimizer_wrapper.py
index ef1db241dd..ba4ca77d11 100644
--- a/tests/test_optim/test_optimizer/test_optimizer_wrapper.py
+++ b/tests/test_optim/test_optimizer/test_optimizer_wrapper.py
@@ -8,7 +8,7 @@
 import torch.distributed as torch_dist
 import torch.nn as nn
 from parameterized import parameterized
-from torch.cuda.amp import GradScaler
+from torch.amp import GradScaler
 from torch.nn.parallel.distributed import DistributedDataParallel
 from torch.optim import SGD, Adam, Optimizer
 

From 5e3172604c65cca2d0d0737ecc7dd774938797c0 Mon Sep 17 00:00:00 2001
From: mgam <312065559@qq.com>
Date: Tue, 4 Nov 2025 19:54:41 +0800
Subject: [PATCH 3/3] Fix lint

---
 mmengine/optim/optimizer/amp_optimizer_wrapper.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmengine/optim/optimizer/amp_optimizer_wrapper.py b/mmengine/optim/optimizer/amp_optimizer_wrapper.py
index 0534627991..b60f3d09b8 100644
--- a/mmengine/optim/optimizer/amp_optimizer_wrapper.py
+++ b/mmengine/optim/optimizer/amp_optimizer_wrapper.py
@@ -1,7 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 from contextlib import contextmanager
-from typing import Union
 from functools import partial
+from typing import Union
 
 import torch
 import torch.nn as nn