diff --git a/tests/see_pool.py b/tests/see_pool.py deleted file mode 100644 index d563664902d3e..0000000000000 --- a/tests/see_pool.py +++ /dev/null @@ -1,37 +0,0 @@ -import numpy as np -import torch -import tvm -from torch import nn -from tvm import te, topi - -if __name__ == "__main__": - """verify function of pool1d""" - input_shape = (1, 1, 10) - kernel = [3] - stride = [1] - dilation = [2] - A = te.placeholder(input_shape, name="A") - B = topi.nn.pool1d( - A, - kernel=kernel, - stride=stride, - dilation=dilation, - padding=[1, 1], - pool_type="avg", - ceil_mode=False, - count_include_pad=False, - ) - - s = te.create_schedule(B.op) - - tgt = tvm.target.Target(target="llvm", host="llvm") - dev = tvm.device(tgt.kind.name, 0) - func = tvm.build(s, [A, B], tgt, name="pool1d") - - print(tvm.lower(s, [A, B], simple_mode=True)) - - a = tvm.nd.array(np.ones(input_shape).astype(A.dtype), dev) - b = tvm.nd.array(np.zeros((1, 1, 8)).astype(A.dtype), dev) - func(a, b) - print(a.asnumpy()) - print(b.asnumpy())