o i@sHddlmZddlmZmZGdddeZGdddeZddZd S) )ir)Visitor CallVisitorc@s,eZdZdZegdZddZddZdS)FastFloatBinOpVisitorzb A pass to add fastmath flag to float-binop instruction if they don't have any flags. )faddfsubfmulfdivfremfcmpcC ||_dSNflagsselfrrT/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/core/fastmathpass.py__init__  zFastFloatBinOpVisitor.__init__cCs6|j|jvr|js|jD] }|j|q dSdSdSr )opname float_binopsrappendrinstrflagrrrvisit_Instructions  z'FastFloatBinOpVisitor.visit_InstructionN)__name__ __module__ __qualname____doc__ frozensetrrrrrrrrs   rc@s eZdZdZddZddZdS)FastFloatCallVisitorzD A pass to change all float function calls to use fastmath. cCr r rrrrrrrzFastFloatCallVisitor.__init__cCs6|jttfvr|jD] }|j|qdSdSr )typer FloatType DoubleTyperfastmathaddrrrr visit_Calls  zFastFloatCallVisitor.visit_CallN)rrrr rr(rrrrr"s r"cCs&|j}t||t||dS)zC Rewrite the given LLVM module to use fastmath everywhere. N)rrvisitr")modoptionsrrrrrewrite_module%sr,N)llvmliterllvmlite.ir.transformsrrrr"r,rrrrs