Update pytorch_checkpointing.py [ci-skip]

argonne-lcf · Jul 9, 2024 · 10ffaa7 · 10ffaa7
1 parent 8d4f473
commit 10ffaa7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlio_benchmark/plugins/experimental/src/checkpoint/pytorch_checkpointing.py b/dlio_benchmark/plugins/experimental/src/checkpoint/pytorch_checkpointing.py
@@ -18,7 +18,7 @@
 import torch
 
 from dlio_benchmark.checkpointing.base_checkpointing import BaseCheckpointing
-from dftracer.logger import fn_interceptor as Profile
+from dlio_benchmark.utils.utility import Profile
 
 from dlio_benchmark.common.constants import MODULE_CHECKPOINT
 from dlio_benchmark.common.enumerations import CheckpointLocationType