tested threshold

Xilinx · Nov 15, 2023 · f7687e9 · f7687e9
1 parent 8736bc7
commit f7687e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/brevitas/graph/gpfq.py b/src/brevitas/graph/gpfq.py
@@ -317,7 +317,7 @@ def single_layer_update(self):
         s = self.layer.quant_weight_scale()
 
         permutation_list = [torch.tensor(range(weight.shape[-1]))]
-        l1_norm = torch.zeros(weight.shape[:-1])
+        l1_norm = torch.zeros(weight.shape[:-1], device=dev)
         for t in range(weight.shape[-1]):
             for group_index in range(self.groups):
                 U[group_index] += torch.matmul(

diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
@@ -212,7 +212,7 @@
     default=None,
     type=int,
     help='Accumulator Bit Width for GPFQ in combination with A2Q (default: None)')
-add_bool_arg(parser, 'gptq', default=True, help='GPTQ (default: enabled)')
+add_bool_arg(parser, 'gptq', default=False, help='GPTQ (default: disabled)')
 add_bool_arg(parser, 'gpfq', default=False, help='GPFQ (default: disabled)')
 add_bool_arg(
     parser, 'gptq-act-order', default=False, help='GPTQ Act order heuristic (default: disabled)')