MiSetProcessPartitionId

VOID __fastcall MiSetProcessPartitionId(INT64 a1, INT16 a2){
  *(_WORD *)(a1 + 1838) = a2;
}

Referenced by:

MmCreateProcessAddressSpace