pci_id.hpp

Go to the documentation of this file.

#pragma once
#ifndef CUDA_API_WRAPPERS_PCI_ID_CUH_
#define CUDA_API_WRAPPERS_PCI_ID_CUH_

#include <cuda/api/types.hpp>
#include <cuda/api/error.hpp>

#include <cuda_runtime_api.h>

#include <string>

namespace cuda {
namespace device {

struct pci_location_t {
    // These are the values CUDA's API provides us with directly
    int domain;
    int bus;
    int device;
    int function;

    operator ::std::string() const;
    // This is not a ctor so as to maintain the PODness
    static pci_location_t parse(const ::std::string& id_str);
    static pci_location_t parse(const char* id_str);
public:
    static constexpr const int unused { -1 };
        // In lieu of making this class a variant with 3 type combinations.
};

namespace detail_ {

inline id_t resolve_id(pci_location_t pci_id)
{
    ::std::string as_string { pci_id };
    id_t cuda_device_id;
    auto result = cuDeviceGetByPCIBusId(&cuda_device_id, as_string.c_str());
    throw_if_error(result,
        "Failed obtaining a CUDA device ID corresponding to PCI id " + as_string);
    return cuda_device_id;
}

} // namespace detail_


} // namespace device
} // namespace cuda

#endif // CUDA_API_WRAPPERS_PCI_ID_HPP_