Skip to content

Commit

Permalink
Rename back to _debug_mode
Browse files Browse the repository at this point in the history
Signed-off-by: ksztenderski <ksztenderski@nvidia.com>
  • Loading branch information
ksztenderski committed Mar 29, 2022
1 parent 8148a56 commit 2da6e95
Show file tree
Hide file tree
Showing 5 changed files with 4 additions and 4 deletions.
File renamed without changes.
2 changes: 1 addition & 1 deletion dali/python/nvidia/dali/external_source.py
Original file line number Diff line number Diff line change
Expand Up @@ -745,7 +745,7 @@ def external_source(source=None, num_outputs=None, *, cycle=None, name=None, dev
which is more performant.
"""

from nvidia.dali.debug_mode import _PipelineDebug
from nvidia.dali._debug_mode import _PipelineDebug

def _external_source(source=None, num_outputs=None, *, cycle=None, name=None, device="cpu", layout=None,
dtype=None, cuda_stream=None, use_copy_kernel=None, batch=True, **kwargs):
Expand Down
2 changes: 1 addition & 1 deletion dali/python/nvidia/dali/fn.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ def op_wrapper(*inputs, **kwargs):
return op_class(**init_args)(*inputs, **call_args)

def fn_wrapper(*inputs, **kwargs):
from nvidia.dali.debug_mode import _PipelineDebug
from nvidia.dali._debug_mode import _PipelineDebug
current_pipeline = _PipelineDebug.current()
if getattr(current_pipeline, '_debug_on', False):
return current_pipeline._wrap_op_call(op_class, *inputs, **kwargs)
Expand Down
2 changes: 1 addition & 1 deletion dali/python/nvidia/dali/ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -714,7 +714,7 @@ def _build_input_sets(self, inputs):


def _prep_input_sets(op, inputs):
from nvidia.dali.debug_mode import _transform_data_to_tensorlist
from nvidia.dali._debug_mode import _transform_data_to_tensorlist
import nvidia.dali.tensors as tensors

inputs = list(inputs)
Expand Down
2 changes: 1 addition & 1 deletion dali/python/nvidia/dali/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -1328,7 +1328,7 @@ def create_pipeline(*args, **kwargs):


def _pipeline_def_experimental(fn=None, **pipeline_kwargs):
from nvidia.dali.debug_mode import _PipelineDebug
from nvidia.dali._debug_mode import _PipelineDebug
pipeline_debug = pipeline_kwargs.pop('debug', False)
def actual_decorator(func):
@functools.wraps(func)
Expand Down

0 comments on commit 2da6e95

Please sign in to comment.