From 79df5c10a660f8a87a2e26a3e08bf66cebfab273 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Thu, 5 Dec 2024 10:54:39 +0000 Subject: [PATCH] precommit --- .../common/learned_round/learned_round_optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas_examples/common/learned_round/learned_round_optimizer.py b/src/brevitas_examples/common/learned_round/learned_round_optimizer.py index 662f9142a..c9927d23b 100644 --- a/src/brevitas_examples/common/learned_round/learned_round_optimizer.py +++ b/src/brevitas_examples/common/learned_round/learned_round_optimizer.py @@ -723,7 +723,7 @@ def skip_full_execution(self, block, next_block, block_forward, cache): # Finally (!), we compute the quantized input of the next block block.eval() if torch.cuda.is_available(): - block.cuda() + block.cuda() next_quant_input = [] pbar = tqdm(range(len(cache)), desc='', leave=False) with torch.no_grad():