Skip to content

Reduce the work to add a new expr: rewrite SubstituteInExpr with OptOutMutator #2202

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Nov 21, 2022
Merged

Reduce the work to add a new expr: rewrite SubstituteInExpr with OptOutMutator #2202

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
ee2f234
structured expr squash merge
zasdfgbnm Nov 17, 2022
24b1a52
save
zasdfgbnm Nov 18, 2022
39ccad3
Merge branch 'devel' of github.com:csarofeen/pytorch into mutator-cle…
zasdfgbnm Nov 18, 2022
c8244e6
fix
zasdfgbnm Nov 18, 2022
ee62179
fix
zasdfgbnm Nov 18, 2022
376dc88
fix
zasdfgbnm Nov 18, 2022
db9686d
no print
zasdfgbnm Nov 18, 2022
7d9c94e
save
zasdfgbnm Nov 18, 2022
305135f
save
zasdfgbnm Nov 18, 2022
e585dcb
rewrite SubstituteInExpr with OptOutMutator
zasdfgbnm Nov 19, 2022
0a7ba35
Merge branch 'devel' into SubstituteInExpr
zasdfgbnm Nov 19, 2022
a97b850
rename
zasdfgbnm Nov 21, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions torch/csrc/jit/codegen/cuda/dispatch.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -302,13 +302,13 @@ class TORCH_CUDA_CU_API OptOutMutator : public PolymorphicBase {
void registerMutation(Val* val, Val* mutation);

Val* maybeMutated(Val* val) {
if (mutations.find(val) == mutations.end()) {
if (mutations_.find(val) == mutations_.end()) {
return val;
}
return mutations.at(val);
return mutations_.at(val);
}

std::unordered_map<Val*, Val*> mutations;
std::unordered_map<Val*, Val*> mutations_;

//****Functions below defined in mutator.cpp*****

Expand All @@ -327,7 +327,8 @@ class TORCH_CUDA_CU_API OptOutMutator : public PolymorphicBase {
virtual void mutate(kir::TensorIndex*);

protected:
void removeExpr(IrContainer*, Expr*);
virtual void removeExpr(IrContainer*, Expr*) const;
virtual void registerNewExpr(Expr*) {}
};

} // namespace cuda
Expand Down
Loading