search

wkcn / MobulaOP

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

Fix CUDA stream #68

Closed wkcn closed 4 years ago

wkcn commented 4 years ago
  1. Synchronize CUDA stream
  2. Fix the number of CUDA threads