Is Kineto planning to support backend extensions?

Question

Is Kineto planning to support backend extensions?

fwenguang opened this issue 5 months ago · comments

Florian commented 5 months ago

Hello, there is 'PrivateUse1' in pytorch to support backend integration. Will Kineto provide similar features?

Aaron Shi · Answer 1 · Wed Jan 31 2024 02:18:14 GMT+0800 (China Standard Time)

Hi, we are open to contributers for that backend. The integration in pytorch was landed by open source contributors.

Florian · Answer 2 · Wed Feb 21 2024 20:32:50 GMT+0800 (China Standard Time)

Thanks for your reply.

I commit a pr #874 that add some ActivityTypes for PrivateUse1.

As a PrivateUse1 user of pytorch. Benefiting from two base classes, IActivityProfilerSession and IActivityProfiler. I have implemented my own kineto plugin. However some additional modifications are still needed. See this pr and pr pytorch/pytorch#120556 for reference.

@aaronenyeshi Could you help review and give some suggestions? Thanks.

Aaron Shi · Answer 3 · Tue Apr 23 2024 23:21:34 GMT+0800 (China Standard Time)

Happy to continue reviewing and landing diffs for PrivateUse1, and we are open to contributions from the open source community. Thank you for your contributions. Closing this issue / question now.