util.hpp

Go to the documentation of this file.

#pragma once
#include "defs.hpp"
#include <cuda_runtime.h>

namespace bnmf_algs {
namespace cuda {
template <typename Integer> void init(Integer device) {
    auto err = cudaSetDevice(device);
    BNMF_ASSERT(err == cudaSuccess, "Error setting CUDA device in cuda::init");
}

template <typename Integer> Integer idiv_ceil(Integer a, Integer b) {
    return a / b + (a % b != 0);
}
} // namespace cuda
} // namespace bnmf_algs