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

2
CMakeLists.txt
View File

@@ -17,9 +17,11 @@ find_package(MPI REQUIRED)
add_executable(mpi_daxpy_gt)
target_sources(mpi_daxpy_gt PRIVATE mpi_daxpy_gt.cc)
target_link_libraries(mpi_daxpy_gt gtensor::gtensor)
target_link_libraries(mpi_daxpy_gt gtensor::blas)
target_link_libraries(mpi_daxpy_gt MPI::MPI_CXX)
if ("${GTENSOR_DEVICE}" STREQUAL "cuda")
enable_language(CUDA)
set_source_files_properties(mpi_daxpy_gt.cc
TARGET_DIRECTORY mpi_daxpy_gt
PROPERTIES LANGUAGE CUDA)

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();