projects
/
platform
/
upstream
/
pytorch.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
e958ceb
)
fix flake8 (#19243)
author
Zachary DeVito
<zdevito@fb.com>
Sat, 13 Apr 2019 17:01:34 +0000
(10:01 -0700)
committer
Facebook Github Bot
<facebook-github-bot@users.noreply.github.com>
Sat, 13 Apr 2019 17:04:39 +0000
(10:04 -0700)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19243
ghimport-source-id:
ae80aed3a5742df21afb6e55979686220a27cce7
Differential Revision:
D14928670
Pulled By: zdevito
fbshipit-source-id:
20ec0d5c8d6f1c515beb55e2e63eddf3b2fc12dd
test/test_jit.py
patch
|
blob
|
history
diff --git
a/test/test_jit.py
b/test/test_jit.py
index
9f09362
..
1492f66
100644
(file)
--- a/
test/test_jit.py
+++ b/
test/test_jit.py
@@
-252,6
+252,7
@@
def enable_cpu_fuser(fn):
torch._C._jit_override_can_fuse_on_cpu(False)
return wrapper
+
# note: not re-entrant, use unnested only
@contextmanager
def disable_autodiff_subgraph_inlining(enabled=True):
@@
-4507,7
+4508,7
@@
a")
return z
graph = _propagate_shapes(test.graph,
- (torch.zeros(2, 1), torch.zeros(1, 2), torch.zeros(1, 1, 1)), False)
+
(torch.zeros(2, 1), torch.zeros(1, 2), torch.zeros(1, 1, 1)), False)
self.assertTrue(next(graph.outputs()).type() == TensorType.get())
out_op_graph_input()