Question about WD Black SN850P with Heatsink for PS5

If the interface is the same, this should work without any problems. You may also need to check whether the WD is not "too fast" for your PC in the sense of paying too much for performance that you cannot use.