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

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

pull/6005/head
botbw 2024-08-15 10:14:42 +08:00 committed by GitHub
parent 597b206001
commit 88fa096d78
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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 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: