@@ -505,8 +505,8 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
505
505
* Eventually we will have a fully 50% fragmented mm.
506
506
*/
507
507
508
- mm_size = PAGE_SIZE << max_order ;
509
- KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , PAGE_SIZE ),
508
+ mm_size = SZ_4K << max_order ;
509
+ KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , SZ_4K ),
510
510
"buddy_init failed\n" );
511
511
512
512
KUNIT_EXPECT_EQ (test , mm .max_order , max_order );
@@ -520,7 +520,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
520
520
}
521
521
522
522
for (order = top ; order -- ;) {
523
- size = get_size (order , PAGE_SIZE );
523
+ size = get_size (order , mm . chunk_size );
524
524
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start ,
525
525
mm_size , size , size ,
526
526
& tmp , flags ),
@@ -534,7 +534,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
534
534
}
535
535
536
536
/* There should be one final page for this sub-allocation */
537
- size = get_size (0 , PAGE_SIZE );
537
+ size = get_size (0 , mm . chunk_size );
538
538
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
539
539
size , size , & tmp , flags ),
540
540
"buddy_alloc hit -ENOMEM for hole\n" );
@@ -544,7 +544,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
544
544
545
545
list_move_tail (& block -> link , & holes );
546
546
547
- size = get_size (top , PAGE_SIZE );
547
+ size = get_size (top , mm . chunk_size );
548
548
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
549
549
size , size , & tmp , flags ),
550
550
"buddy_alloc unexpectedly succeeded at top-order %d/%d, it should be full!" ,
@@ -555,7 +555,7 @@ static void drm_test_buddy_alloc_pathological(struct kunit *test)
555
555
556
556
/* Nothing larger than blocks of chunk_size now available */
557
557
for (order = 1 ; order <= max_order ; order ++ ) {
558
- size = get_size (order , PAGE_SIZE );
558
+ size = get_size (order , mm . chunk_size );
559
559
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
560
560
size , size , & tmp , flags ),
561
561
"buddy_alloc unexpectedly succeeded at order %d, it should be full!" ,
@@ -584,14 +584,14 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
584
584
* page left.
585
585
*/
586
586
587
- mm_size = PAGE_SIZE << max_order ;
588
- KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , PAGE_SIZE ),
587
+ mm_size = SZ_4K << max_order ;
588
+ KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , SZ_4K ),
589
589
"buddy_init failed\n" );
590
590
591
591
KUNIT_EXPECT_EQ (test , mm .max_order , max_order );
592
592
593
593
for (order = 0 ; order < max_order ; order ++ ) {
594
- size = get_size (order , PAGE_SIZE );
594
+ size = get_size (order , mm . chunk_size );
595
595
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
596
596
size , size , & tmp , flags ),
597
597
"buddy_alloc hit -ENOMEM with order=%d\n" ,
@@ -604,7 +604,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
604
604
}
605
605
606
606
/* And now the last remaining block available */
607
- size = get_size (0 , PAGE_SIZE );
607
+ size = get_size (0 , mm . chunk_size );
608
608
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
609
609
size , size , & tmp , flags ),
610
610
"buddy_alloc hit -ENOMEM on final alloc\n" );
@@ -616,7 +616,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
616
616
617
617
/* Should be completely full! */
618
618
for (order = max_order ; order -- ;) {
619
- size = get_size (order , PAGE_SIZE );
619
+ size = get_size (order , mm . chunk_size );
620
620
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
621
621
size , size , & tmp , flags ),
622
622
"buddy_alloc unexpectedly succeeded, it should be full!" );
@@ -632,7 +632,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
632
632
list_del (& block -> link );
633
633
drm_buddy_free_block (& mm , block );
634
634
635
- size = get_size (order , PAGE_SIZE );
635
+ size = get_size (order , mm . chunk_size );
636
636
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
637
637
size , size , & tmp , flags ),
638
638
"buddy_alloc hit -ENOMEM with order=%d\n" ,
@@ -647,7 +647,7 @@ static void drm_test_buddy_alloc_pessimistic(struct kunit *test)
647
647
}
648
648
649
649
/* To confirm, now the whole mm should be available */
650
- size = get_size (max_order , PAGE_SIZE );
650
+ size = get_size (max_order , mm . chunk_size );
651
651
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
652
652
size , size , & tmp , flags ),
653
653
"buddy_alloc (realloc) hit -ENOMEM with order=%d\n" ,
@@ -678,15 +678,15 @@ static void drm_test_buddy_alloc_optimistic(struct kunit *test)
678
678
* try to allocate them all.
679
679
*/
680
680
681
- mm_size = PAGE_SIZE * ((1 << (max_order + 1 )) - 1 );
681
+ mm_size = SZ_4K * ((1 << (max_order + 1 )) - 1 );
682
682
683
- KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , PAGE_SIZE ),
683
+ KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_init (& mm , mm_size , SZ_4K ),
684
684
"buddy_init failed\n" );
685
685
686
686
KUNIT_EXPECT_EQ (test , mm .max_order , max_order );
687
687
688
688
for (order = 0 ; order <= max_order ; order ++ ) {
689
- size = get_size (order , PAGE_SIZE );
689
+ size = get_size (order , mm . chunk_size );
690
690
KUNIT_ASSERT_FALSE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
691
691
size , size , & tmp , flags ),
692
692
"buddy_alloc hit -ENOMEM with order=%d\n" ,
@@ -699,7 +699,7 @@ static void drm_test_buddy_alloc_optimistic(struct kunit *test)
699
699
}
700
700
701
701
/* Should be completely full! */
702
- size = get_size (0 , PAGE_SIZE );
702
+ size = get_size (0 , mm . chunk_size );
703
703
KUNIT_ASSERT_TRUE_MSG (test , drm_buddy_alloc_blocks (& mm , start , mm_size ,
704
704
size , size , & tmp , flags ),
705
705
"buddy_alloc unexpectedly succeeded, it should be full!" );
@@ -716,15 +716,15 @@ static void drm_test_buddy_alloc_limit(struct kunit *test)
716
716
LIST_HEAD (allocated );
717
717
struct drm_buddy mm ;
718
718
719
- KUNIT_EXPECT_FALSE (test , drm_buddy_init (& mm , size , PAGE_SIZE ));
719
+ KUNIT_EXPECT_FALSE (test , drm_buddy_init (& mm , size , SZ_4K ));
720
720
721
721
KUNIT_EXPECT_EQ_MSG (test , mm .max_order , DRM_BUDDY_MAX_ORDER ,
722
722
"mm.max_order(%d) != %d\n" , mm .max_order ,
723
723
DRM_BUDDY_MAX_ORDER );
724
724
725
725
size = mm .chunk_size << mm .max_order ;
726
726
KUNIT_EXPECT_FALSE (test , drm_buddy_alloc_blocks (& mm , start , size , size ,
727
- PAGE_SIZE , & allocated , flags ));
727
+ mm . chunk_size , & allocated , flags ));
728
728
729
729
block = list_first_entry_or_null (& allocated , struct drm_buddy_block , link );
730
730
KUNIT_EXPECT_TRUE (test , block );
@@ -734,10 +734,10 @@ static void drm_test_buddy_alloc_limit(struct kunit *test)
734
734
drm_buddy_block_order (block ), mm .max_order );
735
735
736
736
KUNIT_EXPECT_EQ_MSG (test , drm_buddy_block_size (& mm , block ),
737
- BIT_ULL (mm .max_order ) * PAGE_SIZE ,
737
+ BIT_ULL (mm .max_order ) * mm . chunk_size ,
738
738
"block size(%llu) != %llu\n" ,
739
739
drm_buddy_block_size (& mm , block ),
740
- BIT_ULL (mm .max_order ) * PAGE_SIZE );
740
+ BIT_ULL (mm .max_order ) * mm . chunk_size );
741
741
742
742
drm_buddy_free_list (& mm , & allocated , 0 );
743
743
drm_buddy_fini (& mm );
0 commit comments