torch-tb-profiler doesn't support HDFS file system
yundai424 opened this issue · comments
Hi team, we notice that torch TB profiler plugin doesn't currently support HDFS https://github.com/pytorch/kineto/blob/main/tb_plugin/torch_tb_profiler/io/file.py#L66 . Right now only local FS, S3, GS and azure blob are allowed.
We would like to have this support added (iiuc the only thing needed is to implement it and register) as our profiling result is stored on HDFS.
Feel free to add the changes needed to support HDFS. We can help review the changes.
@aaronenyeshi sorry for the late response. I've added the changes. Would appreciate it if you could help to review #793 Thanks!
Thanks for landing the fix, closing this now.