Update

[ghstack-poisoned]
pytorch · Oct 17, 2024 · 5239f2e · 5239f2e
1 parent a9dbfb3
commit 5239f2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/common/h2d_test.py b/benchmarks/common/h2d_test.py
@@ -97,7 +97,7 @@ def test_to_njt(
         self, benchmark, consolidated, njt_td, default_device, compile_mode, num_threads
     ):
         tensordict_logger.info(
-            f"njtd size {td.bytes() / 1024 / 1024 :.2f} Mb"
+            f"njtd size {njtd.bytes() / 1024 / 1024 :.2f} Mb"
         )
         if consolidated:
             njt_td = njt_td.consolidate()