Optimize fe_mul register usage