o i el@sbdZddlZddlmmZddlmZdZedZ edZ dZ dd Z d d Z d ad dZdS)zHContains function to log if devices are compatible with mixed precision.N) tf_loggingz5Mixed precision compatibility check (mixed_float16): OKWARNINGzIf you will use compatible GPU(s) not attached to this host, e.g. by running a multi-worker model, you can ignore this warning. This message will only be logged oncecCsNg}t|D]\}}tt|}|dkr||q|d||fq|S)aGroups together consecutive identical strings. For example, given: ['GPU 1', 'GPU 2', 'GPU 2', 'GPU 3', 'GPU 3', 'GPU 3'] This function returns: ['GPU 1', 'GPU 2 (x2)', 'GPU 3 (x3)'] Args: device_strs: A list of strings, each representing a device. Returns: A copy of the input, but identical consecutive strings are merged into a single string. z%s (x%d)) itertoolsgroupbylenlistappend) device_strsnew_device_strs device_strvalsnumrc/var/www/myenv/lib/python3.10/site-packages/keras/src/mixed_precision/device_compatibility_check.py_dedup_strings"s  rc CsL|dkrdSg}g}|D]4}|dd}|d}|r9|d|dd|d }|d kr3||q ||q ||d q |r}td }|rN|d 7}nt|d krY|d7}n|d7}t||D] }|d|d 7}qc|d7}|t7}t|dS|stdttfdSt|d krtdt |dfdStdt dS)aiLogs a compatibility check if the devices support the policy. Currently only logs for the policy mixed_float16. Args: policy_name: The name of the dtype policy. gpu_details_list: A list of dicts, one dict per GPU. Each dict is the device details for a GPU, as returned by `tf.config.experimental.get_device_details()`. mixed_float16N device_namez Unknown GPUcompute_capabilityz, compute capability r.r)rz4, no compute capability (probably not an Nvidia GPU) zSome of your GPUs may run slowly with dtype policy mixed_float16 because they do not all have compute capability of at least 7.0. Your GPUs: zYour GPU may run slowly with dtype policy mixed_float16 because it does not have compute capability of at least 7.0. Your GPU: zYour GPUs may run slowly with dtype policy mixed_float16 because they do not have compute capability of at least 7.0. Your GPUs: z z^See https://developer.nvidia.com/cuda-gpus for a list of GPUs and their compute capabilities. z%s The dtype policy mixed_float16 may run slowly because this machine does not have a GPU. Only Nvidia GPUs with compute capability of at least 7.0 run quickly with mixed_float16. %sz~%s Your GPU will likely run quickly with dtype policy mixed_float16 as it has compute capability of at least 7.0. Your GPU: %szx%s Your GPUs will likely run quickly with dtype policy mixed_float16 as they all have compute capability of at least 7.0) getr _COMPAT_CHECK_WARNING_PREFIXrr_COMPAT_CHECK_WARNING_SUFFIXrwarninginfo_COMPAT_CHECK_OK_PREFIX) policy_namegpu_details_listsupported_device_strsunsupported_device_strsdetailsnameccr warning_strrrr_log_device_compatibility_check;sl        r'FcCs4trdSdatjd}dd|D}t||dS)zLogs a compatibility check if the devices support the policy. Currently only logs for the policy mixed_float16. A log is shown only the first time this function is called. Args: policy_name: The name of the dtype policy. NTGPUcSsg|] }tjj|qSr)tfconfig experimentalget_device_details).0grrr sz2log_device_compatibility_check..)_logged_compatibility_checkr)r*list_physical_devicesr')rgpusr rrrlog_device_compatibility_checks  r3)__doc__rtensorflow.compat.v2compatv2r)tensorflow.python.platformr_COMPAT_CHECK_PREFIXrrrrr'r0r3rrrrs W