Unverified Commit fc55f34f by Tianqi Chen Committed by GitHub

[NNPACK] temporary disable nnpack test (#2115)

parent 415f7c4b
...@@ -21,6 +21,7 @@ def test_fully_connected_output(): ...@@ -21,6 +21,7 @@ def test_fully_connected_output():
if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_output", True): if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_output", True):
print("skip because extern function is not available") print("skip because extern function is not available")
return return
return
ctx = tvm.cpu(0) ctx = tvm.cpu(0)
f = tvm.build(s, [A, B, D, bias], target) f = tvm.build(s, [A, B, D, bias], target)
a = tvm.nd.array(np.random.uniform(size=(n, l)).astype(A.dtype), ctx) a = tvm.nd.array(np.random.uniform(size=(n, l)).astype(A.dtype), ctx)
...@@ -51,6 +52,7 @@ def test_fully_connected_inference(): ...@@ -51,6 +52,7 @@ def test_fully_connected_inference():
if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True): if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
print("skip because extern function is not available") print("skip because extern function is not available")
return return
return
ctx = tvm.cpu(0) ctx = tvm.cpu(0)
f = tvm.build(s, [A, B, D, bias], target) f = tvm.build(s, [A, B, D, bias], target)
a = tvm.nd.array(np.random.uniform(size=(l)).astype(A.dtype), ctx) a = tvm.nd.array(np.random.uniform(size=(l)).astype(A.dtype), ctx)
...@@ -128,6 +130,7 @@ def test_convolution_inference(): ...@@ -128,6 +130,7 @@ def test_convolution_inference():
if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True): if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
print("skip because extern function is not available") print("skip because extern function is not available")
return return
return
ctx = tvm.cpu(0) ctx = tvm.cpu(0)
output = nnpack.convolution_inference( output = nnpack.convolution_inference(
data, kernel, bias if with_bias else None, data, kernel, bias if with_bias else None,
...@@ -189,7 +192,7 @@ def test_convolution_inference_without_weight_transform(): ...@@ -189,7 +192,7 @@ def test_convolution_inference_without_weight_transform():
if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True): if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
print("skip because extern function is not available") print("skip because extern function is not available")
return return
return
ctx = tvm.cpu(0) ctx = tvm.cpu(0)
transformed_kernel = nnpack.convolution_inference_weight_transform( transformed_kernel = nnpack.convolution_inference_weight_transform(
kernel, algorithm=algorithm) kernel, algorithm=algorithm)
...@@ -246,6 +249,7 @@ def test_convolution_output(): ...@@ -246,6 +249,7 @@ def test_convolution_output():
if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True): if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
print("skip because extern function is not available") print("skip because extern function is not available")
return return
return
ctx = tvm.cpu(0) ctx = tvm.cpu(0)
f = tvm.build(s, [data, kernel, bias, output], target) f = tvm.build(s, [data, kernel, bias, output], target)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment