Logger integration format (#67962)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67962
Logger integration format for chunks at [dims] -> input_val.shape[dim]
NOTE: Unused typing imports removed
Test Plan:
buck run -c python.package_style=inplace mode/dev-nosan caffe2/torch/fb/fx2trt:test_chunk
out:
RuntimeWarning: Asked for 2000 chunks along dimention 2 on tensor with size (3, 10, 20), chunks will default to 20
Reviewed By: 842974287
Differential Revision: D32233039
fbshipit-source-id: 1fde12c9f743bb80cdb309e0b7be287173d45147