Fixed allocating memory for buffer in TransposeChannelsKernel

hollance / Forge

A neural network toolkit for Metal

MIT License

1.27k stars 172 forks source link

Closed marseel closed 7 years ago

marseel commented 7 years ago

Buffer that was allocated for permutation was too small.

hollance commented 7 years ago

It probably needs to be slices * 4 in case the number of featureChannels is not a multiple of 4.

marseel commented 7 years ago

Yea, true :)

hollance commented 7 years ago

Thanks for fixing this. 😄