U ?hW ã@spddlZddlmZmZddlmZmZddlZdd„Z dd„Z dZed ƒGd d„deƒƒZe dkrle ¡dS) éN)Úskip_on_cudasimÚCUDATestCase)ÚcudaÚfloat64cCsd|d<dS)Nér©)Úxrrú[/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_optimization.pyÚkernel_funcsr cCs|||S)Nr)rÚyÚzrrr Údevice_funcsr )Z__local_depot0zcall.unizst.param.b64z Simulator does not optimize codec@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)ÚTestOptimizationc Cs\tddd…f}t |¡tƒ}| ¡}tD],}|j|d| |||¡W5QRXq*dS)Nr©Úfragment)rrÚjitr Úinspect_asmÚremoved_by_optÚsubTestÚassertNotIn©ÚselfÚsigÚkernelÚptxrrrr Útest_eager_optszTestOptimization.test_eager_optc Cs`tddd…f}tj|ddtƒ}| ¡}tD],}|j|d| |||¡W5QRXq.dS)NrF©Úoptr)rrrr rrrÚassertInrrrr Útest_eager_noopt!sz!TestOptimization.test_eager_nooptc Cspt t¡}tjdtjd}|d|ƒtt| ¡ ¡ƒƒd}t D](}|j|d| ||¡W5QRXqBdS)Nr©Zdtype©rrr) rrr ÚnpÚzerosrÚnextÚiterrÚitemsrrr©rrrrrrrr Ú test_lazy_opt+s zTestOptimization.test_lazy_optc Csvtjddtƒ}tjdtjd}|d|ƒtt| ¡ ¡ƒƒd}t D](}|j|d| ||¡W5QRXqHdS)NFrrr r!r) rrr r"r#rr$r%rr&rrrr'rrr Útest_lazy_noopt8sz TestOptimization.test_lazy_nooptcCs6tttf}tj|ddtƒ}| |¡}| d|¡dS)NT)Údeviceú fma.rn.f64)rrrr rr©rrr*rrrr Útest_device_optEs z TestOptimization.test_device_optcCs8tttf}tj|dddtƒ}| |¡}| d|¡dS)NTF)r*rr+)rrrr rrr,rrr Útest_device_nooptLs z"TestOptimization.test_device_nooptN) Ú__name__Ú __module__Ú__qualname__rrr(r)r-r.rrrr rs rÚ__main__)Únumpyr"Znumba.cuda.testingrrZnumbarrZunittestr r rrr/Úmainrrrr Ús?