From 7aaad059103c93fbc4b866ca50da5b94cb218bdf Mon Sep 17 00:00:00 2001 From: Sarbojit Sarkar Date: Mon, 20 Sep 2021 07:28:25 +0000 Subject: [PATCH] SWDEV-298667 - fix stream operation for Cuda Change-Id: Ida91712e678e324b54293ba48dfca442be390783 --- .../include/hip/nvidia_detail/nvidia_hip_runtime_api.h | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h index 97032b1f72..df8f1a5fa8 100644 --- a/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h +++ b/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h @@ -85,6 +85,12 @@ typedef enum hipMemoryAdvise { #define STREAM_OPS_WAIT_MASK_32 0xFFFFFFFF #define STREAM_OPS_WAIT_MASK_64 0xFFFFFFFFFFFFFFFF +// stream operation flags +#define hipStreamWaitValueGte CU_STREAM_WAIT_VALUE_GEQ +#define hipStreamWaitValueEq CU_STREAM_WAIT_VALUE_EQ +#define hipStreamWaitValueAnd CU_STREAM_WAIT_VALUE_AND +#define hipStreamWaitValueNor CU_STREAM_WAIT_VALUE_NOR + // hipLibraryPropertyType #define hipLibraryPropertyType libraryPropertyType #define HIP_LIBRARY_MAJOR_VERSION MAJOR_VERSION @@ -2380,7 +2386,7 @@ inline static hipError_t hipStreamWriteValue64(hipStream_t stream, inline static hipError_t hipStreamWaitValue32(hipStream_t stream, void* ptr, int32_t value, unsigned int flags, - uint32_t mask) { + uint32_t mask __dparm(0xFFFFFFFF)) { if (value < 0) { printf("Warning! value is negative, CUDA accept positive values\n"); } @@ -2393,7 +2399,7 @@ inline static hipError_t hipStreamWaitValue32(hipStream_t stream, inline static hipError_t hipStreamWaitValue64(hipStream_t stream, void* ptr, int64_t value, unsigned int flags, - uint64_t mask) { + uint64_t mask __dparm(0xFFFFFFFFFFFFFFFF)) { if (value < 0) { printf("Warning! value is negative, CUDA accept positive values\n"); }