180 	struct hns_roce_mr_table *mr_table = &hr_dev->mr_table;  in hns_roce_alloc_mtt_range()  local
187 		buddy = &mr_table->mtt_buddy;  in hns_roce_alloc_mtt_range()
188 		table = &mr_table->mtt_table;  in hns_roce_alloc_mtt_range()
191 		buddy = &mr_table->mtt_cqe_buddy;  in hns_roce_alloc_mtt_range()
192 		table = &mr_table->mtt_cqe_table;  in hns_roce_alloc_mtt_range()
195 		buddy = &mr_table->mtt_srqwqe_buddy;  in hns_roce_alloc_mtt_range()
196 		table = &mr_table->mtt_srqwqe_table;  in hns_roce_alloc_mtt_range()
199 		buddy = &mr_table->mtt_idx_buddy;  in hns_roce_alloc_mtt_range()
200 		table = &mr_table->mtt_idx_table;  in hns_roce_alloc_mtt_range()
253 	struct hns_roce_mr_table *mr_table = &hr_dev->mr_table;  in hns_roce_mtt_cleanup()  local
260 		hns_roce_buddy_free(&mr_table->mtt_buddy, mtt->first_seg,  in hns_roce_mtt_cleanup()
262 		hns_roce_table_put_range(hr_dev, &mr_table->mtt_table,  in hns_roce_mtt_cleanup()
267 		hns_roce_buddy_free(&mr_table->mtt_cqe_buddy, mtt->first_seg,  in hns_roce_mtt_cleanup()
269 		hns_roce_table_put_range(hr_dev, &mr_table->mtt_cqe_table,  in hns_roce_mtt_cleanup()
274 		hns_roce_buddy_free(&mr_table->mtt_srqwqe_buddy, mtt->first_seg,  in hns_roce_mtt_cleanup()
276 		hns_roce_table_put_range(hr_dev, &mr_table->mtt_srqwqe_table,  in hns_roce_mtt_cleanup()
281 		hns_roce_buddy_free(&mr_table->mtt_idx_buddy, mtt->first_seg,  in hns_roce_mtt_cleanup()
283 		hns_roce_table_put_range(hr_dev, &mr_table->mtt_idx_table,  in hns_roce_mtt_cleanup()
580 	ret = hns_roce_bitmap_alloc(&hr_dev->mr_table.mtpt_bitmap, &index);  in hns_roce_mr_alloc()
729 		hns_roce_table_put(hr_dev, &hr_dev->mr_table.mtpt_table,  in hns_roce_mr_free()
732 	hns_roce_bitmap_free(&hr_dev->mr_table.mtpt_bitmap,  in hns_roce_mr_free()
743 	struct hns_roce_mr_table *mr_table = &hr_dev->mr_table;  in hns_roce_mr_enable()  local
746 	ret = hns_roce_table_get(hr_dev, &mr_table->mtpt_table, mtpt_idx);  in hns_roce_mr_enable()
782 	hns_roce_table_put(hr_dev, &mr_table->mtpt_table, mtpt_idx);  in hns_roce_mr_enable()
798 		table = &hr_dev->mr_table.mtt_table;  in hns_roce_write_mtt_chunk()
802 		table = &hr_dev->mr_table.mtt_cqe_table;  in hns_roce_write_mtt_chunk()
806 		table = &hr_dev->mr_table.mtt_srqwqe_table;  in hns_roce_write_mtt_chunk()
810 		table = &hr_dev->mr_table.mtt_idx_table;  in hns_roce_write_mtt_chunk()
917 	struct hns_roce_mr_table *mr_table = &hr_dev->mr_table;  in hns_roce_init_mr_table()  local
920 	ret = hns_roce_bitmap_init(&mr_table->mtpt_bitmap,  in hns_roce_init_mr_table()
927 	ret = hns_roce_buddy_init(&mr_table->mtt_buddy,  in hns_roce_init_mr_table()
933 		ret = hns_roce_buddy_init(&mr_table->mtt_cqe_buddy,  in hns_roce_init_mr_table()
940 		ret = hns_roce_buddy_init(&mr_table->mtt_srqwqe_buddy,  in hns_roce_init_mr_table()
947 		ret = hns_roce_buddy_init(&mr_table->mtt_idx_buddy,  in hns_roce_init_mr_table()
957 		hns_roce_buddy_cleanup(&mr_table->mtt_srqwqe_buddy);  in hns_roce_init_mr_table()
961 		hns_roce_buddy_cleanup(&mr_table->mtt_cqe_buddy);  in hns_roce_init_mr_table()
964 	hns_roce_buddy_cleanup(&mr_table->mtt_buddy);  in hns_roce_init_mr_table()
967 	hns_roce_bitmap_cleanup(&mr_table->mtpt_bitmap);  in hns_roce_init_mr_table()
973 	struct hns_roce_mr_table *mr_table = &hr_dev->mr_table;  in hns_roce_cleanup_mr_table()  local
976 		hns_roce_buddy_cleanup(&mr_table->mtt_idx_buddy);  in hns_roce_cleanup_mr_table()
978 		hns_roce_buddy_cleanup(&mr_table->mtt_srqwqe_buddy);  in hns_roce_cleanup_mr_table()
979 	hns_roce_buddy_cleanup(&mr_table->mtt_buddy);  in hns_roce_cleanup_mr_table()
981 		hns_roce_buddy_cleanup(&mr_table->mtt_cqe_buddy);  in hns_roce_cleanup_mr_table()
982 	hns_roce_bitmap_cleanup(&mr_table->mtpt_bitmap);  in hns_roce_cleanup_mr_table()
1456 		hns_roce_table_put(hr_dev, &hr_dev->mr_table.mtpt_table,  in hns_roce_mw_free()
1460 	hns_roce_bitmap_free(&hr_dev->mr_table.mtpt_bitmap,  in hns_roce_mw_free()
1467 	struct hns_roce_mr_table *mr_table = &hr_dev->mr_table;  in hns_roce_mw_enable()  local
1474 	ret = hns_roce_table_get(hr_dev, &mr_table->mtpt_table, mtpt_idx);  in hns_roce_mw_enable()
1507 	hns_roce_table_put(hr_dev, &mr_table->mtpt_table, mtpt_idx);  in hns_roce_mw_enable()
1525 	ret = hns_roce_bitmap_alloc(&hr_dev->mr_table.mtpt_bitmap, &index);  in hns_roce_alloc_mw()