axi_registers_v2: fix build for rm.AWIDTH != 2