[Static Runtime] Moved NNC operator definitions to separate files. (#63838)
authorRaghavan Raman <raghavanr@fb.com>
Wed, 25 Aug 2021 18:12:57 +0000 (11:12 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 25 Aug 2021 18:18:32 +0000 (11:18 -0700)
commita2399a76e18b31747019ead8f80e2a4eb53e8223
tree69178babc0f36decd116721a151da6076a48a158
parent8a22d4fa5c5953bcb83293b5349b5f79ae08f193
[Static Runtime] Moved NNC operator definitions to separate files. (#63838)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63838

Refactored NNC operator definitions code into separate files.

Made `TEWrapper` a class with a fixed set of methods and added separate definitions for them based on `TORCH_ENABLE_LLVM` to keep the same functionality as before.

Test Plan: Build and ran Static Runtime tests.

Reviewed By: hlu1

Differential Revision: D30405467

fbshipit-source-id: 606ef852bb820d5e23a0f8af1bf5dc122e90bceb
tools/build_variables.bzl
torch/csrc/jit/runtime/static/ops.cpp
torch/csrc/jit/runtime/static/te_wrapper.cpp [new file with mode: 0644]
torch/csrc/jit/runtime/static/te_wrapper.h [new file with mode: 0644]