@@ -300,7 +300,7 @@ def check_get_preferred_vector_width_half(device):
300
300
def check_create_sub_devices_equally (device ):
301
301
try :
302
302
n = int (device .max_compute_units / 2 )
303
- device .create_sub_devices (n )
303
+ device .create_sub_devices (partition = n )
304
304
except SubDeviceCreationError :
305
305
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
306
306
except Exception :
@@ -310,7 +310,7 @@ def check_create_sub_devices_equally(device):
310
310
def check_create_sub_devices_by_counts (device ):
311
311
try :
312
312
n = device .max_compute_units / 2
313
- device .create_sub_devices ([ n , n ] )
313
+ device .create_sub_devices (partition = ( n , n ) )
314
314
except SubDeviceCreationError :
315
315
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
316
316
except Exception :
@@ -319,7 +319,7 @@ def check_create_sub_devices_by_counts(device):
319
319
320
320
def check_create_sub_devices_by_affinity_not_applicable (device ):
321
321
try :
322
- device .create_sub_devices ("not_applicable" )
322
+ device .create_sub_devices (partition = "not_applicable" )
323
323
except SubDeviceCreationError :
324
324
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
325
325
except Exception :
@@ -328,7 +328,7 @@ def check_create_sub_devices_by_affinity_not_applicable(device):
328
328
329
329
def check_create_sub_devices_by_affinity_numa (device ):
330
330
try :
331
- device .create_sub_devices ("numa" )
331
+ device .create_sub_devices (partition = "numa" )
332
332
except SubDeviceCreationError :
333
333
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
334
334
except Exception :
@@ -337,7 +337,7 @@ def check_create_sub_devices_by_affinity_numa(device):
337
337
338
338
def check_create_sub_devices_by_affinity_L4_cache (device ):
339
339
try :
340
- device .create_sub_devices ("L4_cache" )
340
+ device .create_sub_devices (partition = "L4_cache" )
341
341
except SubDeviceCreationError :
342
342
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
343
343
except Exception :
@@ -346,7 +346,7 @@ def check_create_sub_devices_by_affinity_L4_cache(device):
346
346
347
347
def check_create_sub_devices_by_affinity_L3_cache (device ):
348
348
try :
349
- device .create_sub_devices ("L3_cache" )
349
+ device .create_sub_devices (partition = "L3_cache" )
350
350
except SubDeviceCreationError :
351
351
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
352
352
except Exception :
@@ -355,7 +355,7 @@ def check_create_sub_devices_by_affinity_L3_cache(device):
355
355
356
356
def check_create_sub_devices_by_affinity_L2_cache (device ):
357
357
try :
358
- device .create_sub_devices ("L2_cache" )
358
+ device .create_sub_devices (partition = "L2_cache" )
359
359
except SubDeviceCreationError :
360
360
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
361
361
except Exception :
@@ -364,7 +364,7 @@ def check_create_sub_devices_by_affinity_L2_cache(device):
364
364
365
365
def check_create_sub_devices_by_affinity_L1_cache (device ):
366
366
try :
367
- device .create_sub_devices ("L1_cache" )
367
+ device .create_sub_devices (partition = "L1_cache" )
368
368
except SubDeviceCreationError :
369
369
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
370
370
except Exception :
@@ -373,7 +373,7 @@ def check_create_sub_devices_by_affinity_L1_cache(device):
373
373
374
374
def check_create_sub_devices_by_affinity_next_partitionable (device ):
375
375
try :
376
- device .create_sub_devices ("next_partitionable" )
376
+ device .create_sub_devices (partition = "next_partitionable" )
377
377
except SubDeviceCreationError :
378
378
pytest .skip ("create_sub_devices can't create sub-devices on this device" )
379
379
except Exception :
0 commit comments