Franscobec/gpufetch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[v0.02] Add warning message. Hide the waiting message when CUDA initialization is done. Fix compiler warnings

Browse Source
This commit is contained in:
Dr-Noob
2021-08-15 21:56:51 +02:00
parent 2687fa5016
commit 8386052b10
4 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/cuda/cuda.cpp
View File

@@ -6,7 +6,7 @@
#include "uarch.hpp"
#include "../common/global.hpp"
struct cache* get_cache_info(struct gpu_info* gpu, cudaDeviceProp prop) {
struct cache* get_cache_info(cudaDeviceProp prop) {
struct cache* cach = (struct cache*) emalloc(sizeof(struct cache));
cach->L2 = (struct cach*) emalloc(sizeof(struct cach));
@@ -17,7 +17,7 @@ struct cache* get_cache_info(struct gpu_info* gpu, cudaDeviceProp prop) {
return cach;
}
struct topology* get_topology_info(struct gpu_info* gpu, cudaDeviceProp prop) {
struct topology* get_topology_info(cudaDeviceProp prop) {
struct topology* topo = (struct topology*) emalloc(sizeof(struct topology));
topo->streaming_mp = prop.multiProcessorCount;
@@ -63,11 +63,13 @@ struct gpu_info* get_gpu_info() {
struct gpu_info* gpu = (struct gpu_info*) emalloc(sizeof(struct gpu_info));
gpu->pci = NULL;
printf("Waiting for CUDA driver to start...\n");
printf("Waiting for CUDA driver to start...");
fflush(stdout);
int dev = 0;
cudaSetDevice(dev);
cudaDeviceProp deviceProp;
cudaGetDeviceProperties(&deviceProp, dev);
printf("\r");
gpu->freq = deviceProp.clockRate * 1e-3f;
gpu->vendor = GPU_VENDOR_NVIDIA;
@@ -80,9 +82,9 @@ struct gpu_info* get_gpu_info() {
}
gpu->arch = get_uarch_from_cuda(gpu);
gpu->cach = get_cache_info(gpu, deviceProp);
gpu->cach = get_cache_info(deviceProp);
gpu->mem = get_memory_info(gpu, deviceProp);
gpu->topo = get_topology_info(gpu, deviceProp);
gpu->topo = get_topology_info(deviceProp);
gpu->peak_performance = get_peak_performance(gpu);
return gpu;