pytorch
7ae45403 - [static runtime] support aten::__getitem__ natively (#55310)

Commit
3 years ago
[static runtime] support aten::__getitem__ natively (#55310) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55310 Test Plan: Run on the dper generated local/local_ro model ``` ./buck-out/opt/gen/caffe2/caffe2/fb/predictor/ptvsc2_predictor_bench --scripted_model=/data/users/ansha/tmp/adfinder/aug_1x/210616848_0.predictor.disagg.local.local_ro.pt --pt_inputs=/data/users/ansha/tmp/adfinder/aug_1x/210616848_0.predictor.disagg.input_data.container.pt --iters=1000 --warmup_iters=10000 --num_threads=1 --pt_enable_static_runtime=1 --pt_cleanup_activations=true --pt_enable_out_variant=1 --pt_optimize_memory=1 --compare_results=0 --do_profile=0 --adsfinder_compatibility=1 ``` Reviewed By: hlu1 Differential Revision: D27569662 fbshipit-source-id: df68c2fdd95e39a30aec35ddbaf1f5df0bc3a3da
Author
Parents
  • torch/csrc/jit/runtime/static
    • File
      ops.cpp