Skip to content

GitLab

Unverified Commit 3dbbf83f authored by flybird11111's avatar flybird11111 Committed by GitHub
Browse files

fix (#5158) 

fix
parent 368b5e3d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/tensor/colo_parameter.py
View file @ 3dbbf83f
...@@ -7,7 +7,7 @@ from colossalai.tensor.param_op_hook import ColoParamOpHookManager ...@@ -7,7 +7,7 @@ from colossalai.tensor.param_op_hook import ColoParamOpHookManager
from .colo_tensor import _convert_output from .colo_tensor import _convert_output
WHITE_LIST_FUNCS = {torch.Tensor.__getitem__} WHITE_LIST_FUNCS = {torch.Tensor.__getitem__, torch.Tensor.is_floating_point}
def is_no_hook_op(func) -> bool: def is_no_hook_op(func) -> bool:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment