Skip to content

Commit 214872e

Browse files
committed
Fix buffer size
Remove unused script
1 parent c69a5ca commit 214872e

File tree

4 files changed

+7
-44
lines changed

4 files changed

+7
-44
lines changed

python/tvm/contrib/hexagon/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(
5757
remote_kw: dict,
5858
session_name: str = "hexagon-rpc",
5959
remote_stack_size_bytes: int = 256 * 1024, # Min size for main thread in QuRT/sim
60-
rpc_receive_buffer_size_bytes: int = 2 * 1024 * 1024,
60+
rpc_receive_buffer_size_bytes: int = 5 * 1024 * 1024, # Size for passing hexagon tests
6161
):
6262
self._launcher = launcher
6363
self._session_name = session_name

tests/python/contrib/test_hexagon/topi/test_batch_matmul.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ class TestMatMulFloat:
4646

4747
# TODO(mehrdadh): add dynamic testing
4848
@requires_hexagon_toolchain
49-
def test_batch_matmul(hexagon_session, x_batch, y_batch, M, N, K, dtype):
49+
def test_batch_matmul(self, hexagon_session, x_batch, y_batch, M, N, K, dtype):
5050
if dtype == "float16":
5151
pytest.xfail("float16 is not supported.")
5252

@@ -98,7 +98,7 @@ class TestMatMulInt8:
9898
)
9999

100100
@requires_hexagon_toolchain
101-
def test_batch_matmul_int8(hexagon_session, x_batch, y_batch, M, N, K):
101+
def test_batch_matmul_int8(self, hexagon_session, x_batch, y_batch, M, N, K):
102102
dtype = "int8"
103103
out_dtype = "int8"
104104
assert x_batch == y_batch or x_batch == 1 or y_batch == 1

tests/python/contrib/test_hexagon/topi/test_pooling.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ class TestAdaptivePool:
5757
)
5858

5959
@requires_hexagon_toolchain
60-
def test_adaptive_pool(hexagon_session, dshape, out_size, pool_type, layout):
60+
def test_adaptive_pool(self, hexagon_session, dshape, out_size, pool_type, layout):
6161
dtype = "float32"
6262
np_data = np.random.uniform(low=0, high=255, size=dshape).astype(dtype)
6363
np_out = tvm.topi.testing.adaptive_pool(np_data, out_size, pool_type, layout)
@@ -234,6 +234,7 @@ class TestPool1D:
234234

235235
@requires_hexagon_toolchain
236236
def test_pool1d(
237+
self,
237238
hexagon_session,
238239
input_shape,
239240
kernel,
@@ -310,6 +311,7 @@ class TestPool2D:
310311

311312
@requires_hexagon_toolchain
312313
def test_pool2d(
314+
self,
313315
hexagon_session,
314316
input_shape,
315317
kernel,
@@ -707,6 +709,7 @@ class TestPool3D:
707709

708710
@requires_hexagon_toolchain
709711
def test_pool3d(
712+
self,
710713
hexagon_session,
711714
input_shape,
712715
kernel,

tests/scripts/task_python_hexagon_simulator.sh

Lines changed: 0 additions & 40 deletions
This file was deleted.

0 commit comments

Comments
 (0)