pytorch

f04bb519 - [DataPipe] Change DataPipe display name in profiler (#100042)

[DataPipe] Change DataPipe display name in profiler (#100042) Script: ```python from torchdata.datapipes.iter import IterableWrapper from torchdata.dataloader2 import DataLoader2, MultiProcessingReadingService ls = range(16) dp = IterableWrapper(ls).map(fn_2).map(fn_3).map(fn_4) rs = MultiProcessingReadingService(num_workers=0, main_prefetch_cnt=0, worker_prefetch_cnt=0) dl2 = DataLoader2(dp, reading_service=rs) with torch.profiler.profile(activities=[torch.profiler.ProfilerActivity.CPU]) as prof: for x in dl2: pass ``` Output before: ``` --------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ Name Self CPU % Self CPU CPU total % CPU total CPU time avg # of Calls --------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ enumerate(DataPipe)#MapperIterDataPipe 76.37% 1.419ms 213.08% 3.959ms 80.796us 49 enumerate(DataPipe)#IterableWrapperIterDataPipe 12.70% 236.000us 12.70% 236.000us 13.882us 17 ... ``` Output after: ``` ------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ Name Self CPU % Self CPU CPU total % CPU total CPU time avg # of Calls ------------------------------------------------------- ------------ ------------ ------------ ------------ ------------ ------------ Mapper(datapipe=Mapper, fn=fn_4, input_col=None, out... 29.79% 645.000us 99.17% 2.147ms 126.294us 17 Mapper(datapipe=IterableWrapper, fn=fn_2, input_col=... 29.24% 633.000us 42.96% 930.000us 54.706us 17 Mapper(datapipe=Mapper, fn=fn_3, input_col=None, out... 24.76% 536.000us 68.59% 1.485ms 87.353us 17 IterableWrapper(deepcopy=True, iterable=range(0, 16)... 10.58% 229.000us 10.58% 229.000us 13.471us 17 ... ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/100042 Approved by: https://github.com/ejguan