Add relative directory import for checkpoint

Author: shangerxin

test/xpu/test_utils.py

@@ -18,7 +18,7 @@
 import torch.nn as nn
 import torch.utils.cpp_extension
 import torch.utils.data
-from checkpoint import (
+from .checkpoint import (
     _infer_device_type,
     checkpoint,
     checkpoint_sequential,