diff options
| author | James Reed <jamesreed@fb.com> | 2019-04-22 20:49:21 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-22 20:55:17 -0700 |
| commit | e7fc7c732cbde822f9490840704b1f57fe86c50a (patch) | |
| tree | c9013ff34788a1d2cb340ec4839c0b850ed7aaae | |
| parent | d2b03512da991e250401ba02c954e7da1ce82cbe (diff) | |
| download | pytorch-e7fc7c732cbde822f9490840704b1f57fe86c50a.tar.gz pytorch-e7fc7c732cbde822f9490840704b1f57fe86c50a.tar.bz2 pytorch-e7fc7c732cbde822f9490840704b1f57fe86c50a.zip | |
Bugfix for fusion device check (#19594)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19594
I missed a callsite
Reviewed By: wanchaol
Differential Revision: D15041457
fbshipit-source-id: eef76ad51bee06a56d31b4ab64f19250fe2ad8f0
| -rw-r--r-- | torch/csrc/jit/passes/graph_fuser.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/torch/csrc/jit/passes/graph_fuser.cpp b/torch/csrc/jit/passes/graph_fuser.cpp index 7debb1675c..833d73cfa3 100644 --- a/torch/csrc/jit/passes/graph_fuser.cpp +++ b/torch/csrc/jit/passes/graph_fuser.cpp @@ -1153,7 +1153,7 @@ struct GraphFuser { } bool canFuseWithConcat(Value* producer, Node* before_check) { - if (!isFusable(producer->node())) { + if (!isFusableDevice(producer) || !isFusable(producer->node())) { return false; } // NB: it is important that this check happens after isFusable, which checks |