README.md 1.91 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
-D CMAKE_CXX_COMPILER=/opt/rocm/bin/hipcc                             \
-D CMAKE_PREFIX_PATH=/opt/rocm                                        \
29
-D CMAKE_INSTALL_PREFIX=${PATH_TO_CK_INSTALL_DIRECTORY}               \
30
31
32
33
34
35
36
37
38
39
40
..
```

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

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

## Build ckProfiler
```bash
 make -j ckProfiler
```
Instructions for running ckProfiler are under ```profiler/```
JD's avatar
JD committed
50

51
52
53
54
55
56
## Install CK
```bash
make install
```

## Using CK as pre-built kernel library
JD's avatar
JD committed
57
58
59
60
61
62
63
64
65

## 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.