expose option to collect sizes as dynamic (#141153)

Summary:
X-link: pytorch/torchrec#2594

This is to address recompiles from eager nodes that saved dynamic activations

X-link: pytorch/pytorch#141153
Approved by: https://github.com/jansel
ghstack dependencies: #141152

Reviewed By: izaitsevfb

Differential Revision: D66384254

Pulled By: xmfan

fbshipit-source-id: 89d439353219252c9f5be0b44cf67957092860e3

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -3369,7 +3369,7 @@ def inner_compiler(gm_, example_inputs_):
                 gm, backend=inner_compiler, fullgraph=fullgraph, dynamic=dynamic
             )
 
-        with torch._dynamo.compiled_autograd.enable(compiler_fn) as ctx:
+        with torch._dynamo.compiled_autograd._enable(compiler_fn) as ctx:
             yield ctx