bool cache_snoop;
        unsigned int ctx_offset_base;
        unsigned int ctx_offset_stride;
+       unsigned int utlb_offset_base;
 };
 
 struct ipmmu_vmsa_device {
 
 static u32 ipmmu_utlb_reg(struct ipmmu_vmsa_device *mmu, unsigned int reg)
 {
-       return reg;
+       return mmu->features->utlb_offset_base + reg;
 }
 
 static void ipmmu_imuasid_write(struct ipmmu_vmsa_device *mmu,
        .cache_snoop = true,
        .ctx_offset_base = 0,
        .ctx_offset_stride = 0x40,
+       .utlb_offset_base = 0,
 };
 
 static const struct ipmmu_features ipmmu_features_rcar_gen3 = {
        .cache_snoop = false,
        .ctx_offset_base = 0,
        .ctx_offset_stride = 0x40,
+       .utlb_offset_base = 0,
 };
 
 static const struct of_device_id ipmmu_of_ids[] = {