diff --git a/torch_xla/csrc/runtime/cache.h b/torch_xla/csrc/runtime/cache.h index bef5b099ec6..9557b2353b7 100644 --- a/torch_xla/csrc/runtime/cache.h +++ b/torch_xla/csrc/runtime/cache.h @@ -173,6 +173,7 @@ class PersistentCache : public AbstractCache { TORCH_LAZY_COUNTER("PersistentCacheMiss", 1); return nullptr; } + TORCH_LAZY_TIMED("PersistentCacheLoad"); std::stringstream ss; std::ifstream in(path, std::ios::binary); ss << in.rdbuf();