KeSetThreadBamQosLevel

NTSTATUS __stdcall KeSetThreadBamQosLevel(INT64 a1, INT64 a2){
  NTSTATUS result; 
  signed __int32 v3; 

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

Referenced by:

KeSetThreadSchedulerAssist
KiConvertDynamicHeteroPolicy
PsImpersonateContainerOfThread