pytorch
6c0521b9 - [SR] Add native implementations for converted prim ops (#71474)

Commit
2 years ago
[SR] Add native implementations for converted prim ops (#71474) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71474 The PyTorch edge team is working on promoting some prim ops to interpreter instructions (see D33398092). Since the JIT fallback ops will be unavailable soon, we need to implement these ops in static runtime. Ops not included in this diff: * `aten::__is__` and `aten::__isnot__`: disabled in static runtime for unrelated reasons * `prim::NumToTensor` and `aten::__get__.Dict` already exist ghstack-source-id: 148641179 Test Plan: `buck test caffe2/benchmarks/static_runtime:static_runtime_cpptest` Reviewed By: d1jang Differential Revision: D33657816 fbshipit-source-id: 6d15244ae1024a56d3b25e51a433fa104ce8ee5e (cherry picked from commit 33f8f861ff88a6dda6a545c12515e92c893027d4)
Author
Mike Iovine
Committer
Parents
Loading