Floating-point Math Mode#
Overview#
// enums enum dnnl_fpmath_mode_t; // global 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 Documentation#
Global Functions#
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_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_tf32, dnnl_fpmath_mode_any. |
Returns:
dnnl_success on success and a status describing the error otherwise.