iommu/amd: Fix bug in put_pasid_state_wait