提交 6574f616 编写于 作者: Michael Suo's avatar Michael Suo 提交者: Facebook Github Bot

Revert D16554694: [jit] add a test for inline tracing

Differential Revision:
D16554694

Original commit changeset: 0fae4458f18c

fbshipit-source-id: 08aa0c292fa5b2dbdd0d1f0e59f531416edef760
上级 265b498d
......@@ -329,22 +329,6 @@ class TestJit(JitTestCase):
t2 = torch.rand(3, 4)
self.assertEqual(r(t2), m.x + t2)
def test_trace_nested_fn(self):
class TracedInlineDecision(torch.nn.Module):
def forward(self, x, flag):
@torch.jit.script
def make_decision(flag, x):
if flag:
return x
else:
return torch.zeros_like(x)
x = torch.neg(x) # do something random
return make_decision(flag, x)
decision = TracedInlineDecision()
torch.jit.trace(decision, (torch.rand(3, 4), torch.tensor([True], dtype=torch.bool)), check_trace=True)
def test_restore_device(self):
# main purpose is checking map_location works
m = torch.jit.ScriptModule()
......
Markdown 格式
0% or
您添加了 0 到此讨论。请谨慎行事。
先完成此消息的编辑!
想要评论请 注册