hamr_hip_device.cxx

#include "hamr_hip_device.h"

#include <iostream>

#include <hip/hip_runtime.h>


namespace hamr
{

// **************************************************************************
int get_hip_device(const void *ptr, int &device_id)
{
    hipError_t ierr = hipSuccess;
    hipPointerAttribute_t ptrAtts;
    ierr = hipPointerGetAttributes(&ptrAtts, ptr);

    // TODO -- HIP doesn;t yet have this feature of CUDA
    // these types of pointers are NOT accessible on the GPU
    // hipErrorInValue occurs when the pointer is unknown to HIP, as is
    // the case with pointers allocated by malloc or new.
    /*if ((ierr == hipErrorInvalidValue) ||
        ((ierr == hipSuccess) && ((ptrAtts.type == hipMemoryTypeHost) ||
        (ptrAtts.type == hipMemoryTypeUnregistered))))
    {
        // this is host backed memory not associate with a GPU
        device_id = -1;
    }
    else*/ if (ierr != hipSuccess)
    {
        std::cerr << "[" << __FILE__ << ":" << __LINE__ << "] ERROR:"
            " Failed to get pointer attributes for " << ptr << std::endl;
        return -1;
    }
    else
    {
        device_id = ptrAtts.device;
    }

    return 0;
}

// **************************************************************************
int get_active_hip_device(int &dev_id)
{
    hipError_t ierr = hipSuccess;
    if ((ierr = hipGetDevice(&dev_id)) != hipSuccess)
    {
        std::cerr << "Failed to get the active HIP device. "
            << hipGetErrorString(ierr) << std::endl;
        return -1;
    }

    return 0;
}

// **************************************************************************
int set_active_hip_device(int dev_id)
{
    hipError_t ierr = hipSuccess;
    if ((ierr = hipSetDevice(dev_id)) != hipSuccess)
    {
        std::cerr << "Failed to set the active HIP device. "
            << hipGetErrorString(ierr) << std::endl;
        return -1;
    }

    return 0;
}


// --------------------------------------------------------------------------
activate_hip_device::activate_hip_device(int new_dev) : m_device(-1)
{
    int cur_dev = -1;
    if (!get_active_hip_device(cur_dev) && (cur_dev != new_dev) &&
        !set_active_hip_device(new_dev))
    {
        m_device = cur_dev;
    }
}

// --------------------------------------------------------------------------
activate_hip_device::~activate_hip_device()
{
    if (m_device >= 0)
    {
        set_active_hip_device(m_device);
    }
}

}