pytorch
c5dae335 - [PT][StaticRuntime] Move prim op impl to ops.cpp (#48210)

Commit
4 years ago
[PT][StaticRuntime] Move prim op impl to ops.cpp (#48210) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48210 - Move prim op implementation from `ProcessedNode::run` to `getNativeOperation` - Add out variant for `prim::listConstruct` Test Plan: ``` buck test //caffe2/test:static_runtime buck test //caffe2/benchmarks/static_runtime:static_runtime_cpptest buck test //caffe2/caffe2/fb/predictor:pytorch_predictor_test buck run mode/dev //caffe2/caffe2/fb/predictor:ptvsc2_predictor_bench -- \ --scripted_model=/home/hlu/ads/adindexer/adindexer_ctr_mobilefeed/pt/merge/traced_precomputation.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 ``` Reviewed By: ajyu Differential Revision: D24748947 fbshipit-source-id: 12caeeae87b69e60505a6cea31786bd96f5c8684
Author
Hao Lu
Parents
Loading