Skip to content

Commit

Permalink
PyBind - removing undefined functions (#107)
Browse files Browse the repository at this point in the history
  • Loading branch information
LakshmiKumar23 authored Feb 27, 2024
1 parent ba29dc9 commit 6efbed4
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions rocAL_pybind/amd/rocal/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
import rocal_pybind as b
import amd.rocal.types as types
import numpy as np
import cupy as cp
#import cupy as cp
import ctypes
import functools
import inspect
Expand Down Expand Up @@ -153,6 +153,8 @@ def copyToExternalTensor(self, array, multiplier, offset, reverse_channels, ten
b.rocalToTensor(self._handle, ctypes.c_void_p(array.data_ptr()), tensor_format, tensor_dtype,
multiplier[0], multiplier[1], multiplier[2], offset[0], offset[1], offset[2], (1 if reverse_channels else 0), self._output_memory_type, max_roi_height, max_roi_width)

"""
NOTE: getOneHotEncodedLabels is not defined. This function does not have a cpp equivalent. Bring back once cupy/fucntion is fixed
def get_one_hot_encoded_labels(self, array, device):
if device == "cpu":
if (isinstance(array, np.ndarray)):
Expand All @@ -166,7 +168,7 @@ def get_one_hot_encoded_labels(self, array, device):
self._handle, array.data.ptr, self._num_classes, 1)
else: # torch tensor
return b.getOneHotEncodedLabels(self._handle, ctypes.c_void_p(array.data_ptr()), self._num_classes, 1)

"""
def set_outputs(self, *output_list):
b.setOutputs(self._handle, len(output_list), output_list)

Expand Down

0 comments on commit 6efbed4

Please sign in to comment.