That would be called under-provisioning.
I haven’t read anything about how an SSD deals with partitions, so I don’t know for sure.
Since the controller intercepts the calls for specific locations, I’m inclined to believe that the controller does not care about the concept of partitions and does not segregate any chips, thus it would spread all writes across all of the chips.
From the drives I have seen, usually there are 3 write-cache sizes.
Usually the smallest write-cache is for drives 128GB or smaller. Sometimes the 256GB is also here.
Usually the middle size write-cache is for 512GB and sometimes 256GB drives.
Usually the largest write-cache is only in 1TB and bigger drives.
Performance-wise for writes, you want the biggest write cache, so you want at least a 1TB drive.
For the best wear leveling, you want the drive as big as you can afford, while also looking at the makeup of the memory chips. In order of longest lasting listed first: Single Level, Multi Level, Triple Level, Quad Level.