AI/rocm-systems
2
0
انشعاب 0
کد مسائل تقاضاهای واکشی Actions Packages پروژه‌ها انتشارها دانشنامه فعالیت

updated *_wait* APIs to use int rather than roc_shmem_cmps

فهرست منبع
This commit is contained in:
Yiltan Hassan Temucin
2024-10-10 14:13:44 -07:00
والد bcf3fdff10
کامیت 722a5f0731
12فایلهای تغییر یافته به همراه152 افزوده شده و 152 حذف شده
دانلود پرونده وصله دانلود فایل تغییرات diff Expand all files Collapse all files
src/context.hpp
+16 -16
مشاهده پرونده
@@ -74,42 +74,42 @@ class Context {
***************************** DEVICE METHODS *****************************
*************************************************************************/
template <typename T>
__device__ void wait_until(T *ivars, roc_shmem_cmps cmp, T val);
__device__ void wait_until(T *ivars, int cmp, T val);
template <typename T>
__device__ void wait_until_all(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T val);
int cmp, T val);
template <typename T>
__device__ size_t wait_until_any(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T val);
int cmp, T val);
template <typename T>
__device__ size_t wait_until_some(T *ivars, size_t nelems,
size_t* indices,
const int *status,
roc_shmem_cmps cmp, T val);
int cmp, T val);
template <typename T>
__device__ void wait_until_all_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals);
int cmp, T* vals);
template <typename T>
__device__ size_t wait_until_any_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals);
int cmp, T* vals);
template <typename T>
__device__ size_t 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);
template <typename T>
__device__ int test(T *ivars, roc_shmem_cmps cmp, T val);
__device__ int test(T *ivars, int cmp, T val);
__device__ void threadfence_system();
@@ -365,42 +365,42 @@ class Context {
int nreduce);
template <typename T>
__host__ void wait_until(T *ivars, roc_shmem_cmps cmp, T val);
__host__ void wait_until(T *ivars, int cmp, T val);
template <typename T>
__host__ void wait_until_all(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T val);
int cmp, T val);
template <typename T>
__host__ size_t wait_until_any(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T val);
int cmp, T val);
template <typename T>
__host__ size_t wait_until_some(T *ivars, size_t nelems,
size_t* indices,
const int *status,
roc_shmem_cmps cmp, T val);
int cmp, T val);
template <typename T>
__host__ void wait_until_all_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals);
int cmp, T* vals);
template <typename T>
__host__ size_t wait_until_any_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals);
int cmp, T* vals);
template <typename T>
__host__ size_t 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);
template <typename T>
__host__ int test(T *ivars, roc_shmem_cmps cmp, T val);
__host__ int test(T *ivars, int cmp, T val);
public:
/**