/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/unittest/ |
H A D | test_tvmscript_roundtrip.py | 23 @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 D | test_tir_constructor.py | 45 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 D | test_tvmscript_error_report.py | 21 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 D | test_tir_transform_vectorize.py | 24 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 D | test_tir_nodes.py | 73 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 D | test_arith_rewrite_simplify.py | 36 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 D | test_tir_transform_unroll_loop.py | 23 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 D | test_tir_transform_remove_no_op.py | 22 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 D | test_tir_ops.py | 37 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 D | test_tir_transform_loop_partition.py | 70 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 D | test_tir_transform_rewrite_unsafe_select.py | 22 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 D | test_tir_analysis_verify_ssa.py | 24 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 D | openbsd29.c | 71 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 D | openbsd24.c | 65 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 D | driver_api.cc | 76 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 D | intrin.py | 22 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 D | build_module.py | 23 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 D | tensor_intrin.py | 78 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 D | op.cc | 39 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 D | ir_mutator_with_analyzer.h | 44 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 D | openbsd30.c | 83 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 D | attr_functor.h | 83 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 D | tir_analysis_side_effect.cc | 28 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 D | util.py | 24 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 D | transform.py | 198 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 …]
|