ó ~9­\c@ stddlmZmZddlZddlmZddlmZddlm Z m Z ddl m Z edƒZ e r¹edƒjZed ƒjZejƒejƒejƒnidgd 6Zd efd „ƒYZd efd„ƒYZgaeƒadadefd„ƒYZdefd„ƒYZdefd„ƒYZd„Z e ddƒdd„ƒZ"dS(iÿÿÿÿ(tprint_functiontdivisionN(t import_module(tPrinter(tSt IndexedBase(tdoctest_depends_ontllvmlites llvmlite.irsllvmlite.bindingt llvm_callabletLLVMJitPrintercB sheZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z RS( sConvert expressions to LLVM IRcO sƒ|jdiƒ|_ts*tdƒ‚ntt|ƒj||Žtjƒ|_ ||_ ||_ ||_ i|_ i|_dS(Nt func_arg_maps'llvmlite is required for LLVMJITPrinter(tpopR Rt ImportErrortsuperR t__init__tllt DoubleTypetfp_typetmoduletbuildertfntext_fnttmp_var(tselfRRRtargstkwargs((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyRs    cC s||j|Return of multiple expressions not supported for this callback( R…R\RlRbRdtNotImplementedErrorR+RRD( RR€R"t tmp_exprst final_exprsRR;R'R‡((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyR|øs  '  "cC sÄtj|ƒ}tjƒ}d|_tjƒ}|j|ƒ|j|ƒtjjƒj ƒ}tj ||ƒ}|j ƒt j |ƒtr®tdƒt|j|ƒƒn|j|jƒ}|S(NitAssembly(tllvmtparse_assemblytcreate_pass_manager_buildert opt_leveltcreate_module_pass_managertpopulatetruntTargettfrom_default_tripletcreate_target_machinetcreate_mcjit_compilertfinalize_objectt exe_enginestappendtFalseRht emit_assemblytget_function_addressRa(RRtllmodtpmbt pass_managerttarget_machinetexe_engtfptr((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyt_compile_function s        ( RARGRRkRoRuRxR‚R~R|R­(((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyRY—s     # tLLVMJitCodeCallbackcB s#eZd„Zd„Zd„ZRS(cC stt|ƒj|ƒdS(N(R R®R(RR\((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyR%scC sŒx…t|ƒD]w\}}t|tƒra|jj||f|j|>> import sympy.printing.llvmjitcode as jit >>> from sympy.abc import a >>> e = a*a + a + 1 >>> e1 = jit.llvm_callable([a], e) >>> e.subs(a, 1.1) # Evaluate via substitution 3.31000000000000 >>> e1(1.1) # Evaluate using JIT-compiled code 3.3100000000000005 Callbacks for integration functions can be JIT compiled. >>> import sympy.printing.llvmjitcode as jit >>> from sympy.abc import a >>> from sympy import integrate >>> from scipy.integrate import quad >>> e = a*a >>> e1 = jit.llvm_callable([a], e, callback_type='scipy.integrate') >>> integrate(e, (a, 0.0, 2.0)) 2.66666666666667 >>> quad(e1, 0.0, 2.0)[0] 2.66666666666667 The 'cubature' callback is for the Python wrapper around the cubature package ( https://github.com/saullocastro/cubature ) and ( http://ab-initio.mit.edu/wiki/index.php/Cubature ) There are two signatures for the SciPy integration callbacks. The first ('scipy.integrate') is the function to be passed to the integration routine, and will pass the signature checks. The second ('scipy.integrate.test') is only useful for directly calling the function using ctypes variables. It will not pass the signature checks for scipy.integrate. The return value from the cse module can also be compiled. This can improve the performance of the compiled function. If multiple expressions are given to cse, the compiled function returns a tuple. The 'cubature' callback handles multiple expressions (set `fdim` to match in the integration call.) >>> import sympy.printing.llvmjitcode as jit >>> from sympy import cse, exp >>> from sympy.abc import x,y >>> e1 = x*x + y*y >>> e2 = 4*(x*x + y*y) + 8.0 >>> after_cse = cse([e1,e2]) >>> after_cse ([(x0, x**2), (x1, y**2)], [x0 + x1, 4*x0 + 4*x1 + 8.0]) >>> j1 = jit.llvm_callable([x,y], after_cse) >>> j1(1.0, 2.0) (5.0, 28.0) s$llvmlite is required for llvmjitcodesscipy.integratesscipy.integrate.testitcubatureisUnknown callback type: %sN(RR R¶RbRgRXRdR£RlRcReR°RfR±t ValueErrorRmR¹t CFUNCTYPE( RR"R·R\Rmtargt arg_ctypetarg_ctypes_formalR¬tcfunc((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyRis:S           (RR»(#t __future__RRRbtsympy.externalRtsympy.printing.printerRtsympyRRtsympy.utilities.decoratorRRtirRtbindingR–t initializetinitialize_native_targettinitialize_native_asmprintert__doctest_requires__R RIR¢tsetRrRqtobjectRYR®R¶R¹RXR(((s9lib/python2.7/site-packages/sympy/printing/llvmjitcode.pyts0     W &