README.md 1.77 KB
Newer Older
1
2
3
4
5
6
7
8
## Docker script
```bash
docker run                                     \
-it                                            \
--privileged                                   \
--group-add sudo                               \
-w /root/workspace                             \
-v ${PATH_TO_LOCAL_WORKSPACE}:/root/workspace  \
Chao Liu's avatar
Chao Liu committed
9
rocm/tensorflow:rocm5.1-tf2.6-dev              \
10
11
/bin/bash
```
Chao Liu's avatar
Chao Liu committed
12

13
14
15
# Install the new rocm-cmake version
https://github.com/RadeonOpenCompute/rocm-cmake

16
17
18
19
20
21
22
23
24
25
## Build
```bash
mkdir build && cd build
```

```bash
# Need to specify target ID, example below is gfx908 and gfx90a
cmake                                                                 \
-D BUILD_DEV=OFF                                                      \
-D CMAKE_BUILD_TYPE=Release                                           \
26
-D CMAKE_CXX_FLAGS=" --offload-arch=gfx908 --offload-arch=gfx90a -O3" \
27
28
29
30
31
32
33
34
35
36
37
38
39
-D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc                             \
-D CMAKE_PREFIX_PATH=/opt/rocm                                        \
..
```

### Build and Run Examples
```bash
 make -j examples
```
Instructions for running each individual examples are under ```example/```

## Tests
```bash
40
 make -j examples tests
41
42
43
44
45
46
47
48
 make test
```

## Build ckProfiler
```bash
 make -j ckProfiler
```
Instructions for running ckProfiler are under ```profiler/```
JD's avatar
JD committed
49
50
51
52
53
54
55
56
57
58


## Caveat
### Kernel Timing and Verification
CK's own kernel timer will warn up kernel once, and then run it multiple times
to get average kernel time. For some kernels that use atomic add, this will cause
output buffer to be accumulated multiple times, causing verfication failure.
To work around it, do not use CK's own timer and do verification at the same time.
CK's own timer and verification in each example and ckProfiler can be enabled or
disabled from command line.