lightrft.utils.utils ====================================== .. currentmodule:: lightrft.utils.utils .. automodule:: lightrft.utils.utils get_current_device -------------------------------------------- .. autofunction:: get_current_device print_rank_0 -------------------------------------------- .. autofunction:: print_rank_0 get_torch_profiler ---------------------------------------------------------- .. autofunction:: get_torch_profiler DummyProfile ---------------------------------------------------------- .. autoclass:: DummyProfile :members: __init__, __enter__, __exit__, start, stop, step