search

ParCoreLab / Snoopie

Multi-GPU communication profiler and visualizer
Other
14 stars 1 forks source link

AMD GPU Support #5

Open naromero77amd opened 3 months ago

naromero77amd commented 3 months ago

Thank you for creating this much needed tool

Has there be any working on extending this tool to support AMD GPUs? Any sense of the level of effort needed to extended this to ROC+HIP.

msasongko17 commented 3 months ago

We do have a plan to extend this tool to support AMD GPUs. If we leverage NVBit for NVIDIA GPUs, one option that we want to try to extend it to AMD GPUs is https://github.com/ROCm/omnitrace . The level of effort could be incremental because exactly the same algorithm might work on AMD GPUs provided that omnitrace supports the capture of accessed data addresses on GPUs. Other changes that we might have to do is to intercept the driver API calls for AMD GPUs that are equivalent to the API calls that we intercept for NVIDIA GPUs.