Files
Jungfraujoch/common/CUDAWrapper.h

12 lines
233 B
C++

// 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