KEMBAR78
Inductor cpp wrapper: clean up the hard-coded schema for fusion OPs · Issue #112552 · pytorch/pytorch · GitHub
Skip to content

Inductor cpp wrapper: clean up the hard-coded schema for fusion OPs #112552

@chunyuan-w

Description

@chunyuan-w

🐛 Describe the bug

Inductor cpp wrapper uses hard-coded cpp_op_schema for fusion OPs including:

  • MKLPackedLinear
  • LinearUnary
  • LinearBinary
  • ConvolutionUnary
  • ConvolutionBinary
  • ConvolutionBinaryInplace
  • ConvolutionTransposeUnary
  • QConvPointWisePT2E
  • QConvPointWiseBinaryPT2E
  • QLinearPointwisePT2E
  • QLinearPointwiseBinaryPT2E

As suggested in #112373 (comment), we need to clean up these hard-coded strings and add a general way to create the cpp_op_schema for these OPs.

Error logs

No response

Minified repro

No response

Versions

PyTorch version: 2.2.0a0+git17eaa44

cc @ezyang @msaroufim @bdhirsh @anijain2305 @zou3519 @chauhang @wconstab

Metadata

Metadata

Assignees

Labels

oncall: cpu inductorCPU Inductor issues for Intel team to triageoncall: pt2triagedThis 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