apple / apple/coremltools

[Torch.Export] Test Dynamic Shape Originated from 0/1 Static Shape

Open
#2,307 0 comments 1 reaction 0 assignees View on GitHub
torch.export
Dominant language
Python
Stars
5.4k
Forks
850
Avg merge
4d 5h
Merged PRs (30d)
10

Description

If an axis in the example input has size 1, then torch.export refuse to make that axis dynamic due to 1 being special in broadcast. PyTorch team is actively investigating, see [their discussion](https://docs.google.com/document/d/16VPOa3d-Liikf48teAOmxLc92rgvJdfosIy-yoT38Io/edit?fbclid=IwAR3HNwmmexcitV0pbZm_x1a4ykdXZ9th_eJWK-3hBtVgKnrkmemz6Pm5jRQ#heading=h.ez923tomjvyk)

Once they have the 0/1 shape static -> dynamic issue figured out, we will test our converter accordingly

Contributor guide

Open the contributing guide

Research direction

Start with the linked PyTorch discussion about torch.export dynamic shapes and inspect the repository's converter tests. Once torch.export supports converting axes with static size 0 or 1 to dynamic, add coverage for the converter; done means the relevant conversion tests pass for those cases.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning, testing
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.