Skip to content

Add Batch Norm - #111

Open
Ziminli wants to merge 12 commits into
devfrom
add_batch_norm
Open

Add Batch Norm#111
Ziminli wants to merge 12 commits into
devfrom
add_batch_norm

Conversation

@Ziminli

@ZiminliZiminli commented Nov 15, 2024

Copy link
Copy Markdown
Collaborator
  • Add the CPU and CUDA implementation and the frontend python test for Batch Normalization
  • modified f32_to_f16 in common_cpu for more accurate behavior

@ZiminliZiminli added 类别：BUG 漏洞修复 模块：算子 添加新的算子 进度：已完成 开发完成，等待审阅合并 labels Nov 15, 2024
@ZiminliZiminli self-assigned this Nov 15, 2024
@kilinchangekilinchange added 类别：开发 新功能开发 and removed 类别：BUG 漏洞修复 labels Dec 5, 2024
kilinchange
kilinchange requested changes Dec 6, 2024
View reviewed changes
Comment threadsrc/ops/batch_norm/cuda/batch_norm.cc Outdated
Comment threadsrc/ops/batch_norm/cpu/batch_norm_cpu.cc Outdated
@kilinchange

kilinchange commented Dec 6, 2024

Copy link
Copy Markdown
Collaborator

在 operators/include/infini_operators.h 加一下头文件引用

PanZezhong1725
PanZezhong1725 requested changes Dec 31, 2024
View reviewed changes
Comment threadoperatorspy/tests/batch_norm.py Outdated
PanZezhong1725
PanZezhong1725 requested changes Jan 22, 2025
View reviewed changes
Comment threadoperatorspy/utils.py Outdated
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@PanZezhong1725PanZezhong1725PanZezhong1725 requested changes
@kilinchangekilinchangekilinchange approved these changes

Requested changes must be addressed to merge this pull request.

Assignees

@ZiminliZiminli

Labels

模块：算子添加新的算子类别：开发新功能开发进度：已完成开发完成，等待审阅合并

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@Ziminli@kilinchange@PanZezhong1725