Skip to content

Add GEMM & Expand - #93

Merged
PanZezhong1725 merged 14 commits into
devfrom
add_gemm
Nov 6, 2024
Merged

Add GEMM & Expand#93
PanZezhong1725 merged 14 commits into
devfrom
add_gemm

Conversation

@Ziminli

@Ziminli Ziminli commented Oct 31, 2024

Copy link
Copy Markdown
Collaborator
  • Add GEMM operator
  • Add Expand operator
  • Add FP32 support for matmul operator
  • Misc. changes to common utilities

@Ziminli Ziminli added 模块:算子 添加新的算子 进度:进行中 还在开发过程中,勿合 类别:开发 新功能开发 labels Oct 31, 2024
@Ziminli Ziminli self-assigned this Oct 31, 2024
@Ziminli Ziminli changed the title Add gemm Add GEMM Oct 31, 2024
@Ziminli Ziminli changed the title Add GEMM Add GEMM & Expand Oct 31, 2024
@Ziminli Ziminli added 进度:已完成 开发完成,等待审阅合并 and removed 进度:进行中 还在开发过程中,勿合 labels Nov 1, 2024
Comment thread operatorspy/tests/expand.py Outdated
((3, 2, 4, 5), (3, 2, 1, 1), None, None),
((32, 256, 112, 112), (32, 256, 112, 1), None, None),
# ((32, 256, 112, 112), (32, 1, 1, 1), None, None),
# ((32, 150, 51200), (32, 150, 1), None, None),

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

这俩测例是为什么要注释掉?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

因为GitHub CI 用CPU跑会比较慢,主要是用来测较大数据规模时性能的,如果不需要可以删掉

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

已删除

Comment thread src/devices/cpu/common_cpu.cc Outdated
Comment thread src/ops/expand/cuda/expand.cc Outdated
Comment thread src/ops/gemm/operator.cc
Comment thread src/ops/matmul/cuda/matmul_cuda.cu Outdated
Comment thread src/ops/matmul/cuda/matmul_cuda.cu Outdated
@Ziminli Ziminli mentioned this pull request Nov 6, 2024
Comment thread src/ops/expand/cuda/expand.cc Outdated
@PanZezhong1725
PanZezhong1725 merged commit 1e74a5f into dev Nov 6, 2024
@PanZezhong1725
PanZezhong1725 deleted the add_gemm branch November 6, 2024 08:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants