From 9643916281dcd35cf2ffc2b7e44cf61fe6561d7b Mon Sep 17 00:00:00 2001 From: nihui Date: Sun, 7 Apr 2019 22:03:08 +0800 Subject: [PATCH] fix fp16s fp16a deconvolution shader --- src/layer/shader/deconvolution_pack4.comp | 8 ++++---- src/layer/shader/deconvolutiondepthwise_group_pack4.comp | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/layer/shader/deconvolution_pack4.comp b/src/layer/shader/deconvolution_pack4.comp index c79aca157..15402a5b5 100644 --- a/src/layer/shader/deconvolution_pack4.comp +++ b/src/layer/shader/deconvolution_pack4.comp @@ -113,10 +113,10 @@ void main() #if NCNN_fp16_storage && !NCNN_fp16_arithmetic // GL_EXT_shader_16bit_storage does not define f16mat4 type :( afpmat4 k = afpmat4( - afpvec4(weight_data[(w_offset + x) * 4 + 0]), - afpvec4(weight_data[(w_offset + x) * 4 + 1]), - afpvec4(weight_data[(w_offset + x) * 4 + 2]), - afpvec4(weight_data[(w_offset + x) * 4 + 3]) + afpvec4(weight_data[w_offset * 4 + 0]), + afpvec4(weight_data[w_offset * 4 + 1]), + afpvec4(weight_data[w_offset * 4 + 2]), + afpvec4(weight_data[w_offset * 4 + 3]) ); #else afpmat4 k = afpmat4(weight_data[w_offset]); diff --git a/src/layer/shader/deconvolutiondepthwise_group_pack4.comp b/src/layer/shader/deconvolutiondepthwise_group_pack4.comp index 6fb091fe6..e1bb17500 100644 --- a/src/layer/shader/deconvolutiondepthwise_group_pack4.comp +++ b/src/layer/shader/deconvolutiondepthwise_group_pack4.comp @@ -122,10 +122,10 @@ void main() #if NCNN_fp16_storage && !NCNN_fp16_arithmetic // GL_EXT_shader_16bit_storage does not define f16mat4 type :( afpmat4 k = afpmat4( - afpvec4(weight_data[(w_offset + x) * 4 + 0]), - afpvec4(weight_data[(w_offset + x) * 4 + 1]), - afpvec4(weight_data[(w_offset + x) * 4 + 2]), - afpvec4(weight_data[(w_offset + x) * 4 + 3]) + afpvec4(weight_data[w_offset * 4 + 0]), + afpvec4(weight_data[w_offset * 4 + 1]), + afpvec4(weight_data[w_offset * 4 + 2]), + afpvec4(weight_data[w_offset * 4 + 3]) ); #else afpmat4 k = afpmat4(weight_data[w_offset]);