AI/rocm-systems
2
0
Forkuj 0
Kod Zgłoszenia Oczekujące zmiany Actions Packages Projekty Wydania Wiki Aktywność

Merge pull request #43 from ROCm/LWPRHMEM-75-API-differences-bug-fix

Przeglądaj źródła 
Lwprhmem 75 api differences bug fix
This commit is contained in:
Yiltan
2024-10-28 15:45:15 -04:00
zatwierdzone przez GitHub
rodzic 7ae0a54550 98afb41263
commit 794b888d69
4 zmienionych plików z 32 dodań i 32 usunięć
Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files
src/gpu_ib/context_ib_host.hpp
+8 -8
Wyświetl plik
@@ -99,42 +99,42 @@ class GPUIBHostContext : public 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__ 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__ 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_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:
/* Pointer to the backend's host interface */
src/gpu_ib/context_ib_tmpl_host.hpp
+8 -8
Wyświetl plik
@@ -115,21 +115,21 @@ __host__ void GPUIBHostContext::to_all(roc_shmem_team_t team, T *dest,
}
template <typename T>
__host__ void GPUIBHostContext::wait_until(T *ivars, roc_shmem_cmps cmp, T val) {
__host__ void GPUIBHostContext::wait_until(T *ivars, int cmp, T val) {
host_interface->wait_until<T>(ivars, cmp, val, context_window_info);
}
template <typename T>
__host__ void GPUIBHostContext::wait_until_all(T *ivars, size_t nelems,
const int* status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
host_interface->wait_until_all<T>(ivars, nelems, status, cmp, val, context_window_info);
}
template <typename T>
__host__ size_t GPUIBHostContext::wait_until_any(T *ivars, size_t nelems,
const int* status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
return host_interface->wait_until_any<T>(ivars, nelems, status, cmp, val, context_window_info);
}
@@ -137,21 +137,21 @@ template <typename T>
__host__ size_t GPUIBHostContext::wait_until_some(T *ivars, size_t nelems,
size_t* indices,
const int* status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
return host_interface->wait_until_some<T>(ivars, nelems, indices, status, cmp, val, context_window_info);
}
template <typename T>
__host__ void GPUIBHostContext::wait_until_all_vector(T *ivars, size_t nelems,
const int* status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
host_interface->wait_until_all_vector<T>(ivars, nelems, status, cmp, vals, context_window_info);
}
template <typename T>
__host__ size_t GPUIBHostContext::wait_until_any_vector(T *ivars, size_t nelems,
const int* status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
return host_interface->wait_until_any_vector<T>(ivars, nelems, status, cmp, vals, context_window_info);
}
@@ -159,12 +159,12 @@ template <typename T>
__host__ size_t GPUIBHostContext::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 host_interface->wait_until_some_vector<T>(ivars, nelems, indices, status, cmp, vals, context_window_info);
}
template <typename T>
__host__ int GPUIBHostContext::test(T *ivars, roc_shmem_cmps cmp, T val) {
__host__ int GPUIBHostContext::test(T *ivars, int cmp, T val) {
return host_interface->test<T>(ivars, cmp, val, context_window_info);
}
src/reverse_offload/context_ro_host.hpp
+8 -8
Wyświetl plik
@@ -148,42 +148,42 @@ class ROHostContext : public 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);
};
} // namespace rocshmem
src/reverse_offload/context_ro_tmpl_host.hpp
+8 -8
Wyświetl plik
@@ -141,20 +141,20 @@ __host__ void ROHostContext::to_all(roc_shmem_team_t team, T *dest,
}
template <typename T>
__host__ void ROHostContext::wait_until(T *ivars, roc_shmem_cmps cmp, T val) {
__host__ void ROHostContext::wait_until(T *ivars, int cmp, T val) {
host_interface->wait_until<T>(ivars, cmp, val, context_window_info);
}
template <typename T>
__host__ size_t ROHostContext::wait_until_any(T *ivars, size_t nelems,
const int *status, roc_shmem_cmps cmp, T val) {
const int *status, int cmp, T val) {
return host_interface->wait_until_any<T>(ivars, nelems, status, cmp, val, context_window_info);
}
template <typename T>
__host__
void ROHostContext::wait_until_all(T *ivars, size_t nelems,
const int *status, roc_shmem_cmps cmp, T val) {
const int *status, int cmp, T val) {
host_interface->wait_until_all(ivars, nelems, status, cmp, val, context_window_info);
}
@@ -163,7 +163,7 @@ __host__
size_t ROHostContext::wait_until_some(T *ivars, size_t nelems,
size_t* indices,
const int *status,
roc_shmem_cmps cmp, T val) {
int cmp, T val) {
return host_interface->wait_until_some(ivars, nelems, indices, status, cmp, val, context_window_info);
}
@@ -171,14 +171,14 @@ template <typename T>
__host__
void ROHostContext::wait_until_all_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
host_interface->wait_until_all_vector(ivars, nelems, status, cmp, vals, context_window_info);
}
template <typename T>
__host__ size_t ROHostContext::wait_until_any_vector(T *ivars, size_t nelems,
const int *status,
roc_shmem_cmps cmp, T* vals) {
int cmp, T* vals) {
return host_interface->wait_until_any_vector(ivars, nelems, status, cmp, vals, context_window_info);
}
@@ -186,12 +186,12 @@ template <typename T>
__host__ size_t ROHostContext::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 host_interface->wait_until_some_vector(ivars, nelems, indices, status, cmp, vals, context_window_info);
}
template <typename T>
__host__ int ROHostContext::test(T *ivars, roc_shmem_cmps cmp, T val) {
__host__ int ROHostContext::test(T *ivars, int cmp, T val) {
return host_interface->test<T>(ivars, cmp, val, context_window_info);
}