From e33120786f48c6c83256eb8f729f2d9b52a519f0 Mon Sep 17 00:00:00 2001 From: Chao Liu Date: Mon, 20 Jun 2022 23:34:32 -0500 Subject: [PATCH] update readme and script (#290) [ROCm/composable_kernel commit: ccbd8d907be06fa585e5298824760a959829936c] --- README.md | 2 +- script/profile_conv.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 9d7b578046..f6c933bf5b 100644 --- a/README.md +++ b/README.md @@ -6,7 +6,7 @@ docker run \ --group-add sudo \ -w /root/workspace \ -v ${PATH_TO_LOCAL_WORKSPACE}:/root/workspace \ -rocm/tensorflow:rocm4.3.1-tf2.6-dev \ +rocm/tensorflow:rocm5.1-tf2.6-dev \ /bin/bash ``` diff --git a/script/profile_conv.sh b/script/profile_conv.sh index 0e97ceb6c6..42736dd37f 100755 --- a/script/profile_conv.sh +++ b/script/profile_conv.sh @@ -26,7 +26,7 @@ REPEAT=$9 N=${10} -# Resnet50 from Bing +# Resnet50 (no duplicated layer) ######## op datatype in_layout wei_layout out_layout verify init log repeat N__ K___ C___ Y X Hi__ Wi__ Strides Dilations LeftPads RightPads #$DRIVER $OP $DATATYPE $IN_LAYOUT $WEI_LAYOUT $OUT_LAYOUT $VERIFY $INIT $LOG $REPEAT $N 256 1024 1 1 14 14 1 1 1 1 0 0 0 0 #$DRIVER $OP $DATATYPE $IN_LAYOUT $WEI_LAYOUT $OUT_LAYOUT $VERIFY $INIT $LOG $REPEAT $N 512 1024 1 1 14 14 1 1 1 1 0 0 0 0 @@ -50,7 +50,7 @@ REPEAT=$9 #$DRIVER $OP $DATATYPE $IN_LAYOUT $WEI_LAYOUT $OUT_LAYOUT $VERIFY $INIT $LOG $REPEAT $N 64 8 7 7 224 224 2 2 1 1 3 3 3 3 -# Resnet50 from Bing +# Resnet50 fusion ####### op_________________ datatype in_layout wei_layout out_layout verify init log repeat N__ K___ C_ Y X Hi_ Wi__ Strides Dilations LeftPads RightPads $DRIVER conv_fwd_bias_relu $DATATYPE $IN_LAYOUT $WEI_LAYOUT $OUT_LAYOUT $VERIFY $INIT $LOG $REPEAT $N 64 3 7 7 224 224 2 2 1 1 3 3 3 3 $DRIVER conv_fwd_bias_relu $DATATYPE $IN_LAYOUT $WEI_LAYOUT $OUT_LAYOUT $VERIFY $INIT $LOG $REPEAT $N 64 64 1 1 56 56 1 1 1 1 0 0 0 0