| 435 | self._autoname = True |
| 436 | |
| 437 | def _process_trace(self, arguments): |
| 438 | from nvidia.dali._debug_mode import _PipelineDebug |
| 439 | |
| 440 | current_pipeline = _PipelineDebug.current() |
| 441 | is_debug = getattr(current_pipeline, "_debug_on", False) |
| 442 | if _dali_trace.is_tracing_enabled() and not is_debug: |
| 443 | if _Pipeline.current(): |
| 444 | start_frame = _Pipeline.current()._definition_frame_start |
| 445 | else: |
| 446 | start_frame = 0 |
| 447 | end_frame = self._op._definition_frame_end |
| 448 | stack_summary = _dali_trace.extract_stack(start_frame=start_frame, end_frame=end_frame) |
| 449 | filenames, linenos, names, lines = _dali_trace.preprocess_stack_summary(stack_summary) |
| 450 | |
| 451 | arguments["_origin_stack_filename"] = filenames |
| 452 | arguments["_origin_stack_lineno"] = linenos |
| 453 | arguments["_origin_stack_name"] = names |
| 454 | arguments["_origin_stack_line"] = lines |
| 455 | |
| 456 | def _generate_outputs(self): |
| 457 | pipeline = _Pipeline.current() |