1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#pragma once
#include <iostream>
#include <torch/csrc/jit/tensorexpr/fwd_decls.h>
#include <torch/csrc/jit/tensorexpr/ir_visitor.h>
namespace torch {
namespace jit {
namespace tensorexpr {
class Expr;
class ExprHandle;
class Mod;
class And;
class Or;
class Xor;
class Lshift;
class Rshift;
class CompareSelect;
class Ramp;
class Load;
class IfThenElse;
class Intrinsics;
class Stmt;
class ExternalCall;
class Store;
class For;
class Block;
class TORCH_API IRVerifier : public IRVisitor {
public:
IRVerifier() = default;
void visit(ModPtr v) override;
void visit(AndPtr v) override;
void visit(OrPtr v) override;
void visit(XorPtr v) override;
void visit(LshiftPtr v) override;
void visit(RshiftPtr v) override;
void visit(CompareSelectPtr v) override;
void visit(RampPtr v) override;
void visit(LoadPtr v) override;
void visit(IfThenElsePtr v) override;
void visit(IntrinsicsPtr v) override;
void visit(ExternalCallPtr v) override;
void visit(StorePtr v) override;
void visit(ForPtr v) override;
void visit(BlockPtr v) override;
};
TORCH_API void verify(StmtPtr);
TORCH_API void verify(ExprPtr);
TORCH_API void verify(ExprHandle);
} // namespace tensorexpr
} // namespace jit
} // namespace torch
|
