tree 9dbab5c86801d8a7127411e70ece7ec74475bd1b
parent 48ddc9762bb4cd6d53c760ce4e5dd6b291e15c9a
author Raghavan Raman <raghavanr@fb.com> 1618276263 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> 1618276344 -0700

[NNC] API to reorder multiple loops (#55568)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/52690

This PR adds the following APIs:

```
static bool areLoopsPerfectlyNested(const std::vector<For*>& loops);

static std::vector<For*> reorder(
      const std::vector<For*>& loops,
      const std::vector<size_t>& permutation);
```

The first API checks if the given list of loops are perfectly nested. The second API reorders the given list of loops according to the permutation specified.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/55568

Reviewed By: albanD

Differential Revision: D27689734

Pulled By: navahgar

fbshipit-source-id: dc1bffdbee068c3f401188035772b41847cbc7c6