GpuIds.cpp

#include "GpuIds.hpp"
#include <stdlib.h>
#include <string.h>
#include <cuda_runtime_api.h>

GpuIds::~GpuIds() {
    free(m_piDeviceIds); m_piDeviceIds = nullptr;
    m_iCount = 0;
}
GpuIds::GpuIds() : m_piDeviceIds (nullptr), m_iCount(0) {

}
void GpuIds::SetIds(int iCount, int* piDeviceIds) {
    if (iCount > 0 && piDeviceIds != 0) {
        if (m_piDeviceIds) {
            free(m_piDeviceIds); m_piDeviceIds = nullptr;
            m_iCount = 0;
        }
        m_piDeviceIds = (int*)malloc(iCount * sizeof(int));
        if (m_piDeviceIds) {
            for (int iI = 0; iI < iCount; ++iI) {
                m_piDeviceIds[iI] = piDeviceIds[iI];
            }
            m_iCount = iCount;
        }
    }
}

int GpuIds::GetLength() const {
    return m_iCount;
}
int& GpuIds::operator[](int iIndex){
    return m_piDeviceIds[iIndex];
}
int GpuIds::operator[](int iIndex) const {
    return m_piDeviceIds[iIndex];
}

void GpuIds::SetAllGpus(int iTotalDeviceCount) {
    // Set all GPUs for compatibility
    // Makeup valid GpuIds.
    int* aiIds = nullptr;
    if (iTotalDeviceCount == 0) {
        (int*)malloc(iTotalDeviceCount*sizeof(int));
        for (int iI = 0; iI < iTotalDeviceCount; ++iI) {
            aiIds[iI] = iI;
        }
    }
    SetIds(iTotalDeviceCount, aiIds);
    free(aiIds); aiIds = 0;    
}

bool GpuIds::AreEqualDevices() const {
    int deviceCount = this->GetLength();
    const int devicenamelength = 256;  // The length 256 is fixed by spec of cudaDeviceProp::name
    char devicename[devicenamelength];
    cudaDeviceProp deviceProp;
    for (int dev = 0; dev < deviceCount; dev++) {
        // cudaSetDevice(m_piDeviceIds[dev]);
        cudaGetDeviceProperties(&deviceProp, m_piDeviceIds[dev]);
        if (dev>0) {
            if (strcmp(devicename, deviceProp.name) != 0) {
                return false;
            }
        }
        memset(devicename, 0, devicenamelength);
        strcpy(devicename, deviceProp.name);
    }
    return true;
}