Fix uma_dbg registration problem in multithread