Thank you for asking this question on the Microsoft Q&A Platform.
Are you receiving any error?
Please validate that when you are changing the disk type, the VM status must be "Stopped (deallocated)"
The premium disks are supported
Hope this helps,
Carlos Solís Salazar
Please "Accept as Answer" and/or Upvote if any of the above helped so that, it can help others in the community looking for remediation for similar issues.