Browse Source

fix unaryop on moltenvk (#6181)

pull/6183/head
nihui GitHub 10 months ago
parent
commit
a1f5d5be47
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/layer/vulkan/shader/unaryop.comp

+ 1
- 1
src/layer/vulkan/shader/unaryop.comp View File

@@ -41,7 +41,7 @@ void main()
if (op_type == 14) res = atan(v);
if (op_type == 15) res = afp(1.f) / v;
#if NCNN_moltenvk
if (op_type == 16) res = afp(tanh(float(v)));
if (op_type == 16) res = afpvec4(tanh(vec4(v)));
#else
if (op_type == 16) res = tanh(v);
#endif


Loading…
Cancel
Save