From f23586375bee3fb9de0c416eb50e3d7eab411b15 Mon Sep 17 00:00:00 2001 From: zjing14 Date: Tue, 11 Apr 2023 07:46:46 -0500 Subject: [PATCH] fixed quant example (#672) Co-authored-by: root [ROCm/composable_kernel commit: c203bf67117ca06b1dbd45b1f88e49c9b8a41db9] --- .../conv2d_fwd_bias_relu_perchannel_quantization.cpp | 4 ++-- .../conv2d_fwd_bias_tanh_perchannel_quantization.cpp | 4 ++-- .../09_quantization/conv2d_fwd_perchannel_quantization.cpp | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/client_example/09_quantization/conv2d_fwd_bias_relu_perchannel_quantization.cpp b/client_example/09_quantization/conv2d_fwd_bias_relu_perchannel_quantization.cpp index cf6807f0dd..a10dd3e006 100644 --- a/client_example/09_quantization/conv2d_fwd_bias_relu_perchannel_quantization.cpp +++ b/client_example/09_quantization/conv2d_fwd_bias_relu_perchannel_quantization.cpp @@ -73,7 +73,7 @@ int main(int argc, char* argv[]) SimpleDeviceMem in(sizeof(InDataType) * N * Hi * Wi * C); SimpleDeviceMem wei(sizeof(WeiDataType) * K * Y * X * C); SimpleDeviceMem bias(sizeof(BiasDataType) * K * Y * X * C); - SimpleDeviceMem requant_scale(sizeof(RequantScaleDataType) * K * Y * X * C); + SimpleDeviceMem requant_scale(sizeof(RequantScaleDataType) * K); SimpleDeviceMem out(sizeof(OutDataType) * N * Ho * Wo * K); using DeviceOp = ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD< @@ -203,4 +203,4 @@ int main(int argc, char* argv[]) } return 0; -} \ No newline at end of file +} diff --git a/client_example/09_quantization/conv2d_fwd_bias_tanh_perchannel_quantization.cpp b/client_example/09_quantization/conv2d_fwd_bias_tanh_perchannel_quantization.cpp index 7a216f027f..a0e1865d32 100644 --- a/client_example/09_quantization/conv2d_fwd_bias_tanh_perchannel_quantization.cpp +++ b/client_example/09_quantization/conv2d_fwd_bias_tanh_perchannel_quantization.cpp @@ -76,7 +76,7 @@ int main(int argc, char* argv[]) SimpleDeviceMem in(sizeof(InDataType) * N * Hi * Wi * C); SimpleDeviceMem wei(sizeof(WeiDataType) * K * Y * X * C); SimpleDeviceMem bias(sizeof(BiasDataType) * K * Y * X * C); - SimpleDeviceMem requant_scale(sizeof(RequantScaleDataType) * K * Y * X * C); + SimpleDeviceMem requant_scale(sizeof(RequantScaleDataType) * K); SimpleDeviceMem out(sizeof(OutDataType) * N * Ho * Wo * K); using DeviceOp = ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD< @@ -206,4 +206,4 @@ int main(int argc, char* argv[]) } return 0; -} \ No newline at end of file +} diff --git a/client_example/09_quantization/conv2d_fwd_perchannel_quantization.cpp b/client_example/09_quantization/conv2d_fwd_perchannel_quantization.cpp index c1c5a651eb..6439c22e7f 100644 --- a/client_example/09_quantization/conv2d_fwd_perchannel_quantization.cpp +++ b/client_example/09_quantization/conv2d_fwd_perchannel_quantization.cpp @@ -69,7 +69,7 @@ int main(int argc, char* argv[]) SimpleDeviceMem in(sizeof(InDataType) * N * Hi * Wi * C); SimpleDeviceMem wei(sizeof(WeiDataType) * K * Y * X * C); - SimpleDeviceMem requant_scale(sizeof(RequantScaleDataType) * K * Y * X * C); + SimpleDeviceMem requant_scale(sizeof(RequantScaleDataType) * K); SimpleDeviceMem out(sizeof(OutDataType) * N * Ho * Wo * K); using DeviceOp = @@ -196,4 +196,4 @@ int main(int argc, char* argv[]) } return 0; -} \ No newline at end of file +}