Skip to content

Commit 876842a

Browse files
Skip using @_condition.repeat in cupy tests
1 parent 31e8bbb commit 876842a

File tree

1 file changed

+12
-13
lines changed

1 file changed

+12
-13
lines changed

tests/third_party/cupy/linalg_tests/test_decomposition.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -254,22 +254,21 @@ def check_singular(self, shape, xp, dtype):
254254
assert (a == a_copy).all()
255255
return result
256256

257-
@_condition.repeat(3, 10)
257+
# TODO: Use this decorator in all tests when dpcpp_linux-64>2024.1.0
258+
# is available. Bug CMPLRLLVM-53771
259+
# @_condition.repeat(3, 10)
258260
def test_svd_rank2(self):
259261
self.check_usv((3, 7))
260262
self.check_usv((2, 2))
261263
self.check_usv((7, 3))
262264

263-
@_condition.repeat(3, 10)
265+
# @_condition.repeat(3, 10)
264266
def test_svd_rank2_no_uv(self):
265267
self.check_singular((3, 7))
266268
self.check_singular((2, 2))
267269
self.check_singular((7, 3))
268270

269271
@testing.with_requires("numpy>=1.16")
270-
# dpnp.matmul does not support input empty arrays
271-
# TODO: remove it when support is added
272-
@pytest.mark.usefixtures("allow_fall_back_on_numpy")
273272
def test_svd_rank2_empty_array(self):
274273
self.check_usv((0, 3))
275274
self.check_usv((3, 0))
@@ -283,7 +282,7 @@ def test_svd_rank2_empty_array_compute_uv_false(self, xp):
283282
array, full_matrices=self.full_matrices, compute_uv=False
284283
)
285284

286-
@_condition.repeat(3, 10)
285+
# @_condition.repeat(3, 10)
287286
def test_svd_rank3(self):
288287
self.check_usv((2, 3, 4))
289288
self.check_usv((2, 3, 7))
@@ -292,22 +291,22 @@ def test_svd_rank3(self):
292291
self.check_usv((2, 4, 3))
293292
self.check_usv((2, 32, 32))
294293

295-
@_condition.repeat(3, 10)
294+
# @_condition.repeat(3, 10)
296295
def test_svd_rank3_loop(self):
297296
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
298297
self.check_usv((2, 64, 64))
299298
self.check_usv((2, 64, 32))
300299
self.check_usv((2, 32, 64))
301300

302-
@_condition.repeat(3, 10)
301+
# @_condition.repeat(3, 10)
303302
def test_svd_rank3_no_uv(self):
304303
self.check_singular((2, 3, 4))
305304
self.check_singular((2, 3, 7))
306305
self.check_singular((2, 4, 4))
307306
self.check_singular((2, 7, 3))
308307
self.check_singular((2, 4, 3))
309308

310-
@_condition.repeat(3, 10)
309+
# @_condition.repeat(3, 10)
311310
def test_svd_rank3_no_uv_loop(self):
312311
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
313312
self.check_singular((2, 64, 64))
@@ -339,7 +338,7 @@ def test_svd_rank3_empty_array_compute_uv_false2(self, xp):
339338
array, full_matrices=self.full_matrices, compute_uv=False
340339
)
341340

342-
@_condition.repeat(3, 10)
341+
# @_condition.repeat(3, 10)
343342
def test_svd_rank4(self):
344343
self.check_usv((2, 2, 3, 4))
345344
self.check_usv((2, 2, 3, 7))
@@ -348,22 +347,22 @@ def test_svd_rank4(self):
348347
self.check_usv((2, 2, 4, 3))
349348
self.check_usv((2, 2, 32, 32))
350349

351-
@_condition.repeat(3, 10)
350+
# @_condition.repeat(3, 10)
352351
def test_svd_rank4_loop(self):
353352
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
354353
self.check_usv((3, 2, 64, 64))
355354
self.check_usv((3, 2, 64, 32))
356355
self.check_usv((3, 2, 32, 64))
357356

358-
@_condition.repeat(3, 10)
357+
# @_condition.repeat(3, 10)
359358
def test_svd_rank4_no_uv(self):
360359
self.check_singular((2, 2, 3, 4))
361360
self.check_singular((2, 2, 3, 7))
362361
self.check_singular((2, 2, 4, 4))
363362
self.check_singular((2, 2, 7, 3))
364363
self.check_singular((2, 2, 4, 3))
365364

366-
@_condition.repeat(3, 10)
365+
# @_condition.repeat(3, 10)
367366
def test_svd_rank4_no_uv_loop(self):
368367
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
369368
self.check_singular((3, 2, 64, 64))

0 commit comments

Comments
 (0)