From e7c54bf3e54b0a122bca5a8d9a261ae349535b91 Mon Sep 17 00:00:00 2001
From: Fabian Grob <grobfab@gmail.com>
Date: Tue, 14 May 2024 14:44:43 +0100
Subject: [PATCH] Rebase

---
 src/brevitas/graph/gpfq.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/brevitas/graph/gpfq.py b/src/brevitas/graph/gpfq.py
index 0eca7fdd6..b7d135478 100644
--- a/src/brevitas/graph/gpfq.py
+++ b/src/brevitas/graph/gpfq.py
@@ -17,6 +17,7 @@
 from brevitas.graph.gpxq import StopFwdException
 from brevitas.graph.gpxq import SUPPORTED_CONV_OP
 import brevitas.nn as qnn
+from brevitas.quant_tensor import QuantTensor
 
 
 class gpfq_mode(gpxq_mode):