pytorch
62f01e2b - [FIX][QAT] Switch to use `kwargs` when `args` is empty (#89778)

Commit
2 years ago
[FIX][QAT] Switch to use `kwargs` when `args` is empty (#89778) Summary: When `ref_node.args` is empty, the QAT will throw index out of range. Here is an example, line 574 is using `tensors = ....` in torch.cat func, which will be treated as `kwargs` {F800357376} f388506954 To fix the issue, we will use the value of the first kwarg if args is empty Test Plan: f388545532 Reviewed By: bigning, lyoka Differential Revision: D41396771 Pull Request resolved: https://github.com/pytorch/pytorch/pull/89778 Approved by: https://github.com/lyoka, https://github.com/houseroad
Author
Committer
Parents
Loading