1#pragma once
2
3#include <c10/macros/Export.h>
4
5#include <dispatch.h>
6#include <fusion.h>
7#include <ir_all_nodes.h>
8#include <lower_trivial_reductions.h>
9
10#include <vector>
11
12namespace torch {
13namespace jit {
14namespace fuser {
15namespace cuda {
16
17// Replaces trivial reductions with Unary Set Ops
18void trivialReductionReplacement(Fusion*, const TrivialReductionInfo&);
19
20// Transpose, Shift, Gather, and View Ops with Unary Set Ops
21std::vector<Expr*> unarySetOpInserter(const std::vector<Expr*>& exprs);
22
23} // namespace cuda
24} // namespace fuser
25} // namespace jit
26} // namespace torch
27