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
10namespace torch {
11namespace jit {
12namespace fuser {
13namespace 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