U d @sPddlZddlmZmZddZeadaddZddZ d d Z e e dS) N)dyndeplazycCs,tj|std|dSt|dS)aRegisters 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. Unlike dyndep.InitOpsLibrary, this does not actually parse the c++ file and refresh operators until caffe2 is called in a fashion which requires operators. In some large codebases this saves a large amount of time during import. It is safe to use within a program that also uses dyndep.InitOpsLibrary 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)ospathexistsprintformat_LAZY_IMPORTED_DYNDEPSadd)namer =/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/lazy_dyndep.pyRegisterOpsLibrary s rcCs|adS)aRegisters an error handler for errors from registering operators Since the lazy registration may happen at a much later time, having a dedicated error handler allows for custom error handling logic. It is highly recomended to set this to prevent errors from bubbling up in weird parts of the code. Args: handler: a function that takes an exception as a single handler. Returns: None N)_error_handler)handlerr r r SetErrorHandler0srcCsttS)N) _import_lazyrGetImportedOpsLibrariesr r r r rBsrc CsntsdSttD]X}zFztj|ddWn.tk rV}ztrFt|W5d}~XYnXW5t|XqdS)NF)Z trigger_lazy)r listremoverZ InitOpLibrary BaseExceptionr)r er r r rGs r) rZ caffe2.pythonrrrsetr rrrrZRegisterLazyImportr r r r s