search

wkcn / MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit
MIT License
164 stars 21 forks source link

[Bug] Thread Safety #5

Closed wkcn closed 6 years ago

wkcn commented 6 years ago

The function dot_add and linalg_gemm_?? are not thread safe.

wkcn commented 6 years ago

Fix. db016f2 But the performance drops down.

Current implementation is better on CPU. CPU and GPU may need different implementation to get the best performance.