Class GPUDeviceInfo
Defined in File context.h
Inheritance Relationships
Base Type
public mindspore::DeviceInfoContext
(Class DeviceInfoContext)
Class Documentation
-
class GPUDeviceInfo : public mindspore::DeviceInfoContext
Derived from DeviceInfoContext, The configuration of the model running on the GPU.
Public Functions
-
inline virtual enum DeviceType GetDeviceType() const override
Get the type of this DeviceInfoContext.
- Returns
Type of this DeviceInfoContext.
-
void SetDeviceID(uint32_t device_id)
Set device id.
- Parameters
device_id – [in] The device id.
-
uint32_t GetDeviceID() const
Get the device id.
- Returns
The device id.
-
int GetRankID() const
Get the distribution rank id.
- Returns
The device id.
-
int GetGroupSize() const
Get the distribution group size.
- Returns
The device id.
-
inline void SetPrecisionMode(const std::string &precision_mode)
Set the precision mode.
- Parameters
precision_mode – [in] Optional “origin”, “fp16”. “origin” is set as default.
-
inline std::string GetPrecisionMode() const
Get the precision mode.
- Returns
The precision mode.
-
void SetEnableFP16(bool is_fp16)
Set enables to perform the float16 inference.
- Parameters
is_fp16 – [in] Enable float16 inference or not.
-
bool GetEnableFP16() const
Get enables to perform the float16 inference.
- Returns
Whether enable float16 inference.
-
void SetEnableGLTexture(bool is_enable_gl_texture)
Set enables to sharing mem with OpenGL.
- Parameters
is_enable_gl_texture – [in] Enable sharing OpenCL Memory with OpenGL or not.
-
bool GetEnableGLTexture() const
Get enables to sharing mem with OpenGL.
- Returns
Whether enable sharing mem with OpenGL.
-
void SetGLContext(void *gl_context)
Set current OpenGL context.
- Parameters
gl_context – [in] Current OpenGL context.
-
void *GetGLContext() const
Get current OpenGL context.
- Returns
the OpenCL context by OpenGL used.
-
void SetGLDisplay(void *gl_display)
Set current OpenGL display.
- Parameters
gl_display – [in] Current OpenGL display.
-
void *GetGLDisplay() const
Get current OpenGL display.
- Returns
the OpenCL display by OpenGL used.
-
inline virtual enum DeviceType GetDeviceType() const override