Can pytorch profiler add s3 support

I was looking at the new Pytorch profiler here: Introducing PyTorch Profiler - the new and improved performance tool | PyTorch

I tried it and it worked well at local disk, but seems that tensorboard_trace_handler does not support s3 path as log dir. Can we add s3 support?

This might be a good feature request. Could you create it on GitHub and explain your use case a bit?
Also, would you be interested in implementing it, if the code owners would like to add it?

1 Like