[not for merge] Baseline branch for hierarchical compilation

benchmarks/dynamo/common.py

@@ -1729,6 +1729,7 @@ class BenchmarkRunner:
                 self.optimizer = torch.optim.Adam(
                     params, lr=0.01, capturable=True, foreach=True
                 )
+                self.optimizer.step = torch._dynamo.disable(self.optimizer.step)
         else:
             self.optimizer = None