Move check mem usage inside trace call (#2715)

Move check mem usage inside trace call (#2715) #2946