o پi@sddlmZddlmZddlZddlmZddlmZmZddl m Z m Z ddl m Z ddlmZejded efd d Zded efd d Ze ddddeded efddZeGdddeZdS))Tuple) dataclassN)Int32Uint32)T dsl_user_op)llvm) ParamsBasexreturncCs@td}d}tdD]}dd|>|@r|st|}d}q |S)N FT)rcutlassrange)r resdoneirC/home/ubuntu/.local/lib/python3.10/site-packages/quack/fast_math.pyclzsrcCs$tdt|}|||d@dkS)Nrr r)rr)r arrr find_log2srlociprbc CsDttjtt|j||dt|j||dgddddtjjdS)Nrzmul.hi.u32 $0, $1, $2;z=r,r,rF)has_side_effectsis_align_stack asm_dialect) rr inline_asmri32rir_value AsmDialectAD_ATT)rrrrrrrumulhi$s"r%c@sneZdZUeed<eed<eed<ededdfddZej dedefdd Z dede eeffd d Z d S) FastDivmoddivisor multiplier shift_rightr cCsNtdt|}t|}ttd|>|d|}t|d}t|||S)zConstruct the FastDivmod object, in host code. This precomputes some values based on the divisor and is computationally expensive. rr r )rrrUint64r&)r'p divisor_u32r(r)rrrcreate:s   zFastDivmod.createdividendcCs$|jdkrtt||j|j?S|S)Nr )r'rr%r(r))selfr.rrrdivEs zFastDivmod.divcCs ||}|||j}||fS)N)r0r')r/r.quotient remainderrrrdivmodMs zFastDivmod.divmodN) __name__ __module__ __qualname__r__annotations__r staticmethodr-cutejitr0rr3rrrrr&3s  r&)typingr dataclassesrr cutlass.cuter9rrcutlass.cutlass_dslrrcutlass._mlir.dialectsrquack.cute_dsl_utilsr r:rrr%r&rrrrs