This repository has been archived by the owner on Aug 16, 2024. It is now read-only.
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[TensorExpr] Move 'compute*' functions to operators/... (pytorch#65552)
Summary: Pull Request resolved: pytorch#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
- Loading branch information