// Copyright (2019-2023) Paul Scherrer Institute

#include "CUDAWrapper.h"
#include "JFJochException.h"

inline void cuda_err(cudaError_t val) {
    if (val != cudaSuccess)
        throw JFJochException(JFJochExceptionCategory::GPUCUDAError, cudaGetErrorString(val));
}

int32_t get_gpu_count() {
    int device_count;
    cuda_err(cudaGetDeviceCount(&device_count));
    return device_count;
}

void set_gpu(int32_t dev_id) {
    auto dev_count = get_gpu_count();

    if ((dev_id < 0) || (dev_id >= dev_count))
        throw JFJochException(JFJochExceptionCategory::InputParameterInvalid, "Device ID cannot be negative");

    cuda_err(cudaSetDevice(dev_id));
}