From 8abc87798f3ea728c9e03bef6b5ceea4b5c74ff8 Mon Sep 17 00:00:00 2001
From: digger yu <digger-yu@outlook.com>
Date: Mon, 3 Jul 2023 17:10:18 +0800
Subject: [PATCH] fix Tensor is not defined (#4129)

---
 colossalai/kernel/jit/bias_dropout_add.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/colossalai/kernel/jit/bias_dropout_add.py b/colossalai/kernel/jit/bias_dropout_add.py
index 3687dde79..32965c1eb 100644
--- a/colossalai/kernel/jit/bias_dropout_add.py
+++ b/colossalai/kernel/jit/bias_dropout_add.py
@@ -1,4 +1,5 @@
 import torch
+from torch import Tensor
 
 
 def bias_dropout_add(x, bias, residual, prob, training):