mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-17 11:30:02 +00:00
Change 1d,2d,... to 1D,2D,... (#997)
[ROCm/composable_kernel commit: 0abc0f87db]
This commit is contained in:
committed by
GitHub
parent
3ab2ba4e58
commit
a3ec2dc3bb
@@ -2,7 +2,7 @@
|
||||
|
||||
## Run ```example_reduce_blockwise```
|
||||
```bash
|
||||
# -D <xxx> : input 3d/4d/5d tensor lengths
|
||||
# -D <xxx> : input 3D/4D/5D tensor lengths
|
||||
# -R <xxx> : reduce dimension ids
|
||||
# -v <x> : verification (0=no, 1=yes)
|
||||
#arg1: data type (0: fp16, 1: fp32, 3: int8, 5: bp16, 6: fp64, 7: int4)
|
||||
@@ -22,7 +22,7 @@ Perf: 0.238063 ms, 264.285 GB/s, DeviceReduceBlockWise<256,M_C4_S1,K_C64_S1,InSr
|
||||
|
||||
## Run ```example_reduce_multiblock_atomic_add```
|
||||
```bash
|
||||
# -D <xxx> : input 3d/4d/5d tensor lengths
|
||||
# -D <xxx> : input 3D/4D/5D tensor lengths
|
||||
# -R <xxx> : reduce dimension ids
|
||||
# -v <x> : verification (0=no, 1=yes)
|
||||
#arg1: data type (0: fp32, 1: fp64)
|
||||
|
||||
@@ -4,7 +4,7 @@ arg1: verification (0=no, 1=yes)
|
||||
arg2: initialization (0=no init, 1=integer value, 2=decimal value)
|
||||
arg3: time kernel (0=no, 1=yes)
|
||||
Following arguments (depending on number of spatial dims):
|
||||
Number of spatial dimensions (1=Conv1d, 2=Conv2d, 3=Conv3d)
|
||||
Number of spatial dimensions (1=Conv1D, 2=Conv2D, 3=Conv3D)
|
||||
G, N, K, C,
|
||||
<filter spatial dimensions>, (ie Y, X for 2D)
|
||||
<input image spatial dimensions>, (ie Hi, Wi for 2D)
|
||||
|
||||
Reference in New Issue
Block a user