def on_after_backward()

in sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py [0:0]

• 3 lines of code
• 2 McCabe index (conditional complexity)

    def on_after_backward(self):
        if self.backward_nvtx_range:
            nvtx.end_range(self.backward_nvtx_range)