RunWithAffinity Class Reference

#include <GlobalGPUMgr.h>

Public Member Functions
	RunWithAffinity (const bool valid, const int tempCpu)
	~RunWithAffinity ()

Static Public Attributes
static constexpr int	kCudaOffset = 0
static constexpr int	kNvshmemOffset = 1
static constexpr int	kNcclOffset = 2

Detailed Description

Definition at line 30 of file GlobalGPUMgr.h.

Constructor & Destructor Documentation

RunWithAffinity()

RunWithAffinity::RunWithAffinity	(	const bool	valid,
		const int	tempCpu
	)

Definition at line 20 of file GlobalGPUMgr.C.

                                                                    {
  validCpu = valid && (tempCpu >= 0) && (tempCpu < CmiNumCores());

  if (validCpu) {
    current = CPU_ALLOC(CmiNumCores());
    temp    = CPU_ALLOC(CmiNumCores());
    size    = CPU_ALLOC_SIZE(CmiNumCores());

    CPU_ZERO_S(size, current);
    CPU_ZERO_S(size, temp);

    pthread_getaffinity_np(pthread_self(), size, current);

    CPU_SET_S(tempCpu, size, temp);
    pthread_setaffinity_np(pthread_self(), size, temp);
  }
}

~RunWithAffinity()

RunWithAffinity::~RunWithAffinity

(

)

Definition at line 38 of file GlobalGPUMgr.C.

                                  {
  if (validCpu) {
    pthread_setaffinity_np(pthread_self(), size, current);
  
    CPU_FREE(current);
    CPU_FREE(temp);
  }
}

Member Data Documentation

kCudaOffset

constexpr int RunWithAffinity::kCudaOffset = 0

static

Definition at line 36 of file GlobalGPUMgr.h.

kNcclOffset

constexpr int RunWithAffinity::kNcclOffset = 2

static

Definition at line 38 of file GlobalGPUMgr.h.

kNvshmemOffset

constexpr int RunWithAffinity::kNvshmemOffset = 1

static

Definition at line 37 of file GlobalGPUMgr.h.

---

The documentation for this class was generated from the following files:

• GlobalGPUMgr.h
• GlobalGPUMgr.C