[2025秋季][T1-1-4] PPPoint-t #956

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

Jump to bottom

Open

PPPoint-t wants to merge 2 commits into InfiniTensor:main from PPPoint-t:2025-autumn-PPPoint-t-T1-1-4

PPPoint-t commented Jan 21, 2026 •

edited

Loading

描述

Infinicore 赛题 T1-1-4 : 算子 sum, topk, var, var_mean, all。
GPU 使用 ninetoothed 实现，ntops 仓库pr地址。

记录

topk 因 ntl.sort 不支持同时返回索引，暂用迭代选择法实现
var、var_mean 若使用 welford 方法效果应该更好，暂用 two-pass 来实现
摩尔线程 all 算子的 GPU 实现中，若 out 是不连续输出的使用 copy_ 时会有问题，copy 后仍无法对齐，不连续输出的测例无法通过

测试截图

算子	CPU	NVIDIA	摩尔线程
sum
topk
var
var_mean
all

署名

HONOR_CODE.md
REFERENCE.md

PPPoint-t added 2 commits

January 7, 2026 16:58


          Finish T1-1-4: sum、topk、var、var_mean、all

2ae3b59


          Merge remote-tracking branch 'upstream/main' into 2025-autumn-PPPoint…

829d293

…-t-T1-1-4

PPPoint-t requested a review from a team

January 21, 2026 06:30

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet