@@ -329,8 +329,8 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
329
329
* Eventually we will have a fully 50% fragmented mm.
330
330
*/
331
331
332
- mm_size = PAGE_SIZE << max_order ;
333
- KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , PAGE_SIZE ),
332
+ mm_size = SZ_4K << max_order ;
333
+ KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , SZ_4K ),
334
334
"buddy_init failed\n" );
335
335
336
336
KUNIT_EXPECT_EQ (test , mm .max_order , max_order );
@@ -344,7 +344,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
344
344
}
345
345
346
346
for (order = top ; order -- ;) {
347
- size = get_size (order , PAGE_SIZE );
347
+ size = get_size (order , mm . chunk_size );
348
348
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start ,
349
349
mm_size , size , size ,
350
350
& tmp , flags ),
@@ -358,7 +358,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
358
358
}
359
359
360
360
/* There should be one final page for this sub-allocation */
361
- size = get_size (0 , PAGE_SIZE );
361
+ size = get_size (0 , mm . chunk_size );
362
362
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
363
363
size , size , & tmp , flags ),
364
364
"buddy_alloc hit -ENOMEM for hole\n" );
@@ -368,7 +368,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
368
368
369
369
list_move_tail (& block -> link , & holes );
370
370
371
- size = get_size (top , PAGE_SIZE );
371
+ size = get_size (top , mm . chunk_size );
372
372
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
373
373
size , size , & tmp , flags ),
374
374
"buddy_alloc unexpectedly succeeded at top-order %d/%d, it should be full!" ,
@@ -379,7 +379,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
379
379
380
380
/* Nothing larger than blocks of chunk_size now available */
381
381
for (order = 1 ; order <= max_order ; order ++ ) {
382
- size = get_size (order , PAGE_SIZE );
382
+ size = get_size (order , mm . chunk_size );
383
383
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
384
384
size , size , & tmp , flags ),
385
385
"buddy_alloc unexpectedly succeeded at order %d, it should be full!" ,
@@ -408,14 +408,14 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
408
408
* page left.
409
409
*/
410
410
411
- mm_size = PAGE_SIZE << max_order ;
412
- KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , PAGE_SIZE ),
411
+ mm_size = SZ_4K << max_order ;
412
+ KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , SZ_4K ),
413
413
"buddy_init failed\n" );
414
414
415
415
KUNIT_EXPECT_EQ (test , mm .max_order , max_order );
416
416
417
417
for (order = 0 ; order < max_order ; order ++ ) {
418
- size = get_size (order , PAGE_SIZE );
418
+ size = get_size (order , mm . chunk_size );
419
419
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
420
420
size , size , & tmp , flags ),
421
421
"buddy_alloc hit -ENOMEM with order=%d\n" ,
@@ -428,7 +428,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
428
428
}
429
429
430
430
/* And now the last remaining block available */
431
- size = get_size (0 , PAGE_SIZE );
431
+ size = get_size (0 , mm . chunk_size );
432
432
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
433
433
size , size , & tmp , flags ),
434
434
"buddy_alloc hit -ENOMEM on final alloc\n" );
@@ -440,7 +440,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
440
440
441
441
/* Should be completely full! */
442
442
for (order = max_order ; order -- ;) {
443
- size = get_size (order , PAGE_SIZE );
443
+ size = get_size (order , mm . chunk_size );
444
444
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
445
445
size , size , & tmp , flags ),
446
446
"buddy_alloc unexpectedly succeeded, it should be full!" );
@@ -456,7 +456,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
456
456
list_del (& block -> link );
457
457
drm_buddy_free_block (& mm , block );
458
458
459
- size = get_size (order , PAGE_SIZE );
459
+ size = get_size (order , mm . chunk_size );
460
460
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
461
461
size , size , & tmp , flags ),
462
462
"buddy_alloc hit -ENOMEM with order=%d\n" ,
@@ -471,7 +471,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
471
471
}
472
472
473
473
/* To confirm, now the whole mm should be available */
474
- size = get_size (max_order , PAGE_SIZE );
474
+ size = get_size (max_order , mm . chunk_size );
475
475
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
476
476
size , size , & tmp , flags ),
477
477
"buddy_alloc (realloc) hit -ENOMEM with order=%d\n" ,
@@ -502,15 +502,15 @@ static void drm_test_buddy_alloc_optimistic(struct kunit *test)
502
502
* try to allocate them all.
503
503
*/
504
504
505
- mm_size = PAGE_SIZE * ((1 << (max_order + 1 )) - 1 );
505
+ mm_size = SZ_4K * ((1 << (max_order + 1 )) - 1 );
506
506
507
- KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , PAGE_SIZE ),
507
+ KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , SZ_4K ),
508
508
"buddy_init failed\n" );
509
509
510
510
KUNIT_EXPECT_EQ (test , mm .max_order , max_order );
511
511
512
512
for (order = 0 ; order <= max_order ; order ++ ) {
513
- size = get_size (order , PAGE_SIZE );
513
+ size = get_size (order , mm . chunk_size );
514
514
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
515
515
size , size , & tmp , flags ),
516
516
"buddy_alloc hit -ENOMEM with order=%d\n" ,
@@ -523,7 +523,7 @@ static void drm_test_buddy_alloc_optimistic(struct kunit *test)
523
523
}
524
524
525
525
/* Should be completely full! */
526
- size = get_size (0 , PAGE_SIZE );
526
+ size = get_size (0 , mm . chunk_size );
527
527
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
528
528
size , size , & tmp , flags ),
529
529
"buddy_alloc unexpectedly succeeded, it should be full!" );
@@ -540,15 +540,15 @@ static void drm_test_buddy_alloc_limit(struct kunit *test)
540
540
LIST_HEAD (allocated );
541
541
struct drm_buddy mm ;
542
542
543
- KUNIT_EXPECT_FALSE (test , drm_buddy_init (& mm , size , PAGE_SIZE ));
543
+ KUNIT_EXPECT_FALSE (test , drm_buddy_init (& mm , size , SZ_4K ));
544
544
545
545
KUNIT_EXPECT_EQ_MSG (test , mm .max_order , DRM_BUDDY_MAX_ORDER ,
546
546
"mm.max_order(%d) != %d\n" , mm .max_order ,
547
547
DRM_BUDDY_MAX_ORDER );
548
548
549
549
size = mm .chunk_size << mm .max_order ;
550
550
KUNIT_EXPECT_FALSE (test , drm_buddy_alloc_blocks (& mm , start , size , size ,
551
- PAGE_SIZE , & allocated , flags ));
551
+ mm . chunk_size , & allocated , flags ));
552
552
553
553
block = list_first_entry_or_null (& allocated , struct drm_buddy_block , link );
554
554
KUNIT_EXPECT_TRUE (test , block );
@@ -558,10 +558,10 @@ static void drm_test_buddy_alloc_limit(struct kunit *test)
558
558
drm_buddy_block_order (block ), mm .max_order );
559
559
560
560
KUNIT_EXPECT_EQ_MSG (test , drm_buddy_block_size (& mm , block ),
561
- BIT_ULL (mm .max_order ) * PAGE_SIZE ,
561
+ BIT_ULL (mm .max_order ) * mm . chunk_size ,
562
562
"block size(%llu) != %llu\n" ,
563
563
drm_buddy_block_size (& mm , block ),
564
- BIT_ULL (mm .max_order ) * PAGE_SIZE );
564
+ BIT_ULL (mm .max_order ) * mm . chunk_size );
565
565
566
566
drm_buddy_free_list (& mm , & allocated );
567
567
drm_buddy_fini (& mm );
0 commit comments