@@ -254,22 +254,21 @@ def check_singular(self, shape, xp, dtype):
254
254
assert (a == a_copy ).all ()
255
255
return result
256
256
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)
258
260
def test_svd_rank2 (self ):
259
261
self .check_usv ((3 , 7 ))
260
262
self .check_usv ((2 , 2 ))
261
263
self .check_usv ((7 , 3 ))
262
264
263
- @_condition .repeat (3 , 10 )
265
+ # @_condition.repeat(3, 10)
264
266
def test_svd_rank2_no_uv (self ):
265
267
self .check_singular ((3 , 7 ))
266
268
self .check_singular ((2 , 2 ))
267
269
self .check_singular ((7 , 3 ))
268
270
269
271
@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" )
273
272
def test_svd_rank2_empty_array (self ):
274
273
self .check_usv ((0 , 3 ))
275
274
self .check_usv ((3 , 0 ))
@@ -283,7 +282,7 @@ def test_svd_rank2_empty_array_compute_uv_false(self, xp):
283
282
array , full_matrices = self .full_matrices , compute_uv = False
284
283
)
285
284
286
- @_condition .repeat (3 , 10 )
285
+ # @_condition.repeat(3, 10)
287
286
def test_svd_rank3 (self ):
288
287
self .check_usv ((2 , 3 , 4 ))
289
288
self .check_usv ((2 , 3 , 7 ))
@@ -292,22 +291,22 @@ def test_svd_rank3(self):
292
291
self .check_usv ((2 , 4 , 3 ))
293
292
self .check_usv ((2 , 32 , 32 ))
294
293
295
- @_condition .repeat (3 , 10 )
294
+ # @_condition.repeat(3, 10)
296
295
def test_svd_rank3_loop (self ):
297
296
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
298
297
self .check_usv ((2 , 64 , 64 ))
299
298
self .check_usv ((2 , 64 , 32 ))
300
299
self .check_usv ((2 , 32 , 64 ))
301
300
302
- @_condition .repeat (3 , 10 )
301
+ # @_condition.repeat(3, 10)
303
302
def test_svd_rank3_no_uv (self ):
304
303
self .check_singular ((2 , 3 , 4 ))
305
304
self .check_singular ((2 , 3 , 7 ))
306
305
self .check_singular ((2 , 4 , 4 ))
307
306
self .check_singular ((2 , 7 , 3 ))
308
307
self .check_singular ((2 , 4 , 3 ))
309
308
310
- @_condition .repeat (3 , 10 )
309
+ # @_condition.repeat(3, 10)
311
310
def test_svd_rank3_no_uv_loop (self ):
312
311
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
313
312
self .check_singular ((2 , 64 , 64 ))
@@ -339,7 +338,7 @@ def test_svd_rank3_empty_array_compute_uv_false2(self, xp):
339
338
array , full_matrices = self .full_matrices , compute_uv = False
340
339
)
341
340
342
- @_condition .repeat (3 , 10 )
341
+ # @_condition.repeat(3, 10)
343
342
def test_svd_rank4 (self ):
344
343
self .check_usv ((2 , 2 , 3 , 4 ))
345
344
self .check_usv ((2 , 2 , 3 , 7 ))
@@ -348,22 +347,22 @@ def test_svd_rank4(self):
348
347
self .check_usv ((2 , 2 , 4 , 3 ))
349
348
self .check_usv ((2 , 2 , 32 , 32 ))
350
349
351
- @_condition .repeat (3 , 10 )
350
+ # @_condition.repeat(3, 10)
352
351
def test_svd_rank4_loop (self ):
353
352
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
354
353
self .check_usv ((3 , 2 , 64 , 64 ))
355
354
self .check_usv ((3 , 2 , 64 , 32 ))
356
355
self .check_usv ((3 , 2 , 32 , 64 ))
357
356
358
- @_condition .repeat (3 , 10 )
357
+ # @_condition.repeat(3, 10)
359
358
def test_svd_rank4_no_uv (self ):
360
359
self .check_singular ((2 , 2 , 3 , 4 ))
361
360
self .check_singular ((2 , 2 , 3 , 7 ))
362
361
self .check_singular ((2 , 2 , 4 , 4 ))
363
362
self .check_singular ((2 , 2 , 7 , 3 ))
364
363
self .check_singular ((2 , 2 , 4 , 3 ))
365
364
366
- @_condition .repeat (3 , 10 )
365
+ # @_condition.repeat(3, 10)
367
366
def test_svd_rank4_no_uv_loop (self ):
368
367
# This tests the loop-based batched gesvd on CUDA (_gesvd_batched)
369
368
self .check_singular ((3 , 2 , 64 , 64 ))
0 commit comments