Add RPP_AUDIO flag for slice

ROCm · May 20, 2024 · 540735f · 540735f
1 parent a75d775
commit 540735f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/amd_openvx_extensions/amd_rpp/source/tensor/Slice.cpp b/amd_openvx_extensions/amd_rpp/source/tensor/Slice.cpp
@@ -172,6 +172,7 @@ static vx_status VX_CALLBACK processSlice(vx_node node, const vx_reference *para
     SliceLocalData *data = NULL;
     STATUS_ERROR_CHECK(vxQueryNode(node, VX_NODE_LOCAL_DATA_PTR, &data, sizeof(data)));
     refreshSlice(node, parameters, data);
+#if RPP_AUDIO
     if (data->deviceType == AGO_TARGET_AFFINITY_GPU) {
 #if ENABLE_HIP
         rpp_status = rppt_slice_gpu(data->pSrc, data->pSrcGenericDesc, data->pDst, data->pDstGenericDesc, data->pAnchor, data->pShape, data->pFillValues, data->policy, data->pSrcRoi3D, data->handle->rppHandle);
@@ -181,6 +182,9 @@ static vx_status VX_CALLBACK processSlice(vx_node node, const vx_reference *para
         rpp_status = rppt_slice_host(data->pSrc, data->pSrcGenericDesc, data->pDst, data->pDstGenericDesc, data->pAnchor, data->pShape, data->pFillValues, data->policy, data->pSrcRoi3D, data->handle->rppHandle);
         return_status = (rpp_status == RPP_SUCCESS) ? VX_SUCCESS : VX_FAILURE;
     }
+#else
+    return_status = VX_ERROR_NOT_SUPPORTED;
+#endif
     return return_status;
 }