-
Notifications
You must be signed in to change notification settings - Fork 0
/
example-1d.cpp
100 lines (64 loc) · 2.11 KB
/
example-1d.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
/*
please note that the series of optmiztion technology is not in official document.
All the tests are based on AMD MI25 radeon instict and AMD ROCm.
*/
#include <assert.h>
#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include<iostream>
#include "hip/hip_runtime.h"
#define HIP_ASSERT(x) (assert((x)==hipSuccess))
#define TOTAL_THREADS (1024*1024*1024)
#define NUM 1
#define THREADS_PER_BLOCK_X 1
#define THREADS_PER_BLOCK_Y 1
#define THREADS_PER_BLOCK_Z 1
#define MAX_THREADS 10
__global__ void
null_kernel(hipLaunchParm lp,
float* __restrict__ a)
{
//int x = hipBlockDim_x * hipBlockIdx_x + hipThreadIdx_x;
//int y = hipBlockDim_y * hipBlockIdx_y + hipThreadIdx_y;
}
using namespace std;
int main() {
float* hostA;
float* deviceA;
hipDeviceProp_t devProp;
hipGetDeviceProperties(&devProp, 0);
cout << " System minor " << devProp.minor << endl;
cout << " System major " << devProp.major << endl;
cout << " agent prop name " << devProp.name << endl;
cout << "hip Device prop succeeded " << endl ;
hipEvent_t start, stop;
hipEventCreate(&start);
hipEventCreate(&stop);
float eventMs = 1.0f;
int i;
int errors;
hostA = (float*)malloc(NUM * sizeof(float));
HIP_ASSERT(hipMalloc((void**)&deviceA, NUM * sizeof(float)));
hipLaunchKernel(null_kernel,
dim3(1, 1,1),
dim3(2, 2, 1),
0, 0,
deviceA);
for (int i = 1; i < (MAX_THREADS + 1); i++) {
hipEventRecord(start, NULL);
hipLaunchKernel(null_kernel,
dim3(TOTAL_THREADS / (i*i*i), 1,1),
dim3(i, i, i),
0, 0,
deviceA);
hipEventRecord(stop, NULL);
hipEventSynchronize(stop);
hipEventElapsedTime(&eventMs, start, stop);
printf("kernel_time (hipEventElapsedTime) =%6.3fms\n", eventMs);
printf("threads_per_block = [%d,%d,%d],Threads_per_cycle for Vega10 - 1.536GHz = % 3d\n", i,i,i, int(TOTAL_THREADS / eventMs / 1.536 / 1e6+ 0.5));
}
HIP_ASSERT(hipFree(deviceA));
free(hostA);
return errors;
}