add a metrics to track persistent cache loading time (pytorch#7039)

ManfeiBai · May 9, 2024 · c1b745e · c1b745e
1 parent 5bbe5c8
commit c1b745e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_xla/csrc/runtime/cache.h b/torch_xla/csrc/runtime/cache.h
@@ -173,6 +173,7 @@ class PersistentCache : public AbstractCache<K, T, H, E> {
       TORCH_LAZY_COUNTER("PersistentCacheMiss", 1);
       return nullptr;
     }
+    TORCH_LAZY_TIMED("PersistentCacheLoad");
     std::stringstream ss;
     std::ifstream in(path, std::ios::binary);
     ss << in.rdbuf();