iommu: Use right function to get group for device