mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
4f163df41a
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/42286 One more bug to fix. Operators such as If and AsyncIf need special treatment not just in `onnx::SsaRewrite`, but also in `RemoveOpsByType`. The solution needs two steps: 1) add external inputs/outputs of the subnets of If/AsyncIf op to the inputs/outputs of the op 2) if the inputs/outputs of the If/AsyncIf op need to be renamed as a result, the same inputs/outputs of the subnets need to be renamed as well. I also added unit tests to cover this corner case. Test Plan: ``` buck test //caffe2/caffe2/fb/predictor:black_box_predictor_test mkdir /tmp/models rm -rf /tmp/$USER/snntest rm -rf /tmp/snntest buck run mode/opt admarket/lib/ranking/prediction_replayer/snntest_replayer_test/tools:snntest_replay_test -- --serving_paradigm=USER_AD_PRECOMPUTATION_DSNN ``` Differential Revision: D22834028 fbshipit-source-id: c070707316cac694f452a96e5c80255abf4014bc |
||
|---|---|---|
| .. | ||
| emulator | ||
| CMakeLists.txt | ||
| InferenceGraph.h | ||
| predictor_config.cc | ||
| predictor_config.h | ||
| predictor_test.cc | ||
| predictor_utils.cc | ||
| predictor_utils.h | ||
| predictor.cc | ||
| predictor.h | ||
| ThreadLocalPtr.cc | ||
| ThreadLocalPtr.h | ||
| transforms.cc | ||
| transforms.h | ||