torchvision.ops.nms doesn't work with cuda12.9

[zhumakhan]

🐛 Describe the bug

torchvision.ops.nms doesn't work with cuda12.9

Versions

torch==2.8 && cuda==12.9
torchvision==0.23.0 && cuda==12.9

[NicolasHug]

Hi @zhumakhan can you share more on what exactly doesn't work? What's the error message?