VhUddlZddlmZddlmZddlZdadaeeje d<dZ dejfdZ d Z d Z d Zejd Zy) N)Path)OptionalF_TENSOR_QUEUE_GLOBAL_TESTcftrytttt aday)NT)_TORCHBIND_IMPLS_INITIALIZEDload_torchbind_test_libregister_fake_operatorsregister_fake_classes_empty_tensor_queuerW/home/dcms/DCMS/lib/python3.12/site-packages/torch/testing/_internal/torchbind_impls.pyinit_torchbind_implementationsrs+$ 3 5#' r returnctjjjtjdj dS)Nr)torchclasses_TorchScriptTesting _TensorQueueemptyfill_r r rr r s7 == , , 9 9  %) r ctjjdd}tjjdd}tjjdd}tjjdd}d }d }tjjj j jtjjj|tjjjj jtjjj|tjjjj jtjjjd y) Nz*_TorchScriptTesting::takes_foo_python_metac$|j|SN add_tensor)foozs rfake_takes_fooz/register_fake_operators..fake_takes_foo%s~~a  r z_TorchScriptTesting::queue_popc"|jSr)poptqs rfake_queue_popz/register_fake_operators..fake_queue_pop)s vvxr z_TorchScriptTesting::queue_pushc$|j|Sr)push)r$xs rfake_queue_pushz0register_fake_operators..fake_queue_push-swwqzr z_TorchScriptTesting::queue_sizec"|jSr)sizer#s rfake_queue_sizez0register_fake_operators..fake_queue_size1swwyr cr|j|}|j|}|j|}|||gSrr)rr(abcs rmeta_takes_foo_list_returnz;register_fake_operators..meta_takes_foo_list_return5s8 NN1  NN1  NN1 1ayr cN|j|}|j|}||fSrr)rr(r.r/s rmeta_takes_foo_tuple_returnz.meta_takes_foo_tuple_return;s' NN1  NN1 1v r c$|j|Srr)rr(s rz)register_fake_operators..Jss~~a(r ) rlibrary register_fakeopsrtakes_foo_list_returndefaultpy_impl_C DispatchKeyMetatakes_foo_tuple_return takes_foo)r r%r)r,r1r3s rr r $s^ ]]  !MN!O! ]]  !ABC ]]  !BCD ]]  !BCD   II!!77??GG !! " II!!88@@HH !!!# II!!++33;;EHH.FakeFoor(yc ||_||_yrr(rD)selfr(rDs r__init__z/register_fake_classes..FakeFoo.__init__RsDFDFr c$|dit|SNr dictcls flattend_foos r__obj_unflatten__z8register_fake_classes..FakeFoo.__obj_unflatten__V,l+, ,r c:|j|jz|zSrrF)rGrs rrz1register_fake_classes..FakeFoo.add_tensorZsFFTVVOq( (r N)__name__ __module__ __qualname__intrH classmethodrPrr r rFakeFoorCPs. c c   -  - )r rXz$_TorchScriptTesting::_ContainsTensorcBeZdZdejfdZedZdZy)1register_fake_classes..FakeContainsTensortc||_yrr[)rGr[s rrHz:register_fake_classes..FakeContainsTensor.__init___s DFr c$|dit|SrJrKrMs rrPzCregister_fake_classes..FakeContainsTensor.__obj_unflatten__brQr c|jSrr])rGs rgetz5register_fake_classes..FakeContainsTensor.getfs 66Mr N) rSrTrUrTensorrHrWrPr`r r rFakeContainsTensorrZ]s+ ell   -  - r rb)r_libraryregister_fake_class)rXrbs rr r NsT ^^''(CD ) )E ) ^^''(NO  P r cddl}ddlm}m}m}m}m}|r|jd|s|r td}n|r |d}n|d}tjjt|y)Nr)find_library_location IS_FBCODEIS_MACOS IS_SANDCASTLE IS_WINDOWSz+non-portable load_library call used in testz5//caffe2/test/cpp/jit:test_custom_class_registrationsztorchbind_test.dllzlibtorchbind_test.so) unittest$torch.testing._internal.common_utilsrfrgrhrirjSkipTestrrr8 load_librarystr)rkrfrgrhrirj lib_file_paths rrrjsf MNN )TU -.BC -.DE  II3}-.r c#K |j||d|j|=|jjy#|j|=|jjwxYwwr)r; py_kernels_dispatch_cacheclear) op_overloadkeyfns r_register_py_impl_temporarilyrxsf, C $  " "3 '##))+  " "3 '##))+sA3A(A3)A00A3) contextlibpathlibrtypingrrrr ScriptObject__annotations__rr r r rcontextmanagerrxr r rrsp  %:>8E$6$67> (U//'T8/, ,,r