
    sh                         S SK Jr  S SKJr  S SKJrJr  S SKJr  \" S5       " S S\5      5       r\	S:X  a  \R                  " 5         g	g	)
    )ir)nvvm)unittestContextResettingTestCase)skip_on_cudasimz*Inline PTX cannot be used in the simulatorc                       \ rS rSrS rSrg)TestCudaInlineAsm   c                    [         R                  " [        5      nSUl        [        R
                  " U5        [         R                  " [         R                  " 5       [         R                  " [         R                  " 5       5      /5      n[         R                  " XS5      n[         R                  " UR                  S5      5      n[         R                  " [         R                  " 5       [         R                  " 5       /5      n[         R                  " USSSS9nUR                  UR                  S   5      nUR!                  Xg/5      nUR#                  XR                  S   5        UR%                  5         [        R&                  " 5       R(                  Ul        [        R*                  " U5        [-        U5      n	[        R.                  " U	5      n
U R1                  S	[-        U
5      ;   5        g )
Nznvptx64-nvidia-cudacu_rsqrtentryzrsqrt.approx.f32 $0, $1;z=f,fT)side_effectr   zrsqrt.approx.f32)r   Module__name__tripler   add_ir_versionFunctionTypeVoidTypePointerType	FloatTypeFunction	IRBuilderappend_basic_block	InlineAsmloadargscallstoreret_voidNVVMdata_layoutset_cuda_kernelstr
compile_ir
assertTrue)selfmodfntyfnbldrrsqrt_approx_fnty	inlineasmvalresnvvmirptxs              |/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudadrv/test_inline_ptx.pytest_inline_rsqrt#TestCudaInlineAsm.test_inline_rsqrt
   sD   ii!*
C r{{}r~~blln/M.NO[[J/||B11':;OOBLLNR\\^<LMLL!2!;!'T;	 ii
#ii	5)

3
# ))+11R Soof%*c#h67     N)r   
__module____qualname____firstlineno__r2   __static_attributes__r5   r4   r1   r	   r	      s    8r4   r	   __main__N)llvmliter   numba.cuda.cudadrvr   numba.cuda.testingr   r   r   r	   r   mainr5   r4   r1   <module>r?      sK     # A . =>80 8 ?86 zMMO r4   