[test] fixed the triton version for testing (#2608)

2025-09-12 12:47:21 +00:00 · 2023-02-07 13:49:38 +08:00
parent aa7e9e4794
commit 8518263b80
3 changed files with 3 additions and 1 deletions
--- a/colossalai/amp/apex_amp/apex_amp.py
+++ b/colossalai/amp/apex_amp/apex_amp.py
@@ -2,6 +2,7 @@
 # -*- encoding: utf-8 -*-

 import torch.nn as nn
+
 try:
    import apex.amp as apex_amp
 except ImportError: