Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
AdaptiveAvgPool3d算子已有模板性能优化
NC轴很大,输入D,H,W轴很小的case性能较差
部分泛化场景提升算子性能
对于NC轴很大,输入D,H,W轴很小的场景,使用Gather操作将NC轴转到尾轴提升并行度,从而优化性能
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Backgroud(背景信息)
AdaptiveAvgPool3d算子已有模板性能优化
Origin(信息来源)
NC轴很大,输入D,H,W轴很小的case性能较差
Benefit / Necessity (价值/作用)
部分泛化场景提升算子性能
Design(设计方案)
对于NC轴很大,输入D,H,W轴很小的场景,使用Gather操作将NC轴转到尾轴提升并行度,从而优化性能