Home
last modified time | relevance | path

Searched refs:tir (Results 1 – 25 of 1690) sorted by relevance

12345678910>>...68

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/unittest/
H A Dtest_tvmscript_roundtrip.py23 @tvm.script.tir
82 @tvm.script.tir
112 tir.broadcast(tir.float32(0), 32),
335 assert tir.uint64(0) == tir.tvm_struct_get(
364 assert tir.uint64(0) == tir.tvm_struct_get(
396 assert tir.uint64(0) == tir.tvm_struct_get(
2514 assert tir.uint64(0) == tir.tvm_struct_get(
2563 assert tir.uint64(0) == tir.tvm_struct_get(
2615 assert tir.uint64(0) == tir.tvm_struct_get(
2624 tir.evaluate(tir.tvm_struct_set(stack_value, 0, 12, tir.cast(2, "int64"), dtype="int32"))
[all …]
H A Dtest_tir_constructor.py45 x = tvm.tir.Cast("float32", tvm.tir.IntImm("uint32", 1))
54 tvm.tir.Add,
55 tvm.tir.Sub,
61 tvm.tir.LT,
62 tvm.tir.LE,
63 tvm.tir.GT,
64 tvm.tir.GE,
74 for cls in [tvm.tir.And, tvm.tir.Or]:
131 x = tvm.tir.LetStmt(v, 1, tvm.tir.Evaluate(1))
137 x = tvm.tir.AttrStmt(v == 1, "xx", 1, tvm.tir.Evaluate(1))
[all …]
H A Dtest_tvmscript_error_report.py21 from tvm import tir
26 @tvm.script.tir
32 @tvm.script.tir
44 @tvm.script.tir
56 @tvm.script.tir
63 @tvm.script.tir
75 @tvm.script.tir
81 @tvm.script.tir
88 @tvm.script.tir
94 @tvm.script.tir
[all …]
H A Dtest_tir_transform_vectorize.py24 ib = tvm.tir.ir_builder.create()
36 assert isinstance(stmt, tvm.tir.For)
45 ib = tvm.tir.ir_builder.create()
56 assert isinstance(stmt, tvm.tir.For)
65 ib = tvm.tir.ir_builder.create()
86 v = tvm.tir.Var("v", "float32")
87 ib = tvm.tir.ir_builder.create()
101 ib = tvm.tir.ir_builder.create()
116 ib = tvm.tir.ir_builder.create()
132 ib = tvm.tir.ir_builder.create()
[all …]
H A Dtest_tir_nodes.py73 assert isinstance(tvm.tir.max(x, y), tvm.tir.Max)
74 assert isinstance(tvm.tir.min(x, y), tvm.tir.Min)
96 stmt = tvm.tir.LetStmt(x, 10, tvm.tir.Evaluate(x + 1))
111 stmt = tvm.tir.AttrStmt(y, "stride", 10, tvm.tir.Evaluate(x + 1))
131 x = tvm.tir.Evaluate(0)
132 tvm.tir.For(te.var("i"), 0, 1, tvm.tir.For.Serial, 0, x)
157 tvm.tir.any()
190 tvm.tir.all()
280 test(tvm.tir.const(5, "int32"), tvm.tir.const(0, "int32"))
318 stmt = tvm.tir.LetStmt(x, 10, tvm.tir.Evaluate(x + 1))
[all …]
H A Dtest_arith_rewrite_simplify.py36 ck.verify(tvm.tir.Ramp(x, 1, 4) + tvm.tir.Ramp(y, 2, 4), tvm.tir.Ramp(x + y, 3, 4))
46 ck.verify(tvm.tir.Ramp(x, 4, 4) - tvm.tir.Ramp(y, 2, 4), tvm.tir.Ramp(x - y, 2, 4))
81 ck.verify(fld(tvm.tir.Ramp(x, 8, 5), tvm.tir.Broadcast(4, 5)), tvm.tir.Ramp(fld(x, 4), 2, 5))
99 fld(tvm.tir.Ramp(x * 8, 2, 4), tvm.tir.Broadcast(64, 4)), tvm.tir.Broadcast(fld(x, 8), 4)
123 ck.verify(flm(tvm.tir.Ramp(x * 8, 1, 4), tvm.tir.Broadcast(4, 4)), tvm.tir.Ramp(0, 1, 4))
865 ck.verify(tvm.tir.And(tvm.tir.EQ(x, y), tvm.tir.NE(x, y)), tvm.tir.const(False, "bool"))
866 ck.verify(tvm.tir.And(tvm.tir.NE(x, y), tvm.tir.EQ(x, y)), tvm.tir.const(False, "bool"))
867 ck.verify(tvm.tir.And(x > 1, tvm.tir.Not(x > 1)), tvm.tir.const(False, "bool"))
880 ck.verify(tvm.tir.Or(tvm.tir.EQ(x, y), tvm.tir.NE(x, y)), tvm.tir.const(True, "bool"))
881 ck.verify(tvm.tir.Or(tvm.tir.NE(x, y), tvm.tir.EQ(x, y)), tvm.tir.const(True, "bool"))
[all …]
H A Dtest_tir_transform_unroll_loop.py23 ib = tvm.tir.ir_builder.create()
26 Ab = tvm.tir.decl_buffer((n,), dtype)
36 assert isinstance(stmt, tvm.tir.For)
40 assert not isinstance(ret, tvm.tir.For)
44 assert isinstance(ret, tvm.tir.For)
50 assert isinstance(ret, tvm.tir.For)
53 ib = tvm.tir.ir_builder.create()
57 wrapped = tvm.tir.SeqStmt([wrapped, stmt])
58 assert isinstance(ret, tvm.tir.For)
72 ib = tvm.tir.ir_builder.create()
[all …]
H A Dtest_tir_transform_remove_no_op.py22 return tvm.tir.Evaluate(0)
32 Ab = tvm.tir.decl_buffer((n,), dtype)
33 stmt = tvm.tir.For(
39 tvm.tir.For(
45 tvm.tir.For(
51 tvm.tir.IfThenElse((i * m + j + k < n), tvm.tir.Evaluate(m), tvm.tir.Evaluate(n)),
59 assert isinstance(ret, tvm.tir.Evaluate)
60 store = tvm.tir.Store(Ab.data, tvm.tir.Load(dtype, Ab.data, i) + 1, i + 1)
61 stmt2 = tvm.tir.SeqStmt([nop(), tvm.tir.SeqStmt([store, nop()])])
68 stmt3 = tvm.tir.For(i, 0, 0, 0, 0, store)
[all …]
H A Dtest_tir_ops.py37 tmod = tvm.tir.truncmod
53 tmod = tvm.tir.truncmod
54 tdiv = tvm.tir.truncdiv
68 for func in [tvm.tir.all, tvm.tir.any]:
80 tvm_func(tvm.tir.const(v1, "uint1"), tvm.tir.const(v2, "uint1")),
85 true = tvm.tir.const(1, "uint1")
86 false = tvm.tir.const(0, "uint1")
88 assert tvm.tir.all(x, true).same_as(x)
89 assert tvm.tir.all(true, x).same_as(x)
100 x1 = tvm.tir.const(4, "int32")
[all …]
H A Dtest_tir_transform_loop_partition.py70 ib = tvm.tir.ir_builder.create()
90 ib = tvm.tir.ir_builder.create()
160 ib = tvm.tir.ir_builder.create()
165 ib.emit(tvm.tir.Evaluate(tvm.tir.Select(ib.likely(i * 4 + j < n), m, n)))
176 ib = tvm.tir.ir_builder.create()
180 ib.emit(tvm.tir.Evaluate(tvm.tir.Select(ib.likely(tvm.tir.EQ(i, 5)), m, n)))
212 ib = tvm.tir.ir_builder.create()
278 ib = tvm.tir.ir_builder.create()
310 ib = tvm.tir.ir_builder.create()
338 ib = tvm.tir.ir_builder.create()
[all …]
H A Dtest_tir_transform_rewrite_unsafe_select.py22 ib = tvm.tir.ir_builder.create()
25 y = tvm.tir.Select(i > 1, A[i - 1], 1.0)
27 mod = tvm.IRModule.from_expr(tvm.tir.PrimFunc([i], tvm.tir.Evaluate(y)))
28 yy = tvm.tir.transform.RewriteUnsafeSelect()(mod)["main"].body.value
30 z = tvm.tir.Select(tvm.tir.Select(i > 1, A[i - 1], 1.0) > 0.0, A[i], 0.1)
31 mod = tvm.IRModule.from_expr(tvm.tir.PrimFunc([i], tvm.tir.Evaluate(z)))
32 zz = tvm.tir.transform.RewriteUnsafeSelect()(mod)["main"].body.value
34 a = tvm.tir.Select(tvm.tir.floordiv(i, 4) > 10, y, z)
36 mod = tvm.IRModule.from_expr(tvm.tir.PrimFunc([i], tvm.tir.Evaluate(a)))
37 aa = tvm.tir.transform.RewriteUnsafeSelect()(mod)["main"].body.value
[all …]
H A Dtest_tir_analysis_verify_ssa.py24 z = tvm.tir.Evaluate(x + y)
25 assert tvm.tir.analysis.verify_ssa(tvm.tir.PrimFunc([x, y], z))
27 assert not tvm.tir.analysis.verify_ssa(tvm.tir.PrimFunc([x, y], tvm.tir.LetStmt(x, 1, z)))
32 z1 = tvm.tir.Let(x, 1, x + 1)
33 z2 = tvm.tir.Let(x, 2, x + 2)
35 assert tvm.tir.analysis.verify_ssa(tvm.tir.PrimFunc([], tvm.tir.Evaluate(z1 + z1)))
36 assert not tvm.tir.analysis.verify_ssa(tvm.tir.PrimFunc([], tvm.tir.Evaluate(z1 * z2)))
/dports/security/oidentd/oidentd-2.5.1/src/kernel/legacy/
H A Dopenbsd29.c71 struct tcp_ident_mapping tir; in get_user4() local
77 memset(&tir, 0, sizeof(tir)); in get_user4()
79 tir.faddr.ss_family = AF_INET; in get_user4()
87 tir.laddr.ss_family = AF_INET; in get_user4()
93 i = sizeof(tir); in get_user4()
97 if (error == 0 && tir.ruid != -1) in get_user4()
98 return tir.ruid; in get_user4()
118 struct tcp_ident_mapping tir; in get_user6() local
125 memset(&tir, 0, sizeof(tir)); in get_user6()
147 i = sizeof(tir); in get_user6()
[all …]
H A Dopenbsd24.c65 struct tcp_ident_mapping tir; in get_user4() local
71 memset(&tir, 0, sizeof(tir)); in get_user4()
73 tir.faddr.sa_family = AF_INET; in get_user4()
81 tir.laddr.sa_family = AF_INET; in get_user4()
87 i = sizeof(tir); in get_user4()
91 if (error == 0 && tir.ruid != -1) in get_user4()
92 return tir.ruid; in get_user4()
112 struct tcp_ident_mapping tir; in get_user6() local
119 memset(&tir, 0, sizeof(tir)); in get_user6()
141 i = sizeof(tir); in get_user6()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/driver/
H A Ddriver_api.cc76 if (it.as<tir::VarNode>()) { in BufferWithOffsetAlignment()
82 tir::BufferType buffer_type = has_any ? tir::kAutoBroadcast : tir::kDefault; in BufferWithOffsetAlignment()
124 return tir::PrimFunc(nullptr); in Filter()
142 Map<te::Tensor, tir::Buffer> out_binds; in lower()
206 Filter([](const tir::PrimFunc& f) { in SplitDevHostFuncs()
211 tir::transform::LowerTVMBuiltin(), in SplitDevHostFuncs()
213 tir::transform::LowerIntrin(), in SplitDevHostFuncs()
222 Filter([](const tir::PrimFunc& f) { in SplitDevHostFuncs()
227 tir::transform::LowerWarpMemory(), in SplitDevHostFuncs()
228 tir::transform::Simplify(), in SplitDevHostFuncs()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/script/
H A Dintrin.py22 import tvm.tir
28 return tvm.tir.const(imm, "bool")
33 return tvm.tir.const(imm, "int8")
38 return tvm.tir.const(imm, "int16")
43 return tvm.tir.const(imm, "int32")
48 return tvm.tir.const(imm, "int64")
53 return tvm.tir.const(imm, "uint8")
93 return tvm.tir.floordiv(x, y)
98 return tvm.tir.floormod(x, y)
123 return tvm.tir.Evaluate(value)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/driver/
H A Dbuild_module.py23 import tvm.tir
66 buf = tvm.tir.decl_buffer(
75 elif isinstance(x, tvm.tir.Var):
173 tvm.tir.transform.BF16Legalize(),
175 tvm.tir.transform.Simplify(),
188 tvm.tir.transform.UnrollLoop(),
194 tvm.tir.transform.Simplify(),
195 tvm.tir.transform.RemoveNoOp(),
260 tvm.tir.transform.Filter(
265 tvm.tir.transform.Simplify(),
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/x86/
H A Dtensor_intrin.py78 a_buffer = tvm.tir.decl_buffer(
81 b_buffer = tvm.tir.decl_buffer(
87 ib = tvm.tir.ir_builder.create()
101 tvm.tir.const(0, "uint32"),
108 tvm.tir.const(0, "uint32"),
172 a_buffer = tvm.tir.decl_buffer(
180 ib = tvm.tir.ir_builder.create()
196 tvm.tir.const(0, "uint32"),
262 a_buffer = tvm.tir.decl_buffer(
265 b_buffer = tvm.tir.decl_buffer(
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/op/
H A Dop.cc39 using namespace tir;
90 return tir::Call(t, tir::builtin::large_uint_imm(), in LargeUIntImm()
226 namespace tir { namespace
285 return tir::Call(t, tir::builtin::reinterpret(), {value}); in reinterpret()
416 return tir::Call(cond.dtype(), tir::builtin::likely(), {cond}); in likely()
502 return tir::Call(a.dtype(), tir::builtin::shift_right(), {a, b}); in operator >>()
521 return tir::Call(a.dtype(), tir::builtin::shift_left(), {a, b}); in operator <<()
533 return tir::Call(a.dtype(), tir::builtin::bitwise_and(), {a, b}); in operator &()
545 return tir::Call(a.dtype(), tir::builtin::bitwise_or(), {a, b}); in operator |()
557 return tir::Call(a.dtype(), tir::builtin::bitwise_xor(), {a, b}); in operator ^()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/arith/
H A Dir_mutator_with_analyzer.h44 class IRMutatorWithAnalyzer : public tir::StmtExprMutator {
52 tir::Stmt VisitStmt_(const tir::ForNode* op) override;
53 tir::Stmt VisitStmt_(const tir::LetStmtNode* op) override;
54 tir::Stmt VisitStmt_(const tir::IfThenElseNode* op) override;
55 tir::Stmt VisitStmt_(const tir::AttrStmtNode* op) override;
56 tir::Stmt VisitStmt_(const tir::AssertStmtNode* op) override;
57 PrimExpr VisitExpr_(const tir::LetNode* op) override;
58 PrimExpr VisitExpr_(const tir::SelectNode* op) override;
59 PrimExpr VisitExpr_(const tir::CallNode* op) override;
60 PrimExpr VisitExpr_(const tir::ReduceNode* op) override;
/dports/security/oidentd/oidentd-2.5.1/src/kernel/
H A Dopenbsd30.c83 struct tcp_ident_mapping tir; in get_user4() local
89 memset(&tir, 0, sizeof(tir)); in get_user4()
91 tir.faddr.ss_family = AF_INET; in get_user4()
99 tir.laddr.ss_family = AF_INET; in get_user4()
105 i = sizeof(tir); in get_user4()
109 if (error == 0 && tir.ruid != -1) in get_user4()
110 return tir.ruid; in get_user4()
230 struct tcp_ident_mapping tir; in get_user6() local
237 memset(&tir, 0, sizeof(tir)); in get_user6()
259 i = sizeof(tir); in get_user6()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/ir/
H A Dattr_functor.h83 virtual R VisitAttr_(const tir::VarNode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
84 virtual R VisitAttr_(const tir::SizeVarNode* op, Args... args) { in VisitAttr_()
87 virtual R VisitAttr_(const tir::AddNode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
96 virtual R VisitAttr_(const tir::GENode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
97 virtual R VisitAttr_(const tir::GTNode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
98 virtual R VisitAttr_(const tir::LTNode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
99 virtual R VisitAttr_(const tir::LENode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
100 virtual R VisitAttr_(const tir::EQNode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
101 virtual R VisitAttr_(const tir::NENode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
103 virtual R VisitAttr_(const tir::OrNode* op, Args... args) ATTR_FUNCTOR_DEFAULT;
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/cpp/
H A Dtir_analysis_side_effect.cc28 auto A = tir::Var("A", DataType::Handle()); in TEST()
29 auto i = tir::Var("i", DataType::Int(32)); in TEST()
30 CHECK(tir::SideEffect(tir::Load(DataType::Float(32), A, i, tir::const_true(1))) == in TEST()
31 tir::CallEffectKind::kReadState); in TEST()
32 CHECK(tir::SideEffect(exp(tir::Cast(DataType::Float(32), i + 1))) == tir::CallEffectKind::kPure); in TEST()
33 CHECK(tir::SideEffect(tir::Call(DataType::Handle(), tir::builtin::tvm_storage_sync(), {})) == in TEST()
34 tir::CallEffectKind::kUpdateState); in TEST()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/
H A Dutil.py24 from tvm.tir import layout, bijective_layout
85 return tvm.tir.const(1, "int32")
177 if isinstance(elem, (tvm.tir.Var, tvm.tir.expr.Any)):
264 idxd = tvm.tir.indexdiv
265 idxm = tvm.tir.indexmod
291 idxm = tvm.tir.indexmod
294 now = tvm.tir.const(0.0, dtype)
297 now = tvm.tir.Select(
398 return tvm.tir.Select(tvm.tir.Or(bc, ec), tvm.tir.const(False), ss.equal(0))
434 b = tvm.tir.Select(z < b, z - 1, b)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/vta/python/vta/
H A Dtransform.py198 stmt = tvm.tir.LetStmt(
200 tvm.tir.call_extern(
286 return tvm.tir.Evaluate(0)
313 return tvm.tir.SeqStmt([stmt.body, tvm.tir.Evaluate(sync)])
342 idxd = tvm.tir.indexdiv
343 idxm = tvm.tir.indexmod
511 tvm.tir.call_extern(
590 tvm.tir.call_extern(
642 wgt_layout = tvm.tir.decl_buffer(
650 inp_layout = tvm.tir.decl_buffer(
[all …]

12345678910>>...68