hl_get_pb_block
block_num = hl_get_pb_block(hdev, mm_reg_addr + offset, pb_blocks,
block_num = hl_get_pb_block(hdev,