论坛 › C#多线程 › 进程绑定到cpu逻辑内核上

Halcom 发表于 2022-12-14 16:53:02

进程绑定到cpu逻辑内核上

将线程绑定在某个具体的cpu逻辑内核上运行，https://blog.csdn.net/fm0517/article/details/125470768
using System.Runtime.InteropServices;
using System.Diagnostics;
using System.Management;

namespace BandThreadToCPU
{
    class Program
    {
      //SetThreadAffinityMask: Set hThread run on logical processer(LP:) dwThreadAffinityMask
      
      static extern UIntPtr SetThreadAffinityMask(IntPtr hThread, UIntPtr dwThreadAffinityMask);

      //Get the handler of current thread
      
      static extern IntPtr GetCurrentThread();

      //The real function
      public static void ChangeValue(object lpIdx)
      {
            //Bind current thread to a specific logical processer
            ulong LpId = SetCpuID((int)lpIdx);
            SetThreadAffinityMask(GetCurrentThread(), new UIntPtr(LpId));

            // Let program run for a while
            Stopwatch stopwatch = new Stopwatch();
            stopwatch.Start();
            for (int i = 0; i < 1000000; i++)
            {
                for (int j = 0; j < 1000000; j++)
                {
                  int _data = j;
                }
            }
            stopwatch.Stop();
            Console.WriteLine("Running Time: " + stopwatch.ElapsedMilliseconds.ToString());
      }

      //Get CPU id. index:0 -> id:1, 1->2, 2->4, 3->8, 4->16, ...
      static ulong SetCpuID(int lpIdx)
      {
            ulong cpuLogicalProcessorId = 0;
            if (lpIdx < 0 || lpIdx >= System.Environment.ProcessorCount)
            {
                lpIdx = 0;
            }
            cpuLogicalProcessorId |= 1UL << lpIdx;
            return cpuLogicalProcessorId;
      }

      static void Main(string[] args)
      {
            // Specify the cpu logical processor index you want. CPU lp index start from 0
            int LOGICAL_PROCESSOR_IDX = 4;

            // Get the ManagementClass object
            ManagementClass mc = new ManagementClass(new ManagementPath("Win32_Processor"));
            // Get the properties in the class
            ManagementObjectCollection moc = mc.GetInstances();

            foreach (ManagementObject mo in moc)
            {
                PropertyDataCollection properties = mo.Properties;
                // Print CPU properties
                foreach (PropertyData property in properties)
                {
                  Console.WriteLine( property.Name + ": " + property.Value);
                }
            }

            Stopwatch stopwatch = new Stopwatch();
            Thread thread = new Thread(new ParameterizedThreadStart(ChangeValue));
            stopwatch.Start();
            thread.Start(LOGICAL_PROCESSOR_IDX);
            thread.Join();
            stopwatch.Stop();
            Console.WriteLine("Total running time: " + stopwatch.ElapsedMilliseconds.ToString());
            Console.ReadKey();
      }
    }
}

查看完整版本: 进程绑定到cpu逻辑内核上