实现了赛题中的三个算子优化 #14

Closed

wencong wants to merge 3 commits from wencong/GPUKernelContest:dev into main

pull from: wencong/GPUKernelContest:dev

merge into: ccf-ai-infra:main

ccf-ai-infra:main

ccf-ai-infra:S1

Conversation 0 Commits 3 Files Changed 10 +2337 -8

wencong commented 2025-12-09 16:36:33 +08:00

First-time contributor

Copy Link

个人信息

Email: czh50714@gmail.com
赛题ID：43

Benchmark 结果

实验配置

实验机器：曦云 C500 - 64 GB * 1
容器环境：GiteeAI pytorch2.4.0 maca 3.0.0.4 版本

实验结果

Reduce Sum 算子在小数据规模下(1000000)本算子吞吐量略低于参考实现；在大数据规模下(536870912 及以上)本算子吞吐量高于参考实现
Sort 算子在所有数据规模下吞吐量均显著高于参考实现
Top-k 算子在所有数据规模下吞吐量均显著高于参考实现

## 个人信息 Email: czh50714@gmail.com 赛题ID：43 ## Benchmark 结果 实验配置 实验机器：曦云 C500 - 64 GB * 1 容器环境：GiteeAI pytorch2.4.0 maca 3.0.0.4 版本 实验结果 Reduce Sum 算子在小数据规模下(1000000)本算子吞吐量略低于参考实现；在大数据规模下(536870912 及以上)本算子吞吐量高于参考实现 Sort 算子在所有数据规模下吞吐量均显著高于参考实现 Top-k 算子在所有数据规模下吞吐量均显著高于参考实现

wencong added 3 commits 2025-12-09 16:36:34 +08:00

0a10187d26 [feat] implement three optimized kernel

a10e5fdde7 [feat]: implement optimized kernel in my folder

3e3f8f124a implement the sort kernel

wawahejun closed this pull request 2025-12-15 02:20:36 +08:00

Pull request closed

Please reopen this pull request to perform a merge.

Sign in to join this conversation.

Reviewers

No reviewers

Labels

Clear labels

No items

No Label

Milestone

Clear milestone

No items

No Milestone

Projects

Clear projects

No project

Assignees

Clear assignees

No Assignees

1 Participants

Notifications

Subscribe

Due Date

The due date is invalid or out of range. Please use the format 'yyyy-mm-dd'.

No due date set.

Dependencies

No dependencies set.

Reference: ccf-ai-infra/GPUKernelContest#14

No description provided.