diff --git a/platforms/patches/kunlun_p800/sitecustomize_xpu.py b/platforms/patches/kunlun_p800/sitecustomize_xpu.py index b8d8756..079ba8a 100644 --- a/platforms/patches/kunlun_p800/sitecustomize_xpu.py +++ b/platforms/patches/kunlun_p800/sitecustomize_xpu.py @@ -5,3 +5,6 @@ import torch # them on P800 via its own FP8 kernels. if not hasattr(torch, "float8_e8m0fnu"): torch.float8_e8m0fnu = torch.uint8 + +# P800 Timing module +import p800_timing