fix error

Author: zewenli98

tests/py/dynamo/lowering/test_decompositions.py

@@ -1855,7 +1855,11 @@ def forward(self, query, key, value, attn_mask=None):
 
         # Validate that the results between Torch and Torch-TRT are similar
         trt_model = torch_tensorrt.dynamo.compile(
-            exported_program, inputs, enabled_precisions={torch.half}, min_block_size=1
+            exported_program,
+            inputs,
+            enabled_precisions={torch.half},
+            min_block_size=1,
+            use_explicit_typing=False,
         )
         torch.testing.assert_close(
             trt_model(*inputs),
@@ -1951,7 +1955,11 @@ def forward(self, query, key, value, attn_mask=None):
 
         # Validate that the results between Torch and Torch-TRT are similar
         trt_model = torch_tensorrt.dynamo.compile(
-            exported_program, inputs, enabled_precisions={torch.half}, min_block_size=1
+            exported_program,
+            inputs,
+            enabled_precisions={torch.half},
+            min_block_size=1,
+            use_explicit_typing=False,
         )
 
         inputs = [
@@ -2017,7 +2025,11 @@ def forward(self, query, key, value):
 
         # Validate that the results between Torch and Torch-TRT are similar
         trt_model = torch_tensorrt.dynamo.compile(
-            exported_program, inputs, enabled_precisions={torch.half}, min_block_size=1
+            exported_program,
+            inputs,
+            enabled_precisions={torch.half},
+            min_block_size=1,
+            use_explicit_typing=False,
         )
         torch.testing.assert_close(
             trt_model(*inputs),
@@ -2078,7 +2090,11 @@ def forward(self, query, key, value, attn_bias=None):
 
         # Validate that the results between Torch and Torch-TRT are similar
         trt_model = torch_tensorrt.dynamo.compile(
-            exported_program, inputs, enabled_precisions={torch.half}, min_block_size=1
+            exported_program,
+            inputs,
+            enabled_precisions={torch.half},
+            min_block_size=1,
+            use_explicit_typing=False,
         )
         torch.testing.assert_close(
             trt_model(*inputs),
@@ -2139,7 +2155,11 @@ def forward(self, query, key, value, attn_bias=None):
 
         # Validate that the results between Torch and Torch-TRT are similar
         trt_model = torch_tensorrt.dynamo.compile(
-            exported_program, inputs, enabled_precisions={torch.half}, min_block_size=1
+            exported_program,
+            inputs,
+            enabled_precisions={torch.half},
+            min_block_size=1,
+            use_explicit_typing=False,
         )
         torch.testing.assert_close(
             trt_model(*inputs),