Fix for KeyError: 'Size' error (#194)

Author: jansel

torchbench.py

@@ -1113,6 +1113,8 @@ def run_one_model(
     cos_similarity=False,
     skip_accuracy_check=False,
 ):
+    t0 = time.perf_counter()
+
     tolerance = 1e-4
     # Increase the tolerance for torch allclose
     if is_training and current_device == "cuda" and name in REQUIRE_HIGHER_TOLERANCE:
@@ -1170,7 +1172,9 @@ def run_one_model(
             frames_third_pass = 0
 
         if output_filename and "coverage" in output_filename:
-            results.append(f"{ok:3}/{total:3} +{frames_third_pass} frames")
+            results.append(
+                f"{ok:3}/{total:3} +{frames_third_pass} frames {time.perf_counter()-t0:.0f}s"
+            )
 
         results.append(experiment(model, example_inputs))
         print(" ".join(map(str, results)))

torchdynamo/variables/lists.py

@@ -236,15 +236,11 @@ def python_type(self):
         return torch.Size
 
     def reconstruct(self, codegen):
-        load_torch_size = [
-            create_instruction("LOAD_GLOBAL", "torch"),
-            create_instruction("LOAD_METHOD", "Size"),
-        ]
-        codegen.extend_output(load_torch_size)
+        codegen.load_import_from("torch", "Size")
         codegen.foreach(self.items)
         build_torch_size = [
             create_instruction("BUILD_TUPLE", len(self.items)),
-            create_instruction("CALL_METHOD", 1),
+            create_instruction("CALL_FUNCTION", 1),
         ]
         return build_torch_size