From 6ffb64ff8bf6a1776537a5a754a65ce67b68e595 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Mon, 25 Nov 2024 15:06:02 +0000 Subject: [PATCH] Fix tests --- tests/brevitas/export/test_torch_qcdq.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/brevitas/export/test_torch_qcdq.py b/tests/brevitas/export/test_torch_qcdq.py index 6019bf417..6333be174 100644 --- a/tests/brevitas/export/test_torch_qcdq.py +++ b/tests/brevitas/export/test_torch_qcdq.py @@ -13,6 +13,7 @@ @requires_pt_ge('1.9.1') @jit_disabled_for_export() +@torch.no_grad() def test_torch_qcdq_wbiol_export( quant_module, quant_module_impl, @@ -57,6 +58,7 @@ def test_torch_qcdq_wbiol_export( @requires_pt_ge('1.9.1') @jit_disabled_for_export() @parametrize('input_signed', [True, False]) +@torch.no_grad() def test_torch_qcdq_avgpool_export(input_signed, output_bit_width): in_size = (1, IN_CH, FEATURES, FEATURES) inp = torch.randn(in_size)