Collaboration diagram for Floating-point Math Mode:
Functions | |
| dnnl_status_t DNNL_API | dnnl_get_default_fpmath_mode (dnnl_fpmath_mode_t *mode) |
| dnnl_status_t DNNL_API | dnnl_set_default_fpmath_mode (dnnl_fpmath_mode_t mode) |
Detailed Description
Function Documentation
◆ dnnl_get_default_fpmath_mode()
| dnnl_status_t DNNL_API dnnl_get_default_fpmath_mode | ( | dnnl_fpmath_mode_t * | mode | ) |
Returns the floating-point math mode that will be used by default for all subsequently created primitives.
- Parameters
-
mode Output FP math mode.
- Returns
- dnnl_success on success and a status describing the error otherwise.
◆ dnnl_set_default_fpmath_mode()
| dnnl_status_t DNNL_API dnnl_set_default_fpmath_mode | ( | dnnl_fpmath_mode_t | mode | ) |
Sets the floating-point math mode that will be used by default for all subsequently created primitives.
- Parameters
-
mode FP math mode. The possible values are: dnnl_fpmath_mode_strict, dnnl_fpmath_mode_bf16, dnnl_fpmath_mode_f16, dnnl_fpmath_mode_any.
- Returns
- dnnl_success on success and a status describing the error otherwise.
