1 | #pragma once |
---|---|
2 | #include <c10/macros/Export.h> |
3 | |
4 | #include <ir_all_nodes.h> |
5 | #include <kernel_ir.h> |
6 | |
7 | #include <vector> |
8 | |
9 | namespace torch { |
10 | namespace jit { |
11 | namespace fuser { |
12 | namespace cuda { |
13 | |
14 | //! Update predicates with valid bool conditionals |
15 | //! |
16 | std::vector<Expr*> generateConditionalFromPredicate( |
17 | const std::vector<Expr*>& exprs); |
18 | |
19 | } // namespace cuda |
20 | } // namespace fuser |
21 | } // namespace jit |
22 | } // namespace torch |
23 |