Page MenuHomePhabricator

[MLIR][Shape] Lower `shape.get_extent` to `extract_element` when possible
ClosedPublic

Authored by frgossen on Jun 26 2020, 5:03 AM.

Details

Summary

When the origin of a shape is an extent tensor the operation get_extent can be
lowered directly to extract_element.
This choice circumvents the necessity to materialize the shape in memory.

Depends On D82644

Diff Detail

Event Timeline

frgossen created this revision.Jun 26 2020, 5:03 AM
Herald added a project: Restricted Project. · View Herald TranscriptJun 26 2020, 5:03 AM
herhut accepted this revision.Jun 26 2020, 5:14 AM

Nice!

This revision is now accepted and ready to land.Jun 26 2020, 5:14 AM
This revision was automatically updated to reflect the committed changes.