diff options
| -rw-r--r-- | VA/baselines/cpu/app_baseline.c | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/VA/baselines/cpu/app_baseline.c b/VA/baselines/cpu/app_baseline.c index 99865e9..9290488 100644 --- a/VA/baselines/cpu/app_baseline.c +++ b/VA/baselines/cpu/app_baseline.c @@ -55,10 +55,10 @@ static T *B_local; /** * @brief compute output in the host */ -static void vector_addition_host(unsigned int nr_elements, int t) { +static void vector_addition_host(unsigned long nr_elements, int t) { omp_set_num_threads(t); #pragma omp parallel for - for (int i = 0; i < nr_elements; i++) { + for (long i = 0; i < nr_elements; i++) { #if NUMA_MEMCPY C[i] = A_local[i] + B_local[i]; #else @@ -69,7 +69,7 @@ static void vector_addition_host(unsigned int nr_elements, int t) { // Params --------------------------------------------------------------------- typedef struct Params { - int input_size; + long input_size; int n_warmup; int n_reps; int exp; @@ -125,7 +125,7 @@ struct Params input_params(int argc, char **argv) { usage(); exit(0); break; - case 'i': p.input_size = atoi(optarg); break; + case 'i': p.input_size = atol(optarg); break; case 'w': p.n_warmup = atoi(optarg); break; case 'e': p.n_reps = atoi(optarg); break; case 'x': p.exp = atoi(optarg); break; @@ -157,7 +157,7 @@ int main(int argc, char **argv) { struct Params p = input_params(argc, argv); - const unsigned int input_size = p.exp == 0 ? p.input_size * p.n_threads : p.input_size; + const unsigned long input_size = p.exp == 0 ? p.input_size * p.n_threads : p.input_size; // Create an input file with arbitrary data. /** @@ -189,7 +189,7 @@ int main(int argc, char **argv) { C = (T*) malloc(input_size * sizeof(T)); #endif - for (unsigned int i = 0; i < input_size; i++) { + for (unsigned long i = 0; i < input_size; i++) { A[i] = (T) (rand()); B[i] = (T) (rand()); } @@ -320,7 +320,7 @@ int main(int argc, char **argv) { if (rep >= p.n_warmup) { #if NUMA_MEMCPY - printf("[::] VA-CPU-MEMCPY | n_threads=%d e_type=%s n_elements=%d" + printf("[::] VA-CPU-MEMCPY | n_threads=%d e_type=%s n_elements=%ld" " numa_node_in=%d numa_node_local=%d numa_node_out=%d numa_node_cpu=%d numa_node_cpu_memcpy=%d numa_distance_in_cpu=%d numa_distance_cpu_out=%d" " | throughput_MBps=%f", nr_threads, XSTR(T), input_size, @@ -332,7 +332,7 @@ int main(int argc, char **argv) { timer.time[0], timer.time[1], timer.time[2], timer.time[3], timer.time[0] + timer.time[1] + timer.time[2] + timer.time[3]); #else - printf("[::] VA-CPU | n_threads=%d e_type=%s n_elements=%d" + printf("[::] VA-CPU | n_threads=%d e_type=%s n_elements=%ld" #if NUMA " numa_node_in=%d numa_node_out=%d numa_node_cpu=%d numa_distance_in_cpu=%d numa_distance_cpu_out=%d" #endif |