How to customize tracing granularity for GraphAppendingTracer?

hi guys, is there any possible way to control the tracing granularity when using GraphAppendingTracer? I find this example will trace through everything, but if I got some functions or modules need to be wrapped using torch.fx.wrap or overloading is_leaf_module, how can I do that?