🚀 The feature, motivation and pitch
With the current implementation of DimensionNode in LTC, each backend has its own implementation (PyTorch/XLA, TorchScript).
At the same time, shape inference builds off of LTC core classes leading to this PR failing to build shape inference implementation for expand.SymInt. Please make the appropriate backend DimensionNode visible to LTC core.
Solution alternatives based on offline chat with @Krovatkin.
- IR core to access the correct backend implementation
- Use multiple inheritance
CC @wconstab @JackCaoG
Alternatives
No response
Additional context
No response
🚀 The feature, motivation and pitch
With the current implementation of
DimensionNodein LTC, each backend has its own implementation (PyTorch/XLA, TorchScript).At the same time, shape inference builds off of LTC core classes leading to this PR failing to build shape inference implementation for
expand.SymInt. Please make the appropriate backendDimensionNodevisible to LTC core.Solution alternatives based on offline chat with @Krovatkin.
CC @wconstab @JackCaoG
Alternatives
No response
Additional context
No response