fix the cache miss rate reporting.

dmlc · Aug 21, 2024 · f9e634f · f9e634f
1 parent 26841f3
commit f9e634f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -465,9 +465,9 @@ def main():
             args.cpu_feature_cache_policy,
             args.feature_device == "pinned",
         )
-        for k, feature in cached_features.items():
-            features[k] = feature
-            cpu_cache_miss_rate_fn = lambda: feature.miss_rate
+        for k, cpu_cached_feature in cached_features.items():
+            features[k] = cpu_cached_feature
+            cpu_cache_miss_rate_fn = lambda: cpu_cached_feature.miss_rate
     else:
         cpu_cache_miss_rate_fn = lambda: 1
 
@@ -477,9 +477,9 @@ def main():
             feats,
             int(args.gpu_cache_size * (2**30)),
         )
-        for k, feature in cached_features.items():
-            features[k] = feature
-            gpu_cache_miss_rate_fn = lambda: feature.miss_rate
+        for k, gpu_cached_feature in cached_features.items():
+            features[k] = gpu_cached_feature
+            gpu_cache_miss_rate_fn = lambda: gpu_cached_feature.miss_rate
     else:
         gpu_cache_miss_rate_fn = lambda: 1