Incorrect application of activation with GEMM when reduce_k > 1

artyom-beilis / dlprimitives

Deep Learning Primitives and Mini-Framework for OpenCL

http://blog.dlprimitives.org/

MIT License

169 stars 16 forks source link

Closed artyom-beilis closed 2 years ago

artyom-beilis commented 2 years ago

If reduce_k > 1, activation can not be applied withing same gemm kernel.

Make sure separate kernel in called for inplace activation