pytorch - 💡(How to fix) Fix Support unbacked in export Dim API. [1 participants]

TL;DR

Enable support for unbacked Named Dimensions in export by modifying the dynamic shapes specification to include the unbacked=True parameter.

Guidance

• Review the provided Google Doc for context and examples of the desired functionality.
• Update the dynamic shapes specification to include the unbacked=True parameter for Named Dimensions, as shown in the example: dynamic_shapes = {"x": {0: Dim("batch1", min=1, max=100, unbacked=True)}}.
• Consider enabling this feature by default for named dims in export, as it is expected to improve user experience without introducing regressions.
• Plan to extend this support to compile in the near future.

Example

dynamic_shapes = {
   "x": {0: Dim("batch1", min=1, max=100, unbacked=True)},
   "y": {0: Dim("batch1", min=1, max=100, unbacked=True)},
}

Notes

The implementation details and potential edge cases are not explicitly stated in the issue, so further review and testing may be necessary to ensure a smooth rollout of this feature.

Recommendation

Apply workaround: Enable support for unbacked Named Dimensions in export, as it is expected to improve user experience without introducing regressions.