github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-21 17:40:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fp8] update torch.compile for linear_fp8 to >= 2.4.0 (#6004)

Browse Source
This commit is contained in:
botbw
2024-08-15 10:14:42 +08:00
committed by GitHub
parent 597b206001
commit 88fa096d78
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/quantization/fp8.py
View File

@@ -7,7 +7,7 @@ import torch.nn.functional as F
from packaging.version import Version from packaging.version import Version
from torch.distributed import ReduceOp from torch.distributed import ReduceOp
SUPPORT_TORCH_COMPILE = Version(torch.__version__) >= Version("2.3.0") SUPPORT_TORCH_COMPILE = Version(torch.__version__) >= Version("2.4.0")
class Handle: class Handle: