Skip to content

Port addcmul operator from the TH code to Aten #22797

@VitalyFedyunin

Description

@VitalyFedyunin

addcmul is the point-wise math operator so porting if from the TH code to Aten (and TensorIterator) expected to be easy. Such migration will help to clean up the code, simplify dispatch as well as provide immediate 2-3x operator performance gain.

Porting guide: https://github.com/pytorch/pytorch/wiki/TH-to-ATen-porting-guide

Example PR with porting of the adaptive_avg_pool2d: #14714

How to use TensorIterator: https://github.com/pytorch/pytorch/wiki/How-to-use-TensorIterator

Metadata

Metadata

Labels

better-engineeringRelatively self-contained tasks for better engineering contributorsmodule: cpuCPU specific problem (e.g., perf, algorithm)module: cudaRelated to torch.cuda, and CUDA support in generalmodule: portingIssues related to porting TH/THNN legacy to ATen nativetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions