AI disk partitioning