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

[PPPoint-t]

描述

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

记录

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

测试截图

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