bda/gpu-mpi-tests
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gt and cmake fixes

Browse Source
This commit is contained in:
Bryce Allen
2021-07-16 22:07:00 -04:00
parent d791b81cb6
commit df5f830a26
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mpi_daxpy_gt.cc
View File

@@ -25,9 +25,8 @@
void set_rank_device(int n_ranks, int rank) {
int n_devices, device, ranks_per_device;
size_t memory_per_rank;
n_devices = gt::backend::device_get_count();
n_devices = gt::backend::clib::device_get_count();
if (n_ranks > n_devices) {
if (n_ranks % n_devices != 0) {
@@ -42,14 +41,14 @@ void set_rank_device(int n_ranks, int rank) {
device = rank;
}
gt::backend::device_set(device);
gt::backend::clib::device_set(device);
}
int main(int argc, char **argv) {
int n = 1024;
int world_size, world_rank;
uint32_t vendor_id;
double a = 2.0;
double sum = 0.0;
@@ -70,6 +69,7 @@ int main(int argc, char **argv) {
}
set_rank_device(world_size, world_rank);
vendor_id = gt::backend::clib::device_get_vendor_id(gt::backend::clib::device_get());
gt::blas::handle_t* h = gt::blas::create();
@@ -87,7 +87,7 @@ int main(int argc, char **argv) {
//printf("%f\n", y[i]);
sum += y[i];
}
printf("%d/%d SUM = %f\n", world_rank, world_size, sum);
printf("%d/%d [%x] SUM = %f\n", world_rank, world_size, vendor_id, sum);
MPI_Finalize();