j ddlZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z mZmZddlmZddlmZgd Zd Zd Zd Zd ZdZeddZdZdZe dGddeZdS)N)ContextDecorator)logging) DCU_WHITELIST GCU_WHITELISTMETAX_GPU_WHITELIST MLU_WHITELIST NPU_BLACKLIST XPU_WHITELIST)class_requires_depsfunction_requires_depsis_dep_available)DISABLE_DEV_MODEL_WL) import_paddle) cpugpuxpunpumlugcudcu iluvatar_gpu metax_gpuc|dkr|td|r/dtt|}|d|S|S)Nrz"`device_ids` must be None for CPUs,:) ValueErrorjoinmapstr) device_type device_idss d/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddlex/utils/device.py constr_devicer$,sae 6=>>> XXc#z2233 ,, ,,,ctdr\t}|jr3|jjdkrt ddgSdStdr:ddl}|jr|j jrt ddgStdr._setysV  ; ;HC!BJsO MS993999 : : : : ; ;r%rFLAGS_conv_workspace_size_limit2000r0 auto_growthz0pad3d,pad3d_grad,set_value,set_value_with_tensorTrue)FLAGS_npu_jit_compileFLAGS_use_stride_kernelFLAGS_allocator_strategyCUSTOM_DEVICE_BLACK_LISTFLAGS_npu_scale_aclnnFLAGS_npu_split_aclnnr11800pad3d)BKCL_FORCE_SYNC BKCL_TIMEOUTr\XPU_BLACK_LISTXPU_PADDLE_CONV_FLOATrr\r)r\$FLAGS_use_stream_safe_cuda_allocatorr)rr?is_compiled_with_rocm)r!r3rUrRs r#rHrHus __F;;; e##(D(D(F(F#16: T e##%('*(5(Z%+%+    T e##""'*%%(    T k)))3/ T e##'*47   T e##)3/ T $#r%ctrtjddSd}|dkr|tvsJd|d|zdS|dkr|tvsJd|d|zdS|dkr|t vsJd |d|zdS|d kr|t vsJd |d|zdS|d kr|tvsJd |d|zdS|dkr|tvsJd|d|zdSdS)NznSkip checking if model is supported on device because the flag `PADDLE_PDX_DISABLE_DEV_MODEL_WL` has been set.zWYou could set env `PADDLE_PDX_DISABLE_DEV_MODEL_WL` to `true` to disable this checking.rz%The DCU device does not yet support `z` model!rz%The MLU device does not yet support `rz+The METAX_GPU device does not yet support `rz%The NPU device does not yet support `rz%The XPU device does not yet support `rz%The GCU device does not yet support `) rrwarningrr rr r r)r! model_nametipss r#check_supported_device_typerns |     dDe]*** HJ H H H4 O+****   ]*** HJ H H H4 O+****  # #0000 N* N N NQU U10000   ... HJ H H H4 O/....   ]*** HJ H H H4 O+****   ]*** HJ H H H4 O+**  **r%cFt|\}}t||SrD)r8rn)r+rlr!rIs r#check_supported_devicerps#!&))NK &{J ? ??r%c$eZdZdZdZdZdZdS)TemporaryDeviceChangerz? A context manager to temporarily change global device clt}||_|j|_dSrD)r new_devicer+ get_deviceoriginal_device)selfrtr3s r#__init__zTemporaryDeviceChanger.__init__s.$%}7799r%crt}|j|S|j|j|SrD)rrtr+ set_device)rwr3s r# __enter__z TemporaryDeviceChanger.__enter__s5 ? "K   111 r%crt}|jdS|j|jdS)NF)rrtr+rzrv)rwexc_typeexc_valexc_tbr3s r#__exit__zTemporaryDeviceChanger.__exit__s6 ? "5   !5666ur%N)__name__ __module__ __qualname____doc__rxr{rr%r#rrrrsK:::r%rr)rO contextlibrrcustom_device_listrrrr r r depsr r rflagsr import_guardrr@r$r6r8rFrJrHrnrprrrr%r#rs ''''''POOOOOOOOO''''''''''''      .###2666000 ''**('*Z   D@@@ ^$$-%$r%