search

ROCm / torch_migraphx

Libraries integrating migraphx with pytorch
BSD 3-Clause "New" or "Revised" License
5 stars 1 forks source link

support zeros_like and add test cases for remove_ops pass #173

Closed shivadbhavsar closed 1 month ago

shivadbhavsar commented 1 month ago

Merge #172 first: the simplification added here exposes a bug in some models that is fixed by that PR

This PR:

  1. Add tests for the remove_ops pass that is run in the dynamo workflow
  2. Add simplification to remove torch.ops.aten.zeros_like.default and replace it with a constant tensor instead