Skip to content

Commit 4d161ff

Browse files
Cheng CaiWei Wei
authored andcommitted
[oncall] fix flaky test (test_quant_dequant_not_fold) (#43)
Summary: Pull Request resolved: https://github.com/pytorch/fx2trt/pull/43 Test `deeplearning.trt.fx2trt_oss.test.quant.test_quant_trt.TestQuantizeFxTRTOps` always fails due to incorrect sample inputs type. Fix it here. See the task for more details. Reviewed By: scottxu0730 Differential Revision: D35275923 fbshipit-source-id: 6695416a1942aa38c93c0a481f315479d7df9b66
1 parent 256abbe commit 4d161ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/quant/test_quant_trt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -752,7 +752,7 @@ def forward(self, x):
752752
return self.relu(self.linear(x))
753753

754754
model = LinearModule().eval()
755-
inputs = torch.rand(8, 5)
755+
inputs = [torch.rand(8, 5)]
756756

757757
prepared = prepare_fx(model, {"": self.qconfig}, backend_config_dict=self.trt_backend_config_dict)
758758
quantized = convert_fx(

0 commit comments

Comments
 (0)