1 | #pragma once |
2 | |
3 | #include <c10/macros/Export.h> |
4 | |
5 | #include <dispatch.h> |
6 | #include <ir_base_nodes.h> |
7 | |
8 | #include <unordered_map> |
9 | |
10 | namespace torch { |
11 | namespace jit { |
12 | namespace fuser { |
13 | namespace cuda { |
14 | |
15 | /* |
16 | * Mutators are the mechanism used to modify IR nodes. Since most nodes are |
17 | * immutable or at least partially immutable changeing them can require creating |
18 | * a new node. Base mutator at the moment is a dumb sample mutator that takes |
19 | * any float of value 1.0 and converts it to 0.0; It is currently used as a |
20 | * dummy example, however, we should make it a simple instantiation of all the |
21 | * mutate functions on all node types so that people can inherit it, and only |
22 | * specialize those nodes which they want to have a particular transformation. |
23 | */ |
24 | |
25 | } // namespace cuda |
26 | } // namespace fuser |
27 | } // namespace jit |
28 | } // namespace torch |
29 | |