search

wkcn / MobulaOP

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

add thread sync and op zoo #88

Closed wkcn closed 4 years ago

wkcn commented 4 years ago
  1. Add the function __syncthreads
  2. Add the function get_num_threads and get_thread_num
  3. Add OP Zoo, which includes RoIAlign and Softmax1D