Files
composable_kernel/script/profile_layernorm.sh
2025-05-27 08:49:57 +08:00

16 lines
404 B
Bash
Executable File

#!/bin/bash
## GPU visibility
export HIP_VISIBLE_DEVICES=0
DRIVER="../build/bin/ckProfiler"
OP=$1
DATATYPE=$2
VERIFY=$3
INIT=$4
LOG=$5
TIME=$6
######## op datatype verify init log time length
$DRIVER $OP $DATATYPE $VERIFY $INIT $LOG $TIME --length 256 256
$DRIVER $OP $DATATYPE $VERIFY $INIT $LOG $TIME --length 1024 1024
$DRIVER $OP $DATATYPE $VERIFY $INIT $LOG $TIME --length 4096 4096