Skip to content

perf(retrieval): fix NDCG GPU performance regression by replacing torch.unique #3350

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
rclough wants to merge 4 commits into
base: master
Choose a base branch
from
+176 −29

[pre-commit.ci] auto fixes from pre-commit.com hooks

0cc4c42
Select commit
Loading
Failed to load commit list.
Draft

perf(retrieval): fix NDCG GPU performance regression by replacing torch.unique #3350

[pre-commit.ci] auto fixes from pre-commit.com hooks
0cc4c42
Select commit
Loading
Failed to load commit list.
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

View more details on Lit OSS [bot]