Skip to content

refactor: scatter-gather-autograd-refactor#164

Merged
kilinchange merged 2 commits into
masterfrom
refactor/scatter-gather-autograd-refactor
Jun 9, 2026
Merged

refactor: scatter-gather-autograd-refactor#164
kilinchange merged 2 commits into
masterfrom
refactor/scatter-gather-autograd-refactor

Conversation

@kilinchange

@kilinchange kilinchange commented Jun 3, 2026

Copy link
Copy Markdown
Collaborator

#153 前置 pr,主要内容:

  1. 添加 scatter 算子;
  2. 使用命名空间区分 scatter/gather 与同名通信算子;
  3. 重新组织 misc 文件内的函数。

@kilinchange kilinchange force-pushed the refactor/scatter-gather-autograd-refactor branch from a91cb12 to 1135a56 Compare June 3, 2026 02:01
@kilinchange kilinchange changed the title feat: add scatter operator, distinguish tensor and communication APIs… refactor: scatter-gather-autograd-refactor Jun 3, 2026
@kilinchange kilinchange force-pushed the refactor/scatter-gather-autograd-refactor branch from 1135a56 to d85350f Compare June 3, 2026 07:11
@kilinchange

kilinchange commented Jun 3, 2026

Copy link
Copy Markdown
Collaborator Author

精度测试:
image
性能测试:
image
ctest 测试:
image

@Chamberlain0w0 Chamberlain0w0 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM,仅有一个 check 的部分没和其他部分逻辑对齐

Comment thread infini_train/src/kernels/cuda/scatter.cu
@kilinchange kilinchange force-pushed the refactor/scatter-gather-autograd-refactor branch from 4eb273e to 9810d32 Compare June 9, 2026 01:46
@kilinchange kilinchange merged commit 7d81dbc into master Jun 9, 2026
2 checks passed
@kilinchange kilinchange deleted the refactor/scatter-gather-autograd-refactor branch June 9, 2026 05:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants