fixed the device architecture issue