perf(retrieval): fix NDCG GPU performance regression by replacing torch.unique #3350
+176
−29
Lit OSS [bot] / unittests.yml / Lit Job (pytorchlightning/torchmetrics:ubuntu24.04-cuda12.6.3-py3.12-torch2.8)
failed
Mar 31, 2026 in 59m 42s
Job Failed
The CI job failed.
Details
Failure reason: Job ci-run-lightning-ai-torchmetrics-0cc4c42-unittests-yml-lit-job-pytorchlightning-torchmetrics-ubuntu24-04-cuda12-6-3-py3-12-torch2-8-a1c8ca7b failed
Loading