Этот коммит содержится в:
Brandon Potter
2024-11-25 14:12:15 -06:00
родитель 308816bc5e
Коммит fd8dbc7fb6
179 изменённых файлов: 5250 добавлений и 5251 удалений
+4 -4
Просмотреть файл
@@ -20,7 +20,7 @@
* IN THE SOFTWARE.
*****************************************************************************/
#include <roc_shmem/roc_shmem.hpp>
#include <rocshmem/rocshmem.hpp>
#include <vector>
#include "tester.hpp"
@@ -37,7 +37,7 @@ int main(int argc, char *argv[]) {
/***
* Select a GPU
*/
int rank = roc_shmem_my_pe();
int rank = rocshmem_my_pe();
int ndevices, my_device = 0;
CHECK_HIP(hipGetDeviceCount(&ndevices));
my_device = rank % ndevices;
@@ -46,7 +46,7 @@ int main(int argc, char *argv[]) {
/**
* Must initialize rocshmem to access arguments needed by the tester.
*/
roc_shmem_init();
rocshmem_init();
/**
* Now grab the arguments from rocshmem.
@@ -76,7 +76,7 @@ int main(int argc, char *argv[]) {
* The rocshmem library needs to be cleaned up with this call. It pairs
* with the init function above.
*/
roc_shmem_finalize();
rocshmem_finalize();
return 0;
}