mindspore.device_context
The device_context encapsulates the interface for querying the number of devices with whether the currently specified backend is available.
Cpu Device Backend Management
Returns the compute-capable device count of CPU. |
|
Returns whether the CPU backend is available. |
|
Set the number of threads for CPU kernel used. |
Gpu Device Backend Management
Returns the number of GPUs available. |
|
Returns a bool indicating if CUDA is currently available. |
|
Whether to convert FP32 to TF32 for Conv operators. |
|
Whether to convert FP32 to TF32 for Matmul operators. |
|
Specifies convolution data grad algorithm. |
|
Specifies convolution forward algorithm. |
|
Specifies convolution filter grad algorithm. |
Ascend Device Backend Management
Returns the compute-capable device count of Ascend. |
|
Returns whether the Ascend backend is available. |
|
|
Whether to convert FP32 to HF32 for Conv operators. |
|
Whether to convert FP32 to HF32 for Matmul operators. |
Configure mixed precision mode setting. |
|
|
Path to the configuration file for op precision mode. |
Set the maximum duration of executing an operator in seconds. |
|
Enable debugging options for Ascend operators, disabled by default. |
|
Configure the configuration items for the aclInit interface. |
|
Whether to select online compilation. |
|
Configure aclnn cache parameters. |
|
AOE tuning mode setting, which is not set by default. |
|
Set the parameters specific to Ascend Optimization Engine. |