github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-10 04:18:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
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() world_size = dist.get_world_size()
rank = dist.get_rank() dist.get_rank()
input_type = tensor.dtype input_type = tensor.dtype
input_shape = tensor.shape input_shape = tensor.shape
input_device = tensor.device input_device = tensor.device