// Copyright (2019-2023) Paul Scherrer Institute

#ifndef JUNGFRAUJOCH_CUDAWRAPPER_H
#define JUNGFRAUJOCH_CUDAWRAPPER_H

#include <cstdint>

int32_t get_gpu_count();
void set_gpu(int32_t dev_id);

#endif //JUNGFRAUJOCH_CUDAWRAPPER_H