pytorch
df0ae244 - [static runtime] Add out_ variant for aten::stack and aten::nan_to_num (#48150)

Commit
4 years ago
[static runtime] Add out_ variant for aten::stack and aten::nan_to_num (#48150) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48150 With D24767322, the remaining ops without out_ variant are pretty much the sparsenn specific ops, which are a bit trickier to add. Test Plan: ``` buck run //caffe2/test:static_runtime buck run //caffe2/benchmarks/static_runtime:static_runtime_cpptest buck run //caffe2/caffe2/fb/predictor:pytorch_predictor_test buck-out/opt/gen/caffe2/caffe2/fb/predictor/ptvsc2_predictor_bench \ --pred_net=/home/hlu/ads/adindexer/adindexer_ctr_mobilefeed/pt/merge/precomputation_merge_net.pb \ --c2_inputs=/home/hlu/ads/adindexer/adindexer_ctr_mobilefeed/pt/merge/c2_inputs_precomputation_bs1.pb \ --c2_weights=/home/hlu/ads/adindexer/adindexer_ctr_mobilefeed/pt/merge/c2_weights_precomputation.pb \ --scripted_model=/home/hlu/ads/adindexer/adindexer_ctr_mobilefeed/pt/merge/traced_precomputation_partial_dper_fixes.pt \ --pt_inputs=/home/hlu/ads/adindexer/adindexer_ctr_mobilefeed/pt/merge/container_precomputation_bs1.pt \ --iters=1 --warmup_iters=1 --num_threads=1 --pt_enable_static_runtime=true \ --pt_cleanup_activations=true --pt_enable_out_variant=true \ --eps 1e-2 ``` Reviewed By: bwasti Differential Revision: D25016076 fbshipit-source-id: 59a7948d4cca60182b6755217571128c2fc51f4d
Author
Hao Lu
Parents
Loading