Asteroid belt survival through stellar evolution: dependence on the stellar mass