Code

[WIP] Indexing Refactor#2316

Code

Draft

csarofeen wants to merge 36 commits intodevelcsarofeen/pytorch:develfrom indexing_refactoringcsarofeen/pytorch:indexing_refactoringCopy head branch name to clipboard

+5,148-1,914

Commits

Commits on Jan 17, 2023

• Start reducing down IdGraph.

  csarofeen

  committed
  7e43d1d

• Join the different sets into one structure based on MappingMode.

  csarofeen

  committed
  7bbf730

• Small alias.

  csarofeen

  committed
  5581f6e

• More minor refactoring.

  csarofeen

  committed
  1443fcc

• Code movement. Split up IdGraph build process.

  csarofeen

  committed
  08b35c3

• Rename node -> disjoint sets.

  csarofeen

  committed
  a6d824e

• Make IterDomainGraph more self contained with a recursive ID mapping approach.

  csarofeen

  committed
  fb13a0c

• Expand comment for BestEffortReplay.

  csarofeen

  committed
  ada2af5

• Minor interface changes in compute at map, code movement.

  csarofeen

  committed
  4e9d268

• Expose forwarding info for permissive mapping.

  csarofeen

  committed
  ee50ddd

• Make exact and permissive maps self building.

  csarofeen

  committed
  efda269

• Remove BestEffortReplay use from Loop Map construction. Remove consumer/producer map from IterDomainGraph.

  csarofeen

  committed
  fe08382

• Small name refactor, add definition and uses API to IterDomainGraph.

  csarofeen

  committed
  37e81b3

• Remove ComputeAtMap's definition and uses in favor of IterDomainGraphs.

  csarofeen

  committed
  bf202a8

• Remove multi output function in IdGraph building, remove explicit rfactor mapping since it's already implicitly done.

  csarofeen

  committed
  d787043

• IdGraph/ComputeAt Interface Tuning and cleanup.

  csarofeen

  committed
  0321dba

• Minor IdGraph tweaks, continue removing some BestEffortReplay usage.

  csarofeen

  committed
  c29b427

• Merge getMatchedLeafPosWithoutReplay[CasP, PasC] into ...TasR.

  csarofeen

  committed
  6f28803

• Remove fullSelfMatching

  csarofeen

  committed
  23c05ad

• Treat braodcast as the exception not the rule for unskippable inlined dimensions.

  csarofeen

  committed
  a390667

• Minor build fixes.

  csarofeen

  committed
  3b93604

• Minor cleanup.

  csarofeen

  committed
  9511d76

• Draft loop promotion in compute at map, validate with warning initial approach.

  csarofeen

  committed
  14c9844

• Minor loop promotion map fix.

  csarofeen

  committed
  af28d38

Commits on Jan 20, 2023

• minor build fix.

  csarofeen

  committed
  b67e245

Commits on Jan 21, 2023

• Minor refactoring.

  csarofeen

  committed
  4e55bd1

• Some more minor refactoring.

  csarofeen

  committed
  e38e8f7

Commits on Jan 26, 2023

• Update iter domain graph with broadcast promotion logic. WARNING Breaks everything.

  csarofeen

  committed
  0e6a85a

Commits on Jan 27, 2023

• Reduce verbosity, add name only option to lower dump.

  csarofeen

  committed
  33d0e04

• Minor loop promote fix.

  csarofeen

  committed
  6651752

Commits on Jan 28, 2023

• Minor cleanup.

  csarofeen

  committed
  b465a10

• Disable/WAR Gather/Shift for now.

  csarofeen

  committed
  c9e8710

Commits on Feb 6, 2023

• Add id_definitions_, fix idGraph construction.

  csarofeen

  committed
  bd86f5c

Commits on Feb 12, 2023

• Stash current iter domain graph attempt.

  csarofeen

  committed
  33f5dcc

Commits on Feb 21, 2023

• Another draft of loop promotion and introduction of index map.

  csarofeen

  committed
  981a14a

Commits on Feb 25, 2023

• Factor out IdGraph from multiple IterDomainGraphs.

  csarofeen

  committed
  e3cbe70