h SSKrSSKrSSKrSSKJr SSKJr SSKJrJ r \ "5(aSSK r \"5(aSSK r \RS\S\ S\S4S j5rS \S\4S jrg) N) Generator)Trainer)is_mlflow_availableis_wandb_availabletrainernamereturn)NNNc#F# [R"5nSv [R"5nX2- nSURRSU3U0nSURR ;aB[ Rb1URR(a[ R"U5 SURR ;aW[RbEURR(a)[R"XPRRS9 gggg7f)a A context manager function for profiling a block of code. Results are logged to Weights & Biases or MLflow depending on the trainer's configuration. Args: trainer (`~transformers.Trainer`): Trainer object. name (`str`): Name of the block to be profiled. Used as a key in the logged dictionary. Example: ```python from transformers import Trainer from trl.extras.profiling import profiling_context class MyTrainer(Trainer): def some_method(self): A = np.random.rand(1000, 1000) B = np.random.rand(1000, 1000) with profiling_context(self, "matrix_multiplication"): # Code to profile: simulate a computationally expensive operation result = A @ B # Matrix multiplication ``` Nzprofiling/Time taken: .wandbmlflow)step)time perf_counter __class____name__args report_tor run acceleratoris_main_processlogr log_metricsstate global_step)rr start_timeend_timedurationprofiling_metricss N/home/james-whalen/.local/lib/python3.13/site-packages/trl/extras/profiling.pyprofiling_contextr!s6""$J   "H$H1'2C2C2L2L1MQtfUW_`',,(((UYY-BwGZGZGjGj #$7<<)))fjj.DI\I\IlIl,==3L3LMJm.D)sDD!funccF^[R"T5U4Sj5nU$)a Decorator to profile a function and log execution time using [`extras.profiling.profiling_context`]. Args: func (`callable`): Function to be profiled. Example: ```python from transformers import Trainer from trl.extras.profiling import profiling_decorator class MyTrainer(Trainer): @profiling_decorator def some_method(self): A = np.random.rand(1000, 1000) B = np.random.rand(1000, 1000) # Code to profile: simulate a computationally expensive operation result = A @ B ``` c~>[UTR5 T"U/UQ70UD6sSSS5 $!,(df  g=f)N)r!r)selfrkwargsr"s r wrapper$profiling_decorator..wrapper_s/ tT]] 3.t.v.4 3 3s . <) functoolswraps)r"r's` r profiling_decoratorr+Gs'0__T// N) contextlibr)rcollections.abcr transformersrtransformers.integrationsrrr r contextmanagerstrr!callabler+r,r r5s % M $Nw$Nc$Ni@P6Q$N$NNh8r,