diff --git a/topi/tests/python/test_topi_broadcast.py b/topi/tests/python/test_topi_broadcast.py index dbd1c37a221b35248b0467ef5335151c3cc44810..e5f88e9d4df6836675c9c3a5efdb691227be3bd3 100644 --- a/topi/tests/python/test_topi_broadcast.py +++ b/topi/tests/python/test_topi_broadcast.py @@ -12,6 +12,7 @@ def verify_broadcast_to_ele(in_shape, out_shape): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_broadcast(B) ctx = tvm.context(device, 0) @@ -52,6 +53,7 @@ def verify_broadcast_binary_ele(lhs_shape, rhs_shape, typ="add"): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_broadcast(C) ctx = tvm.context(device, 0) diff --git a/topi/tests/python/test_topi_conv2d_hwcn.py b/topi/tests/python/test_topi_conv2d_hwcn.py index eb05b342be844d71bbe7e225b30a2580ff26f508..9e6da1a58f0e18ca0fb8cd8a44c84b763e919b45 100644 --- a/topi/tests/python/test_topi_conv2d_hwcn.py +++ b/topi/tests/python/test_topi_conv2d_hwcn.py @@ -34,6 +34,7 @@ def verify_conv2d_hwcn(batch, in_channel, in_size, num_filter, kernel, stride, p if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) ctx = tvm.context(device, 0) a = tvm.nd.array(a_np, ctx) w = tvm.nd.array(w_np, ctx) diff --git a/topi/tests/python/test_topi_conv2d_nchw.py b/topi/tests/python/test_topi_conv2d_nchw.py index be0019190167211bc56c54b34ab9906d7f9eab14..b8d7bec79d53421bfd1d499f9f485dc7bd671ebe 100644 --- a/topi/tests/python/test_topi_conv2d_nchw.py +++ b/topi/tests/python/test_topi_conv2d_nchw.py @@ -33,6 +33,7 @@ def verify_conv2d_nchw(batch, in_channel, in_size, num_filter, kernel, stride, p if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s1 = topi.generic.schedule_conv2d_nchw([B]) s2 = topi.generic.schedule_conv2d_nchw([C]) diff --git a/topi/tests/python/test_topi_conv2d_transpose_nchw.py b/topi/tests/python/test_topi_conv2d_transpose_nchw.py index 738831d6c701b08347572cf43e4b1a03ee904bbb..02e0853872535238ed642c4b6f834eaabf9353e1 100644 --- a/topi/tests/python/test_topi_conv2d_transpose_nchw.py +++ b/topi/tests/python/test_topi_conv2d_transpose_nchw.py @@ -32,6 +32,7 @@ def verify_conv2d_transpose_nchw(batch, in_channel, in_size, num_filter, kernel, if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s1 = topi.generic.schedule_conv2d_transpose_nchw([B]) s2 = topi.generic.schedule_conv2d_transpose_nchw([C]) diff --git a/topi/tests/python/test_topi_dense.py b/topi/tests/python/test_topi_dense.py index 94badb81769c803f89db2e8d5f637d259e4771c0..0e5984afc011f669a18ac02ab415b18d55f24690 100644 --- a/topi/tests/python/test_topi_dense.py +++ b/topi/tests/python/test_topi_dense.py @@ -32,6 +32,7 @@ def verify_dense(batch, in_dim, out_dim, use_bias=True): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_dense(D) ctx = tvm.context(device, 0) diff --git a/topi/tests/python/test_topi_depthwise_conv2d.py b/topi/tests/python/test_topi_depthwise_conv2d.py index adaad2221fcd0ccc125174e8a1528e5735fb2730..df9a755c078e52e8fd458364b6832c038301e6ab 100644 --- a/topi/tests/python/test_topi_depthwise_conv2d.py +++ b/topi/tests/python/test_topi_depthwise_conv2d.py @@ -27,6 +27,7 @@ def depthwise_conv2d_with_workload_nchw(batch, in_channel, in_height, channel_mu if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): # schedule s1 = topi.generic.schedule_depthwise_conv2d_nchw(DepthwiseConv2d) @@ -111,6 +112,7 @@ def depthwise_conv2d_with_workload_nhwc(batch, in_channel, in_height, channel_mu if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s1 = topi.generic.schedule_depthwise_conv2d_nhwc(DepthwiseConv2d) diff --git a/topi/tests/python/test_topi_depthwise_conv2d_back_input.py b/topi/tests/python/test_topi_depthwise_conv2d_back_input.py index 616c8e57795dfc639e0a4290fc1fc629394f2113..732e31e7e1666ed883dd152f6f968c730279fcfb 100644 --- a/topi/tests/python/test_topi_depthwise_conv2d_back_input.py +++ b/topi/tests/python/test_topi_depthwise_conv2d_back_input.py @@ -35,6 +35,7 @@ def verify_depthwise_conv2d_back_input(batch, in_channel, in_h, channel_multipli if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) ctx = tvm.context(device, 0) # build the kernel f = tvm.build(schedule, [Filter, Out_grad, In_grad], device) diff --git a/topi/tests/python/test_topi_depthwise_conv2d_back_weight.py b/topi/tests/python/test_topi_depthwise_conv2d_back_weight.py index 4a4d2552ccfd14161c19890785d8535fdbdec173..b9a0fcbb3ddf8d0214c29b0bde507fbc0f3c11f3 100644 --- a/topi/tests/python/test_topi_depthwise_conv2d_back_weight.py +++ b/topi/tests/python/test_topi_depthwise_conv2d_back_weight.py @@ -35,6 +35,7 @@ def verify_depthwise_conv2d_back_weight(batch, in_channel, in_h, channel_multipl if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) ctx = tvm.context(device, 0) # build the kernel f = tvm.build(schedule, [Input, Out_grad, Weight_grad], device) diff --git a/topi/tests/python/test_topi_pooling.py b/topi/tests/python/test_topi_pooling.py index 3cddb7dfbc649e06253e062f783a24631f1361ce..86f724a71fe8784d3798dd67c505375d92e71875 100644 --- a/topi/tests/python/test_topi_pooling.py +++ b/topi/tests/python/test_topi_pooling.py @@ -35,6 +35,7 @@ def verify_pool(n, ic, ih, kh, sh, padding, pool_type): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_pool(B) ctx = tvm.context(device, 0) @@ -70,6 +71,7 @@ def verify_global_pool(n, c, h, w, pool_type): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_global_pool(B) ctx = tvm.context(device, 0) diff --git a/topi/tests/python/test_topi_reduce.py b/topi/tests/python/test_topi_reduce.py index a34b7a173ff23a6acc4eeed0e1198c9b27947eef..13cd8fcdcd21649d9e1a8c7d66b578fd5903d343 100644 --- a/topi/tests/python/test_topi_reduce.py +++ b/topi/tests/python/test_topi_reduce.py @@ -50,6 +50,7 @@ def verify_reduce_map_ele(in_shape, axis, keepdims, type="sum"): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_reduce(B) ctx = tvm.context(device, 0) diff --git a/topi/tests/python/test_topi_relu.py b/topi/tests/python/test_topi_relu.py index 04ee3d87b0dc2398bc543164ff1bc2dfc2321fa7..f70f48c501100e45d4611665ec2ffdc842a3c49f 100644 --- a/topi/tests/python/test_topi_relu.py +++ b/topi/tests/python/test_topi_relu.py @@ -16,6 +16,7 @@ def verify_relu(m, n): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_elemwise(B) ctx = tvm.context(device, 0) diff --git a/topi/tests/python/test_topi_softmax.py b/topi/tests/python/test_topi_softmax.py index 5e54e78d6d02f8aafcd4bf49791ca65f673459e4..72d86a43271d273ffabe01992a0a99d258dfa746 100644 --- a/topi/tests/python/test_topi_softmax.py +++ b/topi/tests/python/test_topi_softmax.py @@ -20,6 +20,7 @@ def verify_softmax(m, n): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_softmax(B) ctx = tvm.context(device, 0) @@ -50,6 +51,7 @@ def verify_log_softmax(m, n): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_softmax(B) ctx = tvm.context(device, 0) diff --git a/topi/tests/python/test_topi_transform.py b/topi/tests/python/test_topi_transform.py index 1e47fdc45ad37128839f43d945fb74204596fa8c..b8f44b5633c3ece14f3a4cd3a36db40f34c0086d 100644 --- a/topi/tests/python/test_topi_transform.py +++ b/topi/tests/python/test_topi_transform.py @@ -10,6 +10,7 @@ def verify_expand_dims(in_shape, out_shape, axis, num_newaxis): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_broadcast(B) ctx = tvm.context(device, 0) @@ -32,6 +33,7 @@ def verify_tranpose(in_shape, axes): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_injective(B) ctx = tvm.context(device, 0) @@ -54,6 +56,7 @@ def verify_reshape(src_shape, dst_shape): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_injective(B) ctx = tvm.context(device, 0) @@ -76,6 +79,7 @@ def verify_squeeze(src_shape, axis): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_injective(B) ctx = tvm.context(device, 0) @@ -103,6 +107,7 @@ def verify_concatenate(shapes, axis): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_injective(out_tensor) ctx = tvm.context(device, 0) @@ -125,6 +130,7 @@ def verify_split(src_shape, indices_or_sections, axis): if not tvm.module.enabled(device): print("Skip because %s is not enabled" % device) return + print("Running on target: %s" % device) with tvm.target.create(device): s = topi.generic.schedule_injective(tensor_l) ctx = tvm.context(device, 0)