### Description

refactor memory for tracing
track memory for tasks

### Testing Instructions

<!--
  Give a quick description of steps to test your changes.
-->