[FEATURE + BUG] DGL + ROCm 7 compatibility updates and rocPRIM bug workaround

## Description This PR adds ROCm 7 compatibility with DGL and adds @jamesETsmith 's workaround for the rocPRIM bug. This branch/PR has been successfully tested on the following configurations Rocm | Ubuntu | Python | Pytorch -- | -- | -- | -- 7.0.1 | 24.04 | 3.12 | 2.8 7.0.1 | 24.04 | 3.12 | 2.6 7.0.1 | 22.04 | 3.10 | 2.7 ## Checklist Please feel free to remove inapplicable items for your PR. - [X ] The PR title starts with [$CATEGORY] (such as [NN], [Model], [Doc], [Feature]]) - [ X] I've leverage the [tools](https://docs.google.com/document/d/1iHyj7zlmygKSk5gBPsqIqL5ASPzJSPREaNT_QdsiYA4/edit) to beautify the python and c++ code. - [ X] The PR is complete and small, read the [Google eng practice (CL equals to PR)](https://google.github.io/eng-practices/review/developer/small-cls.html) to understand more about small PR. In DGL, we consider PRs with less than 200 lines of core code change are small (example, test and documentation could be exempted). - [NA ] All changes have test coverage - [ NA] Code is well-documented - [X ] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change - [ NA] Related issue is referred in this PR - [ NA] If the PR is for a new model/paper, I've updated the example index [here](../examples/README.md).