`i ddlZddlZddlZddlZ ddlZdZn #e$rdZYnwxYwddlmZddl m Z  d dZ ej ddddd Z dS) NTF)_optimize_config)profilerc8ttjsJtsJtsJfd}t j}||||jj |j S)Nc|}j} tj||}|jS#t $r(}t |rtjcYd}~S|d}~wwxYw)N) max_duration) max_total_time_per_trialr benchmark gpu_timesmean Exception isinstancemathinf) trialargsmax_total_timeperfe ignore_erroroptimize_config suggest_func target_funcs n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/optimizing/_optimize.py objectivez_optimize..objectives|E""(A %T@@@D>&&(( (   !\** x  s#/A A7A2*A70A22A7)n_trialstimeout) rr_OptimizationConfigcallableoptuna create_study enqueue_trialoptimize max_trialsr best_trial)rrr default_bestrrstudys``` ` r _optimizer(s o'7'K L LLL L K   L ! !!! !           ! !E  %%% NN +')))  )keypathreadonlyc+vKtstdtj}tj|t |}tj||^tj |r| |n)|r'tj d | |V|J|sH|stj |s||tj|dS#tj|wxYw)aContext manager that optimizes kernel launch parameters. In this context, CuPy's routines find the best kernel launch parameter values (e.g., the number of threads and blocks). The found values are cached and reused with keys as the shapes, strides and dtypes of the given inputs arrays. Args: key (string or None): The cache key of optimizations. path (string or None): The path to save optimization cache records. When path is specified and exists, records will be loaded from the path. When readonly option is set to ``False``, optimization cache records will be saved to the path after the optimization. readonly (bool): See the description of ``path`` option. max_trials (int): The number of trials that defaults to 100. timeout (float): Stops study after the given number of seconds. Default is 1. max_total_time_per_trial (float): Repeats measuring the execution time of the routine for the given number of seconds. Default is 0.1. Examples -------- >>> import cupy >>> from cupyx import optimizing >>> >>> x = cupy.arange(100) >>> with optimizing.optimize(): ... cupy.sum(x) ... array(4950) .. note:: Optuna (https://optuna.org) installation is required. Currently it works for reduction operations only. zbOptuna is required to run optimization. See https://optuna.org/ for the installation instructions.Nzz The specified path {} could not be found, and the readonly option is set. The optimization results will never be stored. )_optuna_available RuntimeErrorrget_current_contextget_new_contextr(set_current_contextosr+existsloadwarningswarnformat _is_dirtysave)r*r+r, config_dict old_contextcontexts rr#r#/sDL J IJJ J#688K.sI{KKG(111  7>>$    LL       M F4LL    :  H   "" #"'..*>*> # T""",[99999,[9999s rCs MMM(''''' $&: te=:=:=:=:=:=:=:s ##