[PyTorch] Convert sample tuple to list in cudagraph input reuse (#2426)

Convert sample tuple to list in reuse
Signed-off-by: Robin Zhang <robinz@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

transformer_engine/pytorch/graph.py

@@ -218,9 +218,10 @@ def _make_graphed_callables(
         assert (
             is_training
         ), "`_reuse_graph_input_output_buffers` is only available in training mode."
-        assert isinstance(
-            sample_args, list
-        ), "sample_args must be a list for _reuse_graph_input_output_buffers."
+        if isinstance(sample_args, tuple):
+            sample_args = list(sample_args)
+        if isinstance(sample_kwargs, tuple):
+            sample_kwargs = list(sample_kwargs)
 
         # Reorganize args and kwargs for input tensor reuse.
         # fwd_sample_qs is keyed by model chunk index. The value is a queue of tuples.