Browse Source

fix convolution winograd43 vulkan on fixed shape hint (#4973)

tags/20231027
nihui GitHub 2 years ago
parent
commit
7c4969ed58
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/layer/vulkan/convolution_vulkan.cpp

+ 1
- 1
src/layer/vulkan/convolution_vulkan.cpp View File

@@ -444,7 +444,7 @@ int Convolution_vulkan::create_pipeline(const Option& _opt)
specializations[1].i = activation_type;
specializations[2].f = activation_params.w >= 1 ? activation_params[0] : 0.f;
specializations[3].f = activation_params.w == 2 ? activation_params[1] : 0.f;
specializations[4 + 0].i = shape_winograd_gemm_packed.c;
specializations[4 + 0].i = shape_winograd_gemm_packed.h;
specializations[4 + 1].i = shape_winograd_gemm_packed.cstep;
specializations[4 + 2].i = block_x;
specializations[4 + 3].i = block_y;


Loading…
Cancel
Save