self_*_memory should preceed *_memory

awwong1 · Dec 12, 2020 · 257c282 · 257c282
1 parent f522ae6
commit 257c282
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchprof/display.py b/torchprof/display.py
@@ -9,10 +9,10 @@
         "cpu_total",
         "self_cuda_total",
         "cuda_total",
-        "cpu_memory",
         "self_cpu_memory",
-        "cuda_memory",
+        "cpu_memory",
         "self_cuda_memory",
+        "cuda_memory",
         "occurrences",
     ],
 )