U d@sXddlZddlZddlmZddlmZmZd ddZeZ eZ ddZ d d d Z dS) N)Lock)coreextension_loaderTcCs.tj|std|dSt||ddS)aOLoads a dynamic library that contains custom operators into Caffe2. Since Caffe2 uses static variable registration, you can optionally load a separate .so file that contains custom operators and registers that into the caffe2 core binary. In C++, this is usually done by either declaring dependency during compilation time, or via dynload. This allows us to do registration similarly on the Python side. Args: name: a name that ends in .so, such as "my_custom_op.so". Otherwise, the command will simply be ignored. Returns: None z&Ignoring {} as it is not a valid file.N) trigger_lazy)ospathexistsprintformat _init_impl)namerr 8/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/dyndep.pyInitOpsLibrarys rcCstSN)_IMPORTED_DYNDEPSr r r rGetImportedOpsLibraries*src CsFt8t|tt|W5QRXt|W5QRXdSr) dll_lockraddrZ DlopenGuardctypesCDLLrZRefreshRegisteredOperators)rrr r rr .s   r )T)T) rr threadingrZ caffe2.pythonrrrsetrrrr r r r rs