U ?h@sHddlmZddlmZmZGdddeZGdddeZddZd S) )ir)Visitor CallVisitorc@s4eZdZdZeddddddgZdd Zd d Zd S) FastFloatBinOpVisitorzb A pass to add fastmath flag to float-binop instruction if they don't have any flags. ZfaddZfsubZfmulZfdivZfremZfcmpcCs ||_dSNflagsselfrr I/var/www/html/venv/lib/python3.8/site-packages/numba/core/fastmathpass.py__init__ szFastFloatBinOpVisitor.__init__cCs.|j|jkr*|js*|jD]}|j|qdSr)opname float_binopsrappendr instrflagr r r visit_Instructions  z'FastFloatBinOpVisitor.visit_InstructionN)__name__ __module__ __qualname____doc__ frozensetrr rr r r r rsrc@s eZdZdZddZddZdS)FastFloatCallVisitorzD A pass to change all float function calls to use fastmath. cCs ||_dSrrr r r r r szFastFloatCallVisitor.__init__cCs2|jttfkr.|jD]}|j|qdSr)typerZ FloatTypeZ DoubleTyperZfastmathaddrr r r visit_Calls zFastFloatCallVisitor.visit_CallN)rrrrr rr r r r rsrcCs&|j}t||t||dS)zC Rewrite the given LLVM module to use fastmath everywhere. N)rrvisitr)modoptionsrr r r rewrite_module%sr!N)ZllvmliterZllvmlite.ir.transformsrrrrr!r r r r s