From 03bbd3c550edf22f000dd025516efb3bc37b6503 Mon Sep 17 00:00:00 2001 From: Yong Wu Date: Tue, 15 Sep 2020 12:46:08 +0800 Subject: [PATCH] fix format --- tests/python/frontend/pytorch/test_forward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index 52e80b7860d97..b803c0fc936f1 100644 --- a/tests/python/frontend/pytorch/test_forward.py +++ b/tests/python/frontend/pytorch/test_forward.py @@ -2385,6 +2385,7 @@ def forward(self, *args): input_data = torch.rand(ishape).float() verify_model(ClampInPlace(min, max).float().eval(), input_data=input_data) + @tvm.testing.uses_gpu def test_forward_copy_(): torch.set_grad_enabled(False) @@ -2910,7 +2911,6 @@ def forward(self, *args): verify_model(Addcmul2().float().eval(), input_data=[input_data, t1, t2]) -@tvm.testing.uses_gpu def test_forward_true_divide(): torch.set_grad_enabled(False)