disable cuda graph for output allocator mode

Author: zewenli98

py/torch_tensorrt/dynamo/runtime/_PythonTorchTensorRTModule.py

@@ -545,6 +545,7 @@ def run_cuda_graph() -> torch.Tensor | Tuple[torch.Tensor, ...]:
             return outputs
 
         def run_output_allocator() -> torch.Tensor | Tuple[torch.Tensor, ...]:
+            torch_tensorrt.runtime.set_cudagraphs_mode(False)
             with (
                 torch.autograd.profiler.record_function(
                     "PythonTorchTensorRTModule:ProcessInputs"