From dfcc614f980845e2045922e881e9d0d609557bf5 Mon Sep 17 00:00:00 2001 From: Neil Hickey Date: Wed, 18 Jan 2023 15:29:04 +0000 Subject: [PATCH 1/2] [TOPI] Making test_strided_set require a GPU for testing --- tests/python/topi/python/test_topi_transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/topi/python/test_topi_transform.py b/tests/python/topi/python/test_topi_transform.py index 0f64b486f375..ceac77217bfc 100644 --- a/tests/python/topi/python/test_topi_transform.py +++ b/tests/python/topi/python/test_topi_transform.py @@ -858,7 +858,7 @@ def test_dynamic_strided_slice(): verify_dynamic_strided_slice((3, 4, 3), [1, 1, 0], [4, 4, 3]) verify_dynamic_strided_slice((3, 4, 3), [0, 2, 0], [1, 2, 3]) - +@tvm.testing.requires_gpu @tvm.testing.uses_gpu def test_strided_set(): verify_strided_set((3, 4, 3), (3, 2, 2), [0, 3, 0], [4, 1, 4], [1, -1, 2]) From e6a75eef83379a572acb11d483b43aa6ca730a97 Mon Sep 17 00:00:00 2001 From: Ashutosh Parkhi <86472128+ashutosh-arm@users.noreply.github.com> Date: Wed, 18 Jan 2023 17:44:09 +0000 Subject: [PATCH 2/2] Fixed linting error --- tests/python/topi/python/test_topi_transform.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/python/topi/python/test_topi_transform.py b/tests/python/topi/python/test_topi_transform.py index ceac77217bfc..e34905f15379 100644 --- a/tests/python/topi/python/test_topi_transform.py +++ b/tests/python/topi/python/test_topi_transform.py @@ -858,6 +858,7 @@ def test_dynamic_strided_slice(): verify_dynamic_strided_slice((3, 4, 3), [1, 1, 0], [4, 4, 3]) verify_dynamic_strided_slice((3, 4, 3), [0, 2, 0], [1, 2, 3]) + @tvm.testing.requires_gpu @tvm.testing.uses_gpu def test_strided_set():