KeSetThreadBamQosLevel

INT64 __fastcall KeSetThreadBamQosLevel(INT64 a1, INT64 a2){
  INT64 result; 
  signed __int32 v3; 
  do
  {
    v3 = *(_DWORD *)(a1 + 512);
    result = (unsigned int)_InterlockedCompareExchange((volatile signed __int32 *)(a1 + 512), a2 | v3 & 0xFFFFFF00, v3);
  }
  while( v3 != (_DWORD)result );
  return result;
}

Referenced by:

KeSetThreadSchedulerAssist
KiConvertDynamicHeteroPolicy
PsImpersonateContainerOfThread