updated *_wait* APIs to use int rather than roc_shmem_cmps

Esse commit está contido em:
Yiltan Hassan Temucin
2024-10-10 14:13:44 -07:00
commit 722a5f0731
12 arquivos alterados com 152 adições e 152 exclusões
+8 -8
Ver Arquivo
@@ -199,7 +199,7 @@ __device__ void Context::broadcast(T *dest, const T *source, int nelems,
}
template <typename T>
__device__ __forceinline__ void Context::wait_until(T *ivars, roc_shmem_cmps cmp,
__device__ __forceinline__ void Context::wait_until(T *ivars, int cmp,
T val) {
while (!test(ivars, cmp, val)) {
}
@@ -221,7 +221,7 @@ template <typename T>
__device__ __forceinline__
size_t Context::wait_until_any(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
// zero nelems error condition
if (!nelems) {
return SIZE_MAX;
@@ -251,7 +251,7 @@ template <typename T>
__device__ __forceinline__
void Context::wait_until_all(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
// zero nelems error condition
if (!nelems) {
return;
@@ -278,7 +278,7 @@ __device__ __forceinline__
size_t Context::wait_until_some(T *ivars, size_t nelems,
size_t* indices,
const int *status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
// zero nelems error condition
if (!nelems) {
return 0;
@@ -313,7 +313,7 @@ template <typename T>
__device__ __forceinline__
void Context::wait_until_all_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
;
}
@@ -321,7 +321,7 @@ template <typename T>
__device__ __forceinline__
size_t Context::wait_until_any_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
return 0;
}
@@ -330,12 +330,12 @@ __device__ __forceinline__
size_t Context::wait_until_some_vector(T *ivars, size_t nelems,
size_t* indices,
const int *status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
return 0;
}
template <typename T>
__device__ __forceinline__ int Context::test(T *ivars, roc_shmem_cmps cmp,
__device__ __forceinline__ int Context::test(T *ivars, int cmp,
T val) {
int ret = 0;
volatile T *vol_ivars = reinterpret_cast<T *>(ivars);