google / google/heir

extend shape inference to support more ops/interfaces

Open
#1,784 0 comments 0 reactions 0 assignees View on GitHub
dialect: secret
Dominant language
MLIR
Stars
906
Forks
171
Avg merge
4d 12h
Merged PRs (30d)
32

Description

the shape inference pass (#1594) used by the frontend supports the `RegionBranchControlFlow` interfaces (e.g., `scf`, `affine`, etc) but not the other interfaces defined in MLIR's ControlFlowInterfaces.h (e.g., the ones used by `cf`).

It also does not support other region-bearing ops that have a mapping between operands and block arguments (e.g., `secret.generic` or `linalg.generic`).

For other ops, it currently also only supports `InferTypeOpInterface` but not other interfaces that might be relevant, such as `ReifyRankedShapedTypeOpInterface` or `LinalgStructuredInterface` used by `linalg` ops.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.