IB/umad: Add P_Key index support