mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-15 03:30:11 +00:00
14 lines
522 B
Bash
Executable File
14 lines
522 B
Bash
Executable File
#!/bin/bash
|
|
## GPU visibility
|
|
export HIP_VISIBLE_DEVICES=0
|
|
DRIVER="../build/bin/ckProfiler"
|
|
OP=$1
|
|
DATATYPE=$2
|
|
VERIFY="-v $3"
|
|
INIT=$4
|
|
TIME=$5
|
|
USE=$6
|
|
|
|
######## op datatype updateMovingAverage saveMeanAndInvVariance init time inOutLengths(nhwc) reduceDims verify
|
|
$DRIVER $OP $DATATYPE $USE 0 $INIT $TIME -D 64,64,280,82 -R 1,2,3 $VERIFY
|
|
$DRIVER $OP $DATATYPE $USE 1 $INIT $TIME -D 64,64,280,82 -R 1,2,3 $VERIFY |