PspSetUmsThreadContext

INT64 __fastcall PspSetUmsThreadContext(INT64 a1, INT64 a2, _DWORD *a3){
  unsigned int v3; 
  v3 = 0;
  if( (*a3 & 2) != 0 )
    return(unsigned int)KeCopyContextToUmsContext(**(_QWORD **)(a1 + 496), a2);
  else
    KeCopyContextToUch(*(_QWORD *)(a1 + 504), a2);
  return v3;
}

Referenced by:

PspDisassociateUmsThreadFromPrimary
PspGetSetContextInternal