Skip to content

Commit 475a3ac

Browse files
committed
Fix unsafe return value in Unwrap.
Signed-off-by: Michał Zientkiewicz <[email protected]>
1 parent 2cd34ef commit 475a3ac

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

dali/c_api_2/data_objects.h

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -701,18 +701,12 @@ RefCountedPtr<TensorListWrapper<Backend>> Wrap(std::shared_ptr<TensorList<Backen
701701

702702
template <typename Backend>
703703
const std::shared_ptr<Tensor<Backend>> &ITensor::Unwrap() const & {
704-
if (auto *self = dynamic_cast<const TensorWrapper<Backend> *>(this))
705-
return self->NativePtr();
706-
else
707-
return nullptr;
704+
return dynamic_cast<const TensorWrapper<Backend> &>(this).NativePtr();
708705
}
709706

710707
template <typename Backend>
711708
const std::shared_ptr<TensorList<Backend>> &ITensorList::Unwrap() const & {
712-
if (auto *self = dynamic_cast<const TensorListWrapper<Backend> *>(this))
713-
return self->NativePtr();
714-
else
715-
return nullptr;
709+
return dynamic_cast<const TensorListWrapper<Backend> &>(this).NativePtr();
716710
}
717711

718712
ITensor *ToPointer(daliTensor_h handle);

0 commit comments

Comments
 (0)