[dynamo] support itertools.repeat: call_function repeat in skip_files Builtin repeat, skip reason: filename is None #110286
Description
🐛 Describe the bug
self explanatory
we should use the polyfill approach to implement this see also #109284 (comment) #109972 (comment)
Real world use case in torchrec dist_data.py
batch_size_per_feature = list(
itertools.chain.from_iterable(
itertools.repeat(x, local_split) for x in batch_size_per_rank
)
)
Versions
main
cc @msaroufim @wconstab @bdhirsh @anijain2305 @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx @chenyang78 @aakhundov @kadeng