Files
Jungfraujoch/common/CUDAWrapper.h
2023-07-28 11:07:15 +02:00

13 lines
278 B
C++

// Copyright (2019-2023) Paul Scherrer Institute
// SPDX-License-Identifier: GPL-3.0-or-later
#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