cuda.h

Go to the documentation of this file.

 
#pragma once
 
 
#if PROMEKI_ENABLE_CORE
#include <promeki/namespace.h>
#include <promeki/config.h>
#include <promeki/string.h>
#include <promeki/error.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class CudaDevice {
        public:
                CudaDevice() = default;
 
                explicit CudaDevice(int ordinal);
 
                static bool isAvailable();
 
                static int deviceCount();
 
                static CudaDevice current();
 
                static Error setCurrent(int ordinal);
 
                bool isValid() const { return _ordinal >= 0; }
 
                int ordinal() const { return _ordinal; }
 
                const String &name() const { return _name; }
 
                size_t totalMemory() const { return _totalMem; }
 
                int computeMajor() const { return _ccMajor; }
 
                int computeMinor() const { return _ccMinor; }
 
        private:
                int    _ordinal = -1;
                String _name;
                size_t _totalMem = 0;
                int    _ccMajor = 0;
                int    _ccMinor = 0;
};
 
class CudaBootstrap {
        public:
                static Error ensureRegistered();
 
                static bool isRegistered();
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_CORE