ϑi^SSKJr SSKrSSKrSSKJrJr SSKrSSKJ r \(a2SSK J r "SS\5r "SS \5r "S S \5r"S S \5rS/rSSSjjrg)) annotationsN) TYPE_CHECKING TypedDict)core) NotRequiredc*\rSrSr%S\S'S\S'Srg)_Kernelboolenablezlist[int] | tuple[int, int] tuning_rangeN__name__ __module__ __qualname____firstlineno____annotations____static_attributes__rX/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/incubate/autotune.pyr r s  11rr c \rSrSr%S\S'Srg)_Layoutr r rNrrrrrrs rrc*\rSrSr%S\S'S\S'Srg) _Dataloader"r r int tuning_stepsrNrrrrrr"s  rrc4\rSrSr%S\S'S\S'S\S'Srg ) _ConfigKernel&zNotRequired[_Kernel]kernelzNotRequired[_Layout]layoutzNotRequired[_Dataloader] dataloaderrNrrrrr!r!&s$$$$,,rr! set_configcJUcR[R"5 [R"5 [RR R SS9 g0n[U[5(aUnOA[U[5(a,[US5n[R"U5nSSS5 SU;aUSnSU;ad[US["5(a6US(a[R"5 O,[R$"5 O[R "S 5 S U;ab[US [&5(a4US n[)U5S :Xde[R*"US US 5 O[R "S5 SU;aoUSnSU;ad[US["5(a6US(a[R"5 O,[R,"5 O[R "S5 SU;aUSnSnSU;a4[US["5(aUSnO[R "S5 SU;a[US[.5(a-[RR R XS5 g[R "S5 [RR R U5 ggg!,(df  GN5=f![a/n[SU35 [R "S5 SnAGNlSnAff=f)a Set the configuration for kernel, layout and dataloader auto-tuning. 1. kernel: When it is enabled, exhaustive search method will be used to select and cache the best algorithm for the operator in the tuning iteration. Tuning parameters are as follows: - enable(bool): Whether to enable kernel tuning. - tuning_range(list): Start and end iteration for auto-tuning. Default: [1, 10]. 2. layout: When it is enabled, the best data layout such as NCHW or NHWC will be determined based on the device and data type. When the origin layout setting is not best, layout transformation will be automatically performed to improve model performance. Layout auto-tuning only supports dygraph mode currently. Tuning parameters are as follows: - enable(bool): Whether to enable layout tuning. 3. dataloader: When it is enabled, the best num_workers will be selected to replace the origin dataloader setting. Tuning parameters are as follows: - enable(bool): Whether to enable dataloader tuning. Args: config (dict|str|None, optional): Configuration for auto-tuning. If it is a dictionary, the key is the tuning type, and the value is a dictionary of the corresponding tuning parameters. If it is a string, the path of a json file will be specified and the tuning configuration will be set by the json file. Default: None, auto-tuning for kernel, layout and dataloader will be enabled. Examples: .. code-block:: python >>> import paddle >>> import json >>> # config is a dict. >>> config = { ... "kernel": { ... "enable": True, ... "tuning_range": [1, 5], ... }, ... "layout": { ... "enable": True, ... }, ... "dataloader": { ... "enable": True, ... } >>> } >>> paddle.incubate.autotune.set_config(config) # type: ignore[arg-type] >>> # config is the path of json file. >>> config_json = json.dumps(config) >>> with open('config.json', 'w') as json_file: ... json_file.write(config_json) >>> paddle.incubate.autotune.set_config('config.json') NT) use_autotunerzLoad config error: z*Use default configuration for auto-tuning.r#r ztThe auto-tuning configuration of the kernel is incorrect.The `enable` should be bool. Use default parameter instead.r rzzThe auto-tuning configuration of the kernel is incorrect.The `tuning_range` should be list. Use default parameter instead.r$ztThe auto-tuning configuration of the layout is incorrect.The `enable` should be bool. Use default parameter instead.r%FzxThe auto-tuning configuration of the dataloader is incorrect.The `enable` should be bool. Use default parameter instead.rz}The auto-tuning configuration of the dataloader is incorrect.The `tuning_steps` should be int. Use default parameter instead.)renable_autotuneenable_layout_autotunepaddleioreaderset_autotune_config isinstancedictstropenjsonload Exceptionprintwarningswarnr disable_autotunelistlenset_autotune_rangedisable_layout_autotuner) config config_dict filehandlee kernel_configr layout_configdataloader_configr(s rr&r&/sx~  ##% ,,$,?K&$ FC  Hfc"j"ii 3 # ;#H- } $-1488 *((*))+ R ] *-7>>,^< <(A---'' QaI X;#H- } $-1488 *//1002 R{"' 5 ( (+H5t<<0:  R . .+N;SAA   44 N"C W   44\B /#U#" H 's+ , MMF G G Hs6 K)K)K) K&!K)&K)) L"3$LL")N)rAz_ConfigKernel | str | NonereturnNone) __future__rr6r:typingrrr. paddle.basertyping_extensionsrr rrr!__all__r&rrrrOs_# + -2)2)i- - .FCr