Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
fb11c945
Commit
fb11c945
authored
Mar 18, 2022
by
Chao Liu
Browse files
change min block per cu to 1
parent
9a17e7fb
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
11 additions
and
11 deletions
+11
-11
example/01_gemm/README.md
example/01_gemm/README.md
+5
-5
example/05_conv2d_fwd/README.md
example/05_conv2d_fwd/README.md
+5
-5
include/ck/config.hpp
include/ck/config.hpp
+1
-1
No files found.
example/01_gemm/README.md
View file @
fb11c945
# Instructions
for ```gemm_xdl``` Example
# Instructions
## Docker script
## Docker script
```
bash
```
bash
...
@@ -13,7 +13,7 @@ rocm/tensorflow:rocm4.3.1-tf2.6-dev \
...
@@ -13,7 +13,7 @@ rocm/tensorflow:rocm4.3.1-tf2.6-dev \
/bin/bash
/bin/bash
```
```
## Build
```gemm_xdl```
## Build
Example
```
bash
```
bash
mkdir
build
&&
cd
build
mkdir
build
&&
cd
build
```
```
...
@@ -30,15 +30,15 @@ cmake \
...
@@ -30,15 +30,15 @@ cmake \
```
```
```
bash
```
bash
make
-j
gemm_xdl
make
-j
example_
gemm_xdl
_fp16
```
```
## Run
```gemm_xdl```
## Run
Example
```
bash
```
bash
#arg1: verification (0=no, 1=yes)
#arg1: verification (0=no, 1=yes)
#arg2: initialization (0=no init, 1=integer value, 2=decimal value)
#arg2: initialization (0=no init, 1=integer value, 2=decimal value)
#arg3: run kernel # of times (>1)
#arg3: run kernel # of times (>1)
./example
/
gemm_xdl 0 1 5
./
bin/
example
_
gemm_xdl
_fp16
0 1 5
```
```
Result (MI100 @ 1087Mhz, 133.5TFlops peak FP16)
Result (MI100 @ 1087Mhz, 133.5TFlops peak FP16)
...
...
example/05_conv2d_fwd/README.md
View file @
fb11c945
# Instructions
for ```conv2d_fwd_xdl``` Example
# Instructions
## Docker script
## Docker script
```
bash
```
bash
...
@@ -13,7 +13,7 @@ rocm/tensorflow:rocm4.3.1-tf2.6-dev \
...
@@ -13,7 +13,7 @@ rocm/tensorflow:rocm4.3.1-tf2.6-dev \
/bin/bash
/bin/bash
```
```
## Build
```conv2d_fwd_xdl```
## Build
Example
```
bash
```
bash
mkdir
build
&&
cd
build
mkdir
build
&&
cd
build
```
```
...
@@ -30,16 +30,16 @@ cmake \
...
@@ -30,16 +30,16 @@ cmake \
```
```
```
bash
```
bash
make
-j
conv2d_fwd_xdl
make
-j
example_
conv2d_fwd_xdl
_fp16
```
```
## Run
```conv2d_fwd_xdl```
## Run
Example
```
bash
```
bash
#arg1: verification (0=no, 1=yes)
#arg1: verification (0=no, 1=yes)
#arg2: initialization (0=no init, 1=integer value, 2=decimal value)
#arg2: initialization (0=no init, 1=integer value, 2=decimal value)
#arg3: run kernel # of times (>1)
#arg3: run kernel # of times (>1)
#arg4 to 18: N, K, C, Y, X, Hi, Wi, Sy, Sx, Dy, Dx, LeftPy, LeftPx, RightPy, RightPx
#arg4 to 18: N, K, C, Y, X, Hi, Wi, Sy, Sx, Dy, Dx, LeftPy, LeftPx, RightPy, RightPx
./example
/
conv2d_fwd_xdl 0 1 5
./
bin/
example
_
conv2d_fwd_xdl
_fp16
0 1 5
```
```
Result (MI100 @ 1087Mhz, 133.5TFlops peak FP16)
Result (MI100 @ 1087Mhz, 133.5TFlops peak FP16)
...
...
include/ck/config.hpp
View file @
fb11c945
...
@@ -21,7 +21,7 @@
...
@@ -21,7 +21,7 @@
#ifdef CK_USE_LAUNCH_BOUNDS
#ifdef CK_USE_LAUNCH_BOUNDS
#define CK_MAX_THREAD_PER_BLOCK 256
#define CK_MAX_THREAD_PER_BLOCK 256
#define CK_MIN_BLOCK_PER_CU
2
#define CK_MIN_BLOCK_PER_CU
1
#endif
#endif
// GPU-specific parameters
// GPU-specific parameters
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment