pytorch
015e0079 - [TensorExpr] Move 'compute*' functions to operators/... (#65552)

Commit
4 years ago
[TensorExpr] Move 'compute*' functions to operators/... (#65552) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65552 This PR is mostly a verbatim move of several functions to different files. The goal is to have more consistency in what resides where. With this PR: * All `compute*` functions defining how a given operator needs to be lowered to TE IR will reside in `operators/*.{cpp,h}`. * Auxiliary functions for these functions will reside in `operators/misc.cpp`. `compute*` functions for ops not belonging anywhere else can also go to that file. * `operators/unary.*` is renamed to `operators/pointwise.*` and now includes functions like `computeTwoOperands`. * `kernel.*` now contains *only JIT-related* logic and implementations of `TensorExprKernel` methods. Test Plan: Imported from OSS Reviewed By: navahgar Differential Revision: D31148923 Pulled By: ZolotukhinM fbshipit-source-id: e36ad8e779b8d30a33b49ea4ebf6d6a7438989f4
Author
Mikhail Zolotukhin
Parents
Loading