Class GPUDeviceInfo

Inheritance Relationships

Base Type

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.