From f5ca88ec972f4c3c76e8aadaacce7a078d1f7116 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Mon, 14 Feb 2022 18:04:57 +0800
Subject: [PATCH] fixed apex import (#227)

---
 colossalai/amp/apex_amp/apex_amp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/amp/apex_amp/apex_amp.py b/colossalai/amp/apex_amp/apex_amp.py
index b43228410088..6d7196b33466 100644
--- a/colossalai/amp/apex_amp/apex_amp.py
+++ b/colossalai/amp/apex_amp/apex_amp.py
@@ -5,7 +5,7 @@
 try:
     import apex.amp as apex_amp
 except ImportError:
-    raise ImportError('Cannot import apex.amp correctly.')
+    pass
 
 from torch import Tensor