refactor code, add docstring

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.39k stars 102 forks source link

Closed yushangdi closed 2 years ago

yushangdi commented 2 years ago

Refactor the code to have a single command that first create the chrome traces, and then parse it to get the GPU Utilization. Add doc string.

Chillee commented 2 years ago

LGTM!