Ph` nUddlmZmZddlmZgZeeed<dedefdZdeded eee e e e ffd Z y ) )ListTuple)ChunkStorageMetadata__all__shard1shard2ct|j}t|D]d}|j||j||j|zk\ry|j||j||j|zk\sdyy)zCheck if two shards overlap.FT)lenoffsetsrangesizes)rrndimsis rC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/distributed/checkpoint/resharding.py"_check_shard_metadata_pair_overlapr sy  E 5\ >>! q 1FLLO C C >>! q 1FLLO C C   saved_shard current_shardreturnc6g}tt|j|j|j|jD]T\}\}}}}t ||z||z}|t ||z } ||kDrd} ||z } n||z } d} |j || | | fV|S)a[ Return the overlapping region between saved_shard and current_shard. There returned list has the same number of elements as the tensor's dimension. For each element, we produce a tuple with the following contents: (dimension, `saved_shard` offset, `current_shard` offset, length) Offsets are relative to each shard. r) enumeratezipr r minmaxappend) rrnarrowsdimsaved_shard_offsetcurrent_shard_offsetsaved_shard_sizecurrent_shard_size min_range_endlengthoffset_for_saved_tensoroffset_for_current_tensors r+_shards_get_overlap_region_wrt_saved_tensorr&sG      ! !            !1 1 #5 5 %9;M!NN  4 4&' #"%99 &';=O&O #() % )+Df M 1 8 NrN) typingrr%torch.distributed.checkpoint.metadatarrstr__annotations__rintr&rrr-sec /C Ma .%.6J. %S#s" #$.r