BVhldZddlZddlZejdZej dddgZdZy)zContains GPU utility functions.Nz6name: ([^,]*), (?:.*compute capability: (\d+)\.(\d+))?gpu_infonamecompute_capabilityc tj|j}|s tddS|j dr4t |j dt |j dfnd}t|j d|S)zReturns the GpuInfo given a DeviceAttributes proto. Args: device_attrs: A DeviceAttributes proto. Returns A gpu_info tuple. Both fields are None if `device_attrs` does not have a valid physical_device_desc field. N)"_PHYSICAL_DEVICE_DESCRIPTION_REGEXsearchphysical_device_descGpuInfogroupint) device_attrsmatchccs T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/gpu_util.py#compute_capability_from_device_descrst - 3 3'' )%  4 5:[[^EKKNSQ01" Q $$)__doc__ collectionsrecompiler namedtupler rrrrsG& &0RZZ=&?" !+ f6J-K L%r