chore: updates

Author: peri044

py/torch_tensorrt/dynamo/conversion/impl/quantize.py

@@ -45,7 +45,9 @@ def quantize(
             max_bound = 448
 
         amax = to_torch(amax, None)
+        print(f"=========== AMAX: {amax}")
         scale = torch.divide(amax, max_bound)
+        print(f"=========== SCALE: {scale}")
         scale = get_trt_tensor(ctx, scale, name + "_scale")
         # Add Q node
         quantize_layer = ctx.net.add_quantize(input_tensor, scale)

tests/py/dynamo/models/test_modelopt_models.py

@@ -100,7 +100,7 @@ def calibrate_loop(model):
     # model has INT8 qdq nodes at this point
     output_pyt = model(input_tensor)
 
-    with torch.no_grad():
+    with torchtrt.logging.debug(), torch.no_grad():
         with export_torch_mode():
             exp_program = torch.export.export(model, (input_tensor,))
             trt_model = torchtrt.dynamo.compile(
@@ -111,6 +111,7 @@ def calibrate_loop(model):
                 cache_built_engines=False,
                 reuse_cached_engines=False,
                 truncate_double=True,
+                debug=True,
             )
             outputs_trt = trt_model(input_tensor)
             assert torch.allclose(output_pyt, outputs_trt, rtol=5e-3, atol=1e-2)