github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
pull/5885/head
pre-commit-ci[bot] 2024-07-04 12:47:16 +00:00
parent 6991819a97
commit e17f835df7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/quantization/fp8.py
View File

@ -69,7 +69,7 @@ def all_reduce_fp8(tensor: torch.Tensor, fp8_format="e4m3") -> None:
"""
world_size = dist.get_world_size()
rank = dist.get_rank()
dist.get_rank()
input_type = tensor.dtype
input_shape = tensor.shape
input_device = tensor.device