set device in color simulator

This commit is contained in:
James E McClure
2019-04-23 13:50:55 -04:00
parent 254a7c0bf1
commit e458291ce8

View File

@@ -39,6 +39,12 @@ int main(int argc, char **argv)
printf("Running Color LBM \n");
printf("********************************************************\n");
}
// Initialize compute device
int device=ScaLBL_SetDevice(rank);
printf("Using GPU ID %i for rank %i \n",device,rank);
ScaLBL_DeviceBarrier();
MPI_Barrier(comm);
PROFILE_ENABLE(1);
//PROFILE_ENABLE_TRACE();
//PROFILE_ENABLE_MEMORY();