README.md 2 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
# Install newer version of rocm-cmake
14
15
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
57
Instructions for using CK as a pre-built kernel library are under ```client_example/```
JD's avatar
JD committed
58
59
60
61
62
63
64
65
66

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