school/csci5451
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

print out

Browse source
This commit is contained in:
Michael Zhang 2023-12-10 17:38:06 -06:00
parent d8f5fe13b6
commit 584a919ff1
Signed by: michael
GPG key ID: BDA47A31A3C8EE6B
1 changed files with 28 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
assignments/04/km_cuda.cu
View file

@ -112,7 +112,7 @@ int main(int argc, char **argv) {
// Allocate memory on the GPU
CUDACHECK(
cudaMalloc((void **)&centroids, num_clusters * dim * sizeof(float)));
CUDACHECK(cudaMalloc((void **)&clusterMap, N * sizeof(int)));
CUDACHECK(cudaMallocManaged((void **)&clusterMap, N * sizeof(int)));
CUDACHECK(cudaMallocManaged((void **)&clusterCount,
num_clusters * sizeof(unsigned int)));
CUDACHECK(cudaMalloc((void **)&data, N * dim * sizeof(float)));
@ -170,7 +170,7 @@ int main(int argc, char **argv) {
printf("Is dirty: %d\n", *dirtyBit);
#pragma endregion
#pragma region
#pragma region Iteration
int it = 0;
while (*dirtyBit) {
printf("Iteration %d (dirty=%d)\n", it, *dirtyBit);
@ -205,5 +205,31 @@ int main(int argc, char **argv) {
}
#pragma endregion
#pragma region
{
FILE *fp = fopen("clusters.txt", "w");
for (int i = 0; i < N; ++i)
fprintf(fp, "%d\n", clusterMap[i]);
fclose(fp);
}
{
FILE *fp = fopen("centroids.txt", "w");
fprintf(fp, "%d %d\n", num_clusters, dim);
float *line = (float *)malloc(dim * sizeof(float));
for (int i = 0; i < num_clusters; ++i) {
CUDACHECK(cudaMemcpy(line, &centroids[i * dim], dim * sizeof(float),
cudaMemcpyDeviceToHost));
for (int d = 0; d < dim; ++d)
fprintf(fp, "%.3f ", line[d]);
fprintf(fp, "\n");
}
free(line);
fclose(fp);
}
printf("Done.\n");
#pragma endregion
return 0;
}