🐛 Describe the bug
Note we have aten.squeeze.dim in
|
register_op_strategy_map( |
|
aten.squeeze.dim, torch.squeeze, schema_info=RuntimeSchemaInfo(1) |
|
) |
, but not
aten.squeeze.dims.
Versions
PyTorch from main
cc @H-Huang @awgu @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @pragupta @msaroufim @dcci @aditvenk @tianyu-l @XilunWu @SherlockNoMad @ppwwyyxx @chauhang @penguinwu
🐛 Describe the bug
Note we have
aten.squeeze.diminpytorch/torch/distributed/tensor/_ops/_view_ops.py
Lines 764 to 766 in 45fd56f
aten.squeeze.dims.Versions
PyTorch from main
cc @H-Huang @awgu @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @pragupta @msaroufim @dcci @aditvenk @tianyu-l @XilunWu @SherlockNoMad @ppwwyyxx @chauhang @penguinwu