Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
brodeynewman committed Sep 13, 2024
1 parent 2a1dd0f commit 38e3190
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions client.cu
Original file line number Diff line number Diff line change
Expand Up @@ -145,10 +145,10 @@ nvmlReturn_t nvmlInit_v2()
return send_rpc_message(RPC_nvmlInit_v2);
}

nvmlReturn_t nvmlDeviceGetCountHandler_v2()
nvmlReturn_t nvmlDeviceGetCountHandler_v2(unsigned int* device_count)
{
open_rpc_client();
return send_rpc_message(RPC_nvmlDeviceGetCount_v2);
return send_rpc_message(RPC_nvmlDeviceGetCount_v2, {}, {{device_count, sizeof(unsigned int)}});
}

nvmlReturn_t nvmlShutdown()
Expand Down
6 changes: 3 additions & 3 deletions server.cu
Original file line number Diff line number Diff line change
Expand Up @@ -82,9 +82,9 @@ void *client_handler(void *arg)
printf("Number of devices: %u\n", dcount);
}

if (write(connfd, &result, sizeof(nvmlReturn_t)) < 0)
goto exit;
}
if (write(connfd, &dcount, sizeof(unsigned int)) <= 0)
goto exit;
}
}
}

Expand Down

0 comments on commit 38e3190

Please sign in to comment.