add helpful message to topi test (#592)

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)

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)

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])

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])

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)

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)

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)

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)

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)

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)

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)

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)

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)