add tensor subclass transform output to traces

Signed-off-by: Masaki Kozuki <[email protected]>

thunder/__init__.py

@@ -635,6 +635,7 @@ def get_computation_and_inputs(*args, **kwargs):
                     _tensor_subclass_transform_applied = True
             if not _tensor_subclass_transform_applied:
                 computation_trc = flatten_tensor_subclasses(computation_trc)
+                computation_traces.append(computation_trc)
 
             if backward_trc is None:
                 from thunder.executors.passes import transform_for_execution as transform_for_execution_pass