Skip to content

Commit 8a48f2a

Browse files
authored
Merge branch 'master' into pr-test-argmax
2 parents a49c571 + b16501a commit 8a48f2a

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

src/layer/vulkan/deconvolution_vulkan.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -775,7 +775,7 @@ int Deconvolution_vulkan::forward(const VkMat& bottom_blob, VkMat& top_blob, VkC
775775
int wcut = top_blob_bordered.w - output_w;
776776
int hcut = top_blob_bordered.h - output_h;
777777

778-
VkMat crop_param_blob(4, (size_t)4u, 1, opt.staging_vkallocator);
778+
VkMat crop_param_blob(6, (size_t)4u, 1, opt.staging_vkallocator);
779779
int* crop_params = crop_param_blob.mapped();
780780

781781
if (pad_left == -233 || pad_right == -233 || pad_top == -233 || pad_bottom == -233)

src/layer/vulkan/deconvolutiondepthwise_vulkan.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -447,7 +447,7 @@ int DeconvolutionDepthWise_vulkan::forward(const VkMat& bottom_blob, VkMat& top_
447447
int wcut = top_blob_bordered.w - output_w;
448448
int hcut = top_blob_bordered.h - output_h;
449449

450-
VkMat crop_param_blob(4, (size_t)4u, 1, opt.staging_vkallocator);
450+
VkMat crop_param_blob(6, (size_t)4u, 1, opt.staging_vkallocator);
451451
int* crop_params = crop_param_blob.mapped();
452452

453453
if (pad_left == -233 || pad_right == -233 || pad_top == -233 || pad_bottom == -233)
@@ -592,7 +592,7 @@ int DeconvolutionDepthWise_vulkan::forward(const VkMat& bottom_blob, VkMat& top_
592592
int wcut = top_blob_bordered.w - output_w;
593593
int hcut = top_blob_bordered.h - output_h;
594594

595-
VkMat crop_param_blob(4, (size_t)4u, 1, opt.staging_vkallocator);
595+
VkMat crop_param_blob(6, (size_t)4u, 1, opt.staging_vkallocator);
596596
int* crop_params = crop_param_blob.mapped();
597597

598598
if (pad_left == -233 || pad_right == -233 || pad_top == -233 || pad_bottom == -233)

0 commit comments

Comments
 (0)